diff options
Diffstat (limited to 'brain/py2stdlib.py')
| -rw-r--r-- | brain/py2stdlib.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/brain/py2stdlib.py b/brain/py2stdlib.py index 25c7122d..b3da2a53 100644 --- a/brain/py2stdlib.py +++ b/brain/py2stdlib.py @@ -5,11 +5,14 @@ Currently help understanding of : * hashlib.md5 and hashlib.sha1 """ -from astroid import MANAGER +from astroid import MANAGER, nodes from astroid.builder import AstroidBuilder MODULE_TRANSFORMS = {} + +# module specific transformation functions ##################################### + def hashlib_transform(module): fake = AstroidBuilder(MANAGER).string_build(''' @@ -177,6 +180,6 @@ def transform(module): tr(module) from astroid import MANAGER -MANAGER.register_transformer(transform) +MANAGER.register_transform(nodes.Module, transform) |