diff options
Diffstat (limited to 'modules/scripts.py')
| -rw-r--r-- | modules/scripts.py | 66 |
1 files changed, 0 insertions, 66 deletions
diff --git a/modules/scripts.py b/modules/scripts.py deleted file mode 100644 index 5b87767d..00000000 --- a/modules/scripts.py +++ /dev/null @@ -1,66 +0,0 @@ - -import glob, json, os, subprocess -from src import IRCObject, ModuleManager, utils - -class Module(ModuleManager.BaseModule): - def on_load(self): - self._directory = os.path.join(self.bot.directory, "modules", "scripts") - self._hooks = [] - self._load_scripts() - - def _load_scripts(self): - for filename in glob.glob(os.path.join(self._directory, "*")): - name = os.path.basename(filename) - for hashflag, value in utils.parse.hashflags(filename): - if hashflag == "name" and value: - name = value - elif hashflag == "hook" and value: - hook_fn = self._make_hook(filename, name) - hook = self.events.on(value).hook(hook_fn) - self._hooks.append([value, hook]) - - def _make_hook(self, filename, name): - return lambda event: self.call(event, filename, name) - - @utils.hook("received.command.reloadscripts", permission="reloadscripts") - def reload(self, event): - for event_name, hook in self._hooks: - self.events.on(event_name).unhook(hook) - self._load_scripts() - event["stdout"].write("Reloaded all scripts") - - def call(self, event, filename, name): - env = { - "HOME": os.environ["HOME"], - "PATH": os.environ["PATH"] - } - - env["EVENT"] = event.name - for key, value in event.kwargs.items(): - if isinstance(value, (str,)): - env[key.upper()] = value - elif isinstance(value, (bool,)): - env[key.upper()] = str(int(value)) - elif isinstance(value, (list, dict)): - env[key.upper()] = json.dumps(value) - elif isinstance(value, (IRCObject.Object,)): - env[key.upper()] = str(value) - - proc = subprocess.Popen([filename], env=env, - stdout=subprocess.PIPE, stderr=subprocess.STDOUT) - try: - proc.wait(5) - except subprocess.TimeoutExpired as e: - proc.kill() - return - - out = proc.stdout.read().decode("utf8").strip("\n") - if out: - if proc.returncode == 0: - if "stdout" in event: - event["stdout"].set_prefix(name) - event["stdout"].write(out) - else: - if "stderr" in event: - event["stderr"].set_prefix(name) - event["stderr"].write(out) |
