diff options
| author | 2019-02-05 16:56:21 +0000 | |
|---|---|---|
| committer | 2019-02-05 16:56:21 +0000 | |
| commit | fd08e23bd46d209a0c8bae84c83ad901a4b9cc12 (patch) | |
| tree | 4e88c1bf0f35820f44abd6f5858fb1ae390b2c00 /modules/scripts.py | |
| parent | CAP capabilities would not be index `2` if the line is multiline (diff) | |
| signature | ||
Move scripts module to it's own directory
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) |
