From 7397e3683773bfb04ea531e498e3affcd69899fb Mon Sep 17 00:00:00 2001
From: jesopo <github@lolnerd.net>
Date: Mon, 27 Jan 2020 23:22:45 +0000
Subject: [PATCH] add data_directory() func to BaseModule, to get path to data
 file

---
 src/ModuleManager.py | 36 +++++++++++++++++++++---------------
 1 file changed, 21 insertions(+), 15 deletions(-)

diff --git a/src/ModuleManager.py b/src/ModuleManager.py
index 10e4fb58..7878dd13 100644
--- a/src/ModuleManager.py
+++ b/src/ModuleManager.py
@@ -1,5 +1,5 @@
-import enum, gc, glob, importlib, importlib.util, io, inspect, os, sys
-import typing, uuid
+import dataclasses, enum, gc, glob, importlib, importlib.util, io, inspect, os
+import sys, typing, uuid
 from src import Config, EventManager, Exports, IRCBot, Logging, Timers, utils
 
 class ModuleException(Exception):
@@ -42,19 +42,15 @@ class TryReloadResult(object):
         self.success = success
         self.message = message
 
+@dataclasses.dataclass
 class BaseModule(object):
-    def __init__(self,
-            bot: "IRCBot.Bot",
-            events: EventManager.Events,
-            exports: Exports.Exports,
-            timers: Timers.Timers,
-            log: Logging.Log):
-        self.bot = bot
-        self.events = events
-        self.exports = exports
-        self.timers = timers
-        self.log = log
-        self.on_load()
+    definition: "ModuleDefinition"
+    bot: "IRCBot.Bot"
+    events: EventManager.Events
+    exports: Exports.Exports
+    timers: Timers.Timers
+    log: Logging.Log
+
     def on_load(self):
         pass
     def unload(self):
@@ -65,6 +61,15 @@ class BaseModule(object):
     def on_resume(self):
         pass
 
+    def data_directory(self, filename: str):
+        path, filename = os.path.split(filename)
+        path = os.path.join(self.bot.data_directory, "mod-data",
+            self.definition.name, path)
+
+        if not os.path.isdir(path):
+            os.makedirs(path)
+        return os.path.join(path, filename)
+
 class ModuleDefinition(object):
     def __init__(self,
             name: str,
@@ -242,8 +247,9 @@ class ModuleManager(object):
         context_events = self.events.new_context(context)
         context_exports = self.exports.new_context(context)
         context_timers = self.timers.new_context(context)
-        module_object = module_object_pointer(bot, context_events,
+        module_object = module_object_pointer(definition, bot, context_events,
             context_exports, context_timers, self.log)
+        module_object.on_load()
 
         module_title = (getattr(module_object, "_name", None) or
             definition.name.title())