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

Implement --no-pin option

Browse Source
This commit is contained in:
dxsh 2017-10-23 16:32:02 +02:00
parent 4d35819bbd
commit 7e819c91aa
1 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pipreqs/pipreqs.py
View File

@ -21,6 +21,7 @@ Options:
--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. --clean <file> Clean up requirements.txt by removing modules that are not imported in project.
--no-pin Omit package version from requirements file.
""" """
from __future__ import print_function, absolute_import from __future__ import print_function, absolute_import
import os import os
@ -115,16 +116,22 @@ def filter_line(l):
return len(l) > 0 and l[0] != "#" return len(l) > 0 and l[0] != "#"
def generate_requirements_file(path, imports): def generate_requirements_file(path, imports, omit_version=None):
with open(path, "w") as out_file: with open(path, "w") as out_file:
logging.debug('Writing {num} requirements: {imports} to {file}'.format( logging.debug('Writing {num} requirements: {imports} to {file}'.format(
num=len(imports), num=len(imports),
file=path, file=path,
imports=", ".join([x['name'] for x in imports]) imports=", ".join([x['name'] for x in imports])
)) ))
fmt = '{name}=={version}'
out_file.write('\n'.join(fmt.format(**item) if item['version'] else '{name}'.format(**item) if omit_version is True:
for item in imports) + '\n') fmt = '{name}\n'
out_file.write(''.join(fmt.format(**item) for item in imports))
else:
fmt = '{name}=={version}\n'
out_file.write(''.join(fmt.format(**item) if item['version']
else '{name}'.format(**item)
for item in imports))
def output_requirements(imports): def output_requirements(imports):
logging.debug('Writing {num} requirements: {imports} to stdout'.format( logging.debug('Writing {num} requirements: {imports} to stdout'.format(
@ -383,11 +390,16 @@ def init(args):
"use --force to overwrite it") "use --force to overwrite it")
return return
if args["--no-pin"]:
omit_version = True
else:
omit_version = False
if args["--print"]: if args["--print"]:
output_requirements(imports) output_requirements(imports)
logging.info("Successfully output requirements") logging.info("Successfully output requirements")
else: else:
generate_requirements_file(path, imports) generate_requirements_file(path, imports, omit_version)
logging.info("Successfully saved requirements file in " + path) logging.info("Successfully saved requirements file in " + path)