/brz/remove-bazaar : revision 7144

To get this branch, use:

bzr branch
http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

Viewing changes to tools/brzflakes.py

Add a tool that monkey patches pyflakes to support lazy imports.

Merged from https://code.launchpad.net/~jelmer/brz/flake8/+merge/358593

added added

removed removed

#!/usr/bin/python3

from __future__ import absolute_import

import ast

from pyflakes.checker import Checker, ImportationFrom

# Do some monkey patching..

def CALL(self, node):

if isinstance(node.func, ast.Name) and node.func.id == 'lazy_import':

self.LAZY_IMPORT(node)

elif (isinstance(node.func, ast.Attribute) and

node.func.attr == 'lazy_import' and

node.func.value.id == 'lazy_import'):

self.LAZY_IMPORT(node)

return self.handleChildren(node)

Checker.CALL = CALL

def LAZY_IMPORT(self, node):

from breezy.lazy_import import ImportProcessor

processor = ImportProcessor()

if not isinstance(node.args[1], ast.Str):

# Not sure how to deal with this..

return

import_text = node.args[1].s

scope = {}

processor.lazy_import(scope, import_text)

for name, (path, sub, scope) in processor.imports.items():

importation = ImportationFrom(name, node, '.'.join(path), scope)

self.addBinding(node, importation)

Checker.LAZY_IMPORT = LAZY_IMPORT

if __name__ == '__main__':

from pyflakes import __main__

__main__.main()