X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/blobdiff_plain/d0e6afe33606ebb3084b8259c0949db7735167aa..83bd151be4cb424fc444ae60985d79bbe2a3567d:/on-modify-pirate?ds=sidebyside

diff --git a/on-modify-pirate b/on-modify-pirate
index 8333d3d..61bb8f6 100755
--- a/on-modify-pirate
+++ b/on-modify-pirate
@@ -4,7 +4,7 @@ import glob
 import imp
 import os
 
-from tasklib.task import TaskWarrior, Task
+from tasklib import TaskWarrior, Task
 
 
 def find_hooks(file_prefix):
@@ -21,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
@@ -39,4 +40,4 @@ task = Task.from_input()
 for hook in find_hooks('pirate_mod'):
     hook(task)
 
-print task.export_data()
+print(task.export_data())