X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/blobdiff_plain/34b3fe8edd2a93a6e7d97c5096fdc364ad580b14..c0d91dae27c2a1f73ed9fc103f7fbfcfc3ca0ae9:/on-modify-pirate?ds=inline diff --git a/on-modify-pirate b/on-modify-pirate index 018b2ef..b7b4986 100755 --- a/on-modify-pirate +++ b/on-modify-pirate @@ -1,16 +1,17 @@ -#!/usr/bin/python +#!/usr/bin/env python import glob import imp import os -from tasklib.task import TaskWarrior, Task +from tasklib import TaskWarrior, Task def find_hooks(file_prefix): # Find all files in subdirectories whose names start with 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 = [] @@ -20,6 +21,7 @@ def find_hooks(file_prefix): 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 @@ -38,4 +40,4 @@ task = Task.from_input() for hook in find_hooks('pirate_mod'): hook(task) -print task.export_data() +print(task.export_data())