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

Add Code Changes

Browse Source
This commit is contained in:
Devashish Gaikwad 2021-05-13 23:24:46 +05:30
parent 564842c29e
commit be812e4884
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pipreqs/pipreqs.py
View File

@ -47,6 +47,8 @@ from docopt import docopt
import requests import requests
from yarg import json2package from yarg import json2package
from yarg.exceptions import HTTPError from yarg.exceptions import HTTPError
from urllib3 import disable_warnings
from urllib3.exceptions import InsecureRequestWarning
from pipreqs import __version__ from pipreqs import __version__
@ -180,13 +182,13 @@ def output_requirements(imports):
def get_imports_info( def get_imports_info(
imports, pypi_server="https://pypi.python.org/pypi/", proxy=None): imports, pypi_server="https://pypi.python.org/pypi/", proxy=None, verify_ssl=True):
result = [] result = []
for item in imports: for item in imports:
try: try:
response = requests.get( response = requests.get(
"{0}{1}/json".format(pypi_server, item), proxies=proxy) "{0}{1}/json".format(pypi_server, item), proxies=proxy, verify=verify_ssl)
if response.status_code == 200: if response.status_code == 200:
if hasattr(response.content, 'decode'): if hasattr(response.content, 'decode'):
data = json2package(response.content.decode()) data = json2package(response.content.decode())
@ -397,6 +399,7 @@ def init(args):
encoding = args.get('--encoding') encoding = args.get('--encoding')
extra_ignore_dirs = args.get('--ignore') extra_ignore_dirs = args.get('--ignore')
follow_links = not args.get('--no-follow-links') follow_links = not args.get('--no-follow-links')
verify_ssl = not args.get('--trusted-host')
input_path = args['<path>'] input_path = args['<path>']
if input_path is None: if input_path is None:
input_path = os.path.abspath(os.curdir) input_path = os.path.abspath(os.curdir)
@ -404,6 +407,9 @@ def init(args):
if extra_ignore_dirs: if extra_ignore_dirs:
extra_ignore_dirs = extra_ignore_dirs.split(',') extra_ignore_dirs = extra_ignore_dirs.split(',')
if not verify_ssl:
disable_warnings(InsecureRequestWarning)
candidates = get_all_imports(input_path, candidates = get_all_imports(input_path,
encoding=encoding, encoding=encoding,
extra_ignore_dirs=extra_ignore_dirs, extra_ignore_dirs=extra_ignore_dirs,
@ -430,7 +436,8 @@ def init(args):
if x.lower() not in [z['name'].lower() for z in local]] if x.lower() not in [z['name'].lower() for z in local]]
imports = local + get_imports_info(difference, imports = local + get_imports_info(difference,
proxy=proxy, proxy=proxy,
pypi_server=pypi_server) pypi_server=pypi_server,
verify_ssl=verify_ssl)
path = (args["--savepath"] if args["--savepath"] else path = (args["--savepath"] if args["--savepath"] else
os.path.join(input_path, "requirements.txt")) os.path.join(input_path, "requirements.txt"))