Python 3 fix for func_name

neo4j-contrib · Jun 17, 2013 · 9218afd · 9218afd
1 parent c5de7f6
commit 9218afd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/neomodel/signals.py b/neomodel/signals.py
@@ -20,8 +20,9 @@ def exec_hook(hook_name, self, *args, **kwargs):
 
 def hooks(fn):
     def hooked(self, *args, **kwargs):
-        exec_hook('pre_' + fn.func_name, self, *args, **kwargs)
+        fn_name = fn.func_name if hasattr(fn, 'func_name') else fn.__name__
+        exec_hook('pre_' + fn_name, self, *args, **kwargs)
         val = fn(self, *args, **kwargs)
-        exec_hook('post_' + fn.func_name, self, *args, **kwargs)
+        exec_hook('post_' + fn_name, self, *args, **kwargs)
         return val
     return hooked