make it work on current business template format

erp5flakes/__init__.py

 
-import compiler
+import _ast
 import sys
 import os
 from pyflakes.checker import Checker
@@ -29,17 +29,12 @@ except ImportError:
     def begin_transaction():
         get_transaction().begin()
 
-# by default, we ignore warnings about unused imports
-fatal_messages = ( messages.ImportStarUsed, messages.UndefinedName,
-           messages.DuplicateArgument, messages.RedefinedFunction )
+ignored_messages = (messages.ReturnOutsideFunction, )
 
 def check(codeString, filename, bound_names=()):
-    try:
     bound_names = [n for n in bound_names if not hasattr(__builtin__, n)]
-        for name in bound_names:
-            setattr(__builtin__, name, 1)
     try:
-            tree = compiler.parse(codeString)
+        tree = compile(codeString, filename, "exec", _ast.PyCF_ONLY_AST)
     except (SyntaxError, IndentationError):
         value = sys.exc_info()[1]
         try:
@@ -55,17 +50,13 @@ def check(codeString, filename, bound_names=()):
         return 1
     else:
         warnings = 0
-            w = Checker(tree, filename)
+        w = Checker(tree, filename, builtins=bound_names)
         w.messages.sort(lambda a, b: cmp(a.lineno, b.lineno))
         for warning in w.messages:
-                if isinstance(warning, fatal_messages):
+            if not isinstance(warning, ignored_messages):
                 warnings += 1
                 print warning
         return warnings
-    finally:
-        for name in bound_names:
-            if hasattr(__builtin__, name):
-                delattr(__builtin__, name)
 
 
 def checkZopeProduct(path):
@@ -118,7 +109,7 @@ def checkBusinessTemplate(path):
         global _connection
         if _connection is not None:
             return _connection
-        db = ZODB.DB(DemoStorage(quota=(1<<20)))
+        db = ZODB.DB(DemoStorage())
         _connection = db.open()
         begin_transaction()
         return _connection
@@ -131,8 +122,7 @@ def checkBusinessTemplate(path):
             if not filename.endswith('.xml'):
                 continue
             filename = os.path.join(dirpath, filename)
-            file_obj = file(filename)
-            try:
+            with open(filename) as file_obj:
                 if 'Products.PythonScripts.PythonScript' in file_obj.read():
                     file_obj.seek(0)
                     obj = _getConnection().importFile(file_obj,
@@ -140,10 +130,15 @@ def checkBusinessTemplate(path):
                     blacklist_names = []
                     if 'WorkflowTemplateItem' in dirpath:
                         blacklist_names = ['context']
-                    warnings += check(obj._body, filename,
+                    python_code = obj._body
+                    python_filename = filename[:-4] + '.py'
+                    if os.path.exists(python_filename):
+                        # "new" business template format, where python code is
+                        # in a separate .py file
+                        python_code = file(python_filename, "U").read()
+                        filename = python_filename
+                    warnings += check(python_code, filename,
                         getValidNames(obj, blacklist_names=blacklist_names))
-            finally:
-                file_obj.close()
 
     warnings += checkZopeProduct(os.path.join(path, 'TestTemplateItem'))
     warnings += checkZopeProduct(os.path.join(path, 'ExtensionTemplateItem'))
@@ -158,7 +153,7 @@ def isProduct(path):
     return os.path.exists(os.path.join(path, '__init__.py'))
 
 def isBusinessTemplate(path):
-    return os.path.exists(os.path.join(path, 'bt', 'revision'))
+    return os.path.exists(os.path.join(path, 'bt', 'title'))
 
 def main():
     warnings = 0