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

Complete function clean

Browse Source
This commit is contained in:
kxrd 2017-06-13 22:03:21 +02:00
parent d3efb942d5
commit 54be2d1c24
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pipreqs/pipreqs.py
View File

@ -19,6 +19,7 @@ Options:
--print Output the list of requirements in the standard output --print Output the list of requirements in the standard output
--force Overwrite existing requirements.txt --force Overwrite existing requirements.txt
--diff <file> Compare modules in requirements.txt to project imports. --diff <file> Compare modules in requirements.txt to project imports.
--clean <file> Clean up requirements.txt by removing modules that are not imported in project.
""" """
from __future__ import print_function, absolute_import from __future__ import print_function, absolute_import
import os import os
@ -298,8 +299,22 @@ def diff(file_, imports):
def clean(file_, imports): def clean(file_, imports):
"""Remove modules that aren't imported in project from file.""" """Remove modules that aren't imported in project from file."""
modules_not_imported = compare_modules(file_, imports)
re_remove = re.compile("|".join(modules_not_imported))
to_write = []
with open(file_, "r+") as f:
for i in f.readlines():
if re_remove.match(i) is None:
to_write.append(i)
f.seek(0)
f.truncate()
for i in to_write:
f.write(i)
logging.info("Successfully cleaned up requirements in " + file_)
def init(args): def init(args):
encoding = args.get('--encoding') encoding = args.get('--encoding')
@ -342,6 +357,10 @@ def init(args):
diff(args["--diff"], imports) diff(args["--diff"], imports)
return return
if args["--clean"]:
clean(args["--clean"], imports)
return
if not args["--print"] and not args["--savepath"] and not args["--force"] and os.path.exists(path): if not args["--print"] and not args["--savepath"] and not args["--force"] and os.path.exists(path):
logging.warning("Requirements.txt already exists, " logging.warning("Requirements.txt already exists, "
"use --force to overwrite it") "use --force to overwrite it")