fix(pipreqs): correct regex for docstring

This commit is contained in:
Vadim Kravcenko 2016-01-20 09:36:58 +01:00
parent 5a01ec01e9
commit d9434a5ac9

View File

@ -57,7 +57,7 @@ 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 = re.sub(re.compile("'''.+?'''", re.DOTALL), '', f.read())
contents = re.sub(re.compile('""".+"""', re.DOTALL), "", contents)
contents = re.sub(re.compile('""".+?"""', re.DOTALL), "", contents)
lines = contents.split("\n")
lines = filter(
filter_line, map(lambda l: l.partition("#")[0].strip(), lines))