github/pipreqs
1
1
Fork 0
mirror of https://github.com/bndr/pipreqs.git synced 2025-06-06 03:25:21 +00:00
Code Issues 3 Packages Projects Releases Wiki Activity

fix(pipreqs.py): remove ''' and """ comments before going through imports

Browse Source
This commit is contained in:
Vadim Kravcenko 2015-12-04 20:40:38 +01:00
parent a4a40efe26
commit 1ee081454b
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pipreqs/pipreqs.py
View File

@ -53,8 +53,11 @@ def get_all_imports(path, encoding=None):
candidates += [os.path.splitext(fn)[0] for fn in files]
for file_name in files:
with open_func(os.path.join(root, file_name), "r", encoding=encoding) as f:
contents = re.sub(re.compile("'''.+?'''", re.DOTALL), '', f.read())
contents = re.sub(re.compile('""".+"""', re.DOTALL), "", contents)
lines = contents.split("\n")
lines = filter(
filter_line, map(lambda l: l.partition("#")[0].strip(), f))
filter_line, map(lambda l: l.partition("#")[0].strip(), lines))
for line in lines:
if "(" in line:
break