Refactoring for clarity (suggested by Mustapha Benali).

zopefoundation · Nov 28, 2007 · 15d5327 · 15d5327
1 parent 3defe5c
commit 15d5327
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/z3c/jbot/__init__.py b/z3c/jbot/__init__.py
@@ -2,9 +2,10 @@
 from zope.pagetemplate.pagetemplatefile import PageTemplate
 
 # add registration hook to ``zope.app.pagetemplate``
-pt_render = PageTemplate.pt_render
-def patched_pt_render(self, *args, **kwargs):
-    manager.getGlobalTemplateManager().registerTemplate(self)
-    return pt_render(self, *args, **kwargs)
+def jbot(pt_render):
+    def render(self, *args, **kwargs): 
+        manager.getGlobalTemplateManager().registerTemplate(self)
+        return pt_render(self, *args, **kwargs)        
+    return render
 
-PageTemplate.pt_render = patched_pt_render
+PageTemplate.pt_render = jbot(PageTemplate.pt_render)