From 415ede7691519252eda87d33481adea19bf3d7ba Mon Sep 17 00:00:00 2001 From: Kay Sackey Date: Thu, 28 Jan 2016 01:08:38 -0500 Subject: [PATCH] Cleanup Moved the import statement to the top of the page where it belongs --- pipreqs/pipreqs.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pipreqs/pipreqs.py b/pipreqs/pipreqs.py index b2d63e2..e20939c 100755 --- a/pipreqs/pipreqs.py +++ b/pipreqs/pipreqs.py @@ -23,7 +23,7 @@ import sys import re import logging import codecs - +import ast, traceback from docopt import docopt import requests from yarg import json2package @@ -42,8 +42,6 @@ else: open_func = codecs.open -import ast, traceback - def get_all_imports(path, encoding=None): imports = set() raw_imports = set() @@ -60,6 +58,9 @@ def get_all_imports(path, encoding=None): for file_name in files: with open_func(os.path.join(root, file_name), "r", encoding=encoding) as f: contents = f.read() + #contents = re.sub(re.compile("'''.+?'''", re.DOTALL), '', f.read()) + #contents = re.sub(re.compile('""".+?"""', re.DOTALL), "", contents) + try: tree = ast.parse(contents) except Exception, e: