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

Rename function clean to compare_modules, have it return a tuple of modules not imported, add docstring

Browse Source
This commit is contained in:
kxrd 2017-06-10 20:43:54 +02:00
parent 1055a7a28b
commit d8b497ea91
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pipreqs/pipreqs.py
View File

@ -253,15 +253,26 @@ def parse_requirements(file_):
return (modules, parameters) return (modules, parameters)
def clean(file_, imports):
def compare_modules(file_, imports):
"""Compare modules in a file to imported modules in a project.
Args:
file_ (str): File to parse for modules to be compared.
imports (tuple): Modules being imported in the project.
Returns:
tuple: The modules not imported in the project, but do exist in the
specified file.
"""
modules, parameters = parse_requirements(file_) modules, parameters = parse_requirements(file_)
imports = [imports[i]["name"] for i in range(len(imports))] imports = [imports[i]["name"] for i in range(len(imports))]
modules = [modules[i]["name"] for i in range(len(modules))] modules = [modules[i]["name"] for i in range(len(modules))]
modules_not_imported = set(modules) - set(imports) modules_not_imported = set(modules) - set(imports)
print("These modules are in {} but does not seem to be imported: " return modules_not_imported
"\n{}".format(file_, "\n".join(x for x in modules_not_imported)))
def init(args): def init(args):
encoding = args.get('--encoding') encoding = args.get('--encoding')