# HG changeset patch
# User Sylvain Thénault <sylvain.thenault@logilab.fr>
# Date 1318439182 -7200
# Node ID 018b5deca73ee708ea7a98eda07c1f3fbd581587
# Parent  e107204d3b27619b9f46f84089a2bca31df622d6
pylint extension

diff -r e107204d3b27 -r 018b5deca73e pylintext.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/pylintext.py	Wed Oct 12 19:06:22 2011 +0200
@@ -0,0 +1,41 @@
+"""https://pastebin.logilab.fr/show/860/"""
+
+from logilab.astng import MANAGER, nodes, scoped_nodes
+
+def turn_function_to_class(node):
+    """turn a Function node into a Class node (in-place)"""
+    node.__class__ = scoped_nodes.Class
+    node.bases = ()
+    # remove return nodes so that we don't get warned about 'return outside
+    # function' by pylint
+    for rnode in node.nodes_of_class(nodes.Return):
+        rnode.parent.body.remove(rnode)
+    # that seems to be enough :)
+
+
+def cubicweb_transform(module):
+    # handle objectify_selector decorator. Only look at module level functions,
+    # should be enough
+    for assnodes in module.locals.values():
+        for node in assnodes:
+            if isinstance(node, scoped_nodes.Function) and node.decorators:
+                for decorator in node.decorators.nodes:
+                    for infered in decorator.infer():
+                        if infered.name == 'objectify_selector':
+                            turn_function_to_class(node)
+                            break
+                    else:
+                        continue
+                    break
+    # add yams base types into 'yams.buildobjs', astng doesn't grasp globals()
+    # magic in there
+    if module.name == 'yams.buildobjs':
+        from yams import BASE_TYPES
+        for etype in BASE_TYPES:
+            module.locals[etype] = [scoped_nodes.Class(etype, None)]
+
+MANAGER.register_transformer(cubicweb_transform)
+
+def register(linter):
+    """called when loaded by pylint --load-plugins, nothing to do here"""
+