Fix import statements

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

diff --git a/on-modify-pirate b/on-modify-pirate
index 8333d3d04d5679aa64c3813e449a7889881dbfc2..5093d18edda3d7bd9d6989bca9a772cdf631ce89 100755 (executable)
--- a/on-modify-pirate
+++ b/on-modify-pirate
@@ -1,10 +1,10 @@
-#!/usr/bin/python
+#!/usr/bin/env python2

 
 import glob
 import imp
 import os
 
 
 import glob
 import imp
 import os
 

-from tasklib.task import TaskWarrior, Task
+from tasklib import TaskWarrior, Task

 
 
 def find_hooks(file_prefix):
 
 
 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_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 = imp.load_source(module_name, module_path)
 
         # Find all hook methods available