fix memory leak when excecute BEMRender.render() with entrypoint argu…

…ment
arhibot · Jan 18, 2013 · d19192d · d19192d
1 parent 1a5514b
commit d19192d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pybem/pybem.py b/pybem/pybem.py
@@ -124,10 +124,12 @@ def render(self, pagedir, context, env, entrypoint,
             ctx.locals.context = context
             ctx.locals.env = env
             if entrypoint:
-                ctx.locals.bemjson = ctx.eval('%s(context, env)' % entrypoint)
+                if return_bemjson:
+                    return ctx.eval('JSON.stringify(%s(context, env))' % entrypoint)
+                return ctx.eval('BEMHTML.apply(%s(context, env))' % entrypoint)
             else:
                 ctx.locals.bemjson = context
-            if return_bemjson:
-                return ctx.eval('JSON.stringify(bemjson)')
-            return ctx.eval('BEMHTML.apply(bemjson)')
+                if return_bemjson:
+                    return ctx.eval('JSON.stringify(bemjson)')
+                return ctx.eval('BEMHTML.apply(bemjson)')