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

Add and implement function diff, improve inline comments in function parse_requirements

Browse Source
This commit is contained in:
kxrd 2017-06-10 21:24:39 +02:00
parent a78203dc2b
commit 7549b1c416
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pipreqs/pipreqs.py
View File

@ -18,6 +18,7 @@ Options:
--savepath <file> Save the list of requirements in the given file --savepath <file> Save the list of requirements in the given file
--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.
""" """
from __future__ import print_function, absolute_import from __future__ import print_function, absolute_import
import os import os
@ -289,6 +290,14 @@ def compare_modules(file_, imports):
return modules_not_imported return modules_not_imported
def diff(file_, imports):
"""Display the difference between modules in file a and imported modules."""
modules_not_imported = compare_modules(file_, imports)
logging.info("The following modules are in {} but do not seem to be imported: "
"{}".format(file_, ", ".join(x for x in modules_not_imported)))
def init(args): def init(args):
encoding = args.get('--encoding') encoding = args.get('--encoding')
extra_ignore_dirs = args.get('--ignore') extra_ignore_dirs = args.get('--ignore')
@ -326,6 +335,10 @@ def init(args):
path = (args["--savepath"] if args["--savepath"] else path = (args["--savepath"] if args["--savepath"] else
os.path.join(args['<path>'], "requirements.txt")) os.path.join(args['<path>'], "requirements.txt"))
if args["--diff"]:
diff(args["--diff"], 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")