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

Strip alias from import name for multiple imports on the same line. Fix for #13

Browse Source
This commit is contained in:
Tiago Costa 2015-05-08 16:54:21 +01:00
parent a0c796bc6b
commit b30777fb41
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pipreqs/pipreqs.py
View File

@ -27,7 +27,6 @@ REGEXP = [
re.compile(r'^from ((?!\.+).*?) import (?:.*)$')
]
def get_all_imports(start_path):
imports = []
packages = []
@ -52,11 +51,9 @@ def get_all_imports(start_path):
for item in s.groups():
if "," in item:
for match in item.split(","):
imports.append(match.strip())
imports.append(get_import_name_without_alias(match))
else:
to_append = item.partition(
' as ')[0].partition('.')[0]
imports.append(to_append.strip())
imports.append(get_import_name_without_alias(item))
third_party_packages = set(imports) - set(set(packages) & set(imports))
logging.debug(
'Found third-party packages: {0}'.format(third_party_packages))
@ -128,6 +125,8 @@ def get_pkg_names_from_import_names(pkgs):
result.append(toappend)
return result
def get_import_name_without_alias(import_name):
return import_name.partition(' as ')[0].partition('.')[0].strip()
def init(args):
print("Looking for imports")