Use python in current environment instead of /usr/bin/python

[etc/taskwarrior.git] / on-add-pirate

diff --git a/on-add-pirate b/on-add-pirate
index 8d123ed0d9f7b1cf0cf8d5ef0965d4638de9f11a..9fae1c46333fa90cf1350148fdbc97a5b69cb2ba 100755 (executable)
--- a/on-add-pirate
+++ b/on-add-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_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 = []
@@ -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_add'):
     hook(task)
 
-print task.export_data()
+print(task.export_data())