X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/blobdiff_plain/9aa8aa5122d0208fd4177f436e9b60ab868e204c..a1b88a4611f98012432130e7d9ff6c5ed32b9eae:/.config/taskwarrior/taskpirate/on-modify-pirate

diff --git a/.config/taskwarrior/taskpirate/on-modify-pirate b/.config/taskwarrior/taskpirate/on-modify-pirate
new file mode 100755
index 0000000..b7b4986
--- /dev/null
+++ b/.config/taskwarrior/taskpirate/on-modify-pirate
@@ -0,0 +1,43 @@
+#!/usr/bin/env python
+
+import glob
+import imp
+import os
+
+from tasklib import TaskWarrior, Task
+
+
+def find_hooks(file_prefix):
+    # Find all files in subdirectories whose names start with <file_prefix>
+    file_pattern = os.path.dirname(__file__) + '/*/' + file_prefix + "*.py"
+    module_paths = [f for f in glob.glob(file_pattern) if os.path.isfile(f)]
+    module_paths.sort()
+
+    # Gather all hooks in these files
+    hooks = []
+
+    for module_path in module_paths:
+        # Load the module
+        module_dir = os.path.dirname(module_path)
+        module_filename = os.path.basename(module_path)
+        module_name = 'pirate_{0}_{1}'.format(module_dir, module_filename)
+        module_name = module_name.replace('.', '_')
+        module = imp.load_source(module_name, module_path)
+
+        # Find all hook methods available
+        module_hooks = [
+            getattr(module, hook_name)
+            for hook_name in dir(module)
+            if hook_name.startswith('hook_')
+        ]
+
+        hooks += module_hooks
+
+    return hooks
+
+task = Task.from_input()
+
+for hook in find_hooks('pirate_mod'):
+    hook(task)
+
+print(task.export_data())