mirror of
https://github.com/J3rome/py-requirements-guesser.git
synced 2024-11-24 19:00:10 +01:00
117 lines
3.5 KiB
Python
117 lines
3.5 KiB
Python
import re
|
|
import os
|
|
from urllib.request import urlretrieve
|
|
|
|
def user_response_multi_choices(message, choices):
|
|
print(message)
|
|
for i, choice in enumerate(choices):
|
|
print(f' {i+1}. {choice}')
|
|
|
|
|
|
nb_choices = len(choices)
|
|
resp = input(f'Choose option [1-{nb_choices}]\n')
|
|
|
|
if not resp.isdigit() or int(resp) not in range(1,nb_choices+1):
|
|
print("")
|
|
return user_response_multi_choices(message, choices)
|
|
|
|
return int(resp)
|
|
|
|
|
|
def user_response_yes_no(message):
|
|
|
|
|
|
resp = input(message + ' [Y/n]\n').lower()
|
|
|
|
if resp not in ['y', 'n']:
|
|
print("")
|
|
return user_response_yes_no(message)
|
|
|
|
return resp == 'y'
|
|
|
|
|
|
def get_mapping_files_from_pipreqs(tmp_path="/tmp/.py-req-guesser"):
|
|
"""
|
|
Retrieve import to package name mapping file and standard lib module list
|
|
This list comes from https://github.com/bndr/pipreqs
|
|
"""
|
|
|
|
skip_download = False
|
|
|
|
if not os.path.exists(tmp_path):
|
|
os.mkdir(tmp_path)
|
|
|
|
mapping_filepath = f"{tmp_path}/mapping"
|
|
stdlib_filepath = f"{tmp_path}/stdlib"
|
|
|
|
if os.path.exists(mapping_filepath) and os.path.exists(stdlib_filepath):
|
|
# File have already been downloaded
|
|
skip_download = True
|
|
|
|
if not skip_download:
|
|
msg = "We will download a mapping file from https://github.com/bndr/pipreqs\n" \
|
|
"Thanks to the maintainers of Pipreqs for keeping the mapping file "\
|
|
"and the STDlib module list up to date\n" \
|
|
f"Do you agree to downloading these files in '{tmp_path}' ?"
|
|
|
|
if not user_response_yes_no(msg):
|
|
print("\n\n[ERROR]Pipreqs mapping files are required, I encourage you to inspect the code to make sure everything is safe and rerun this")
|
|
exit(0)
|
|
|
|
print("")
|
|
# FIXME : This is not really scalable...
|
|
mapping_url = "https://raw.githubusercontent.com/bndr/pipreqs/90102acdbb23c09574d27df8bd1f568d34e0cfd3/pipreqs/mapping"
|
|
stdlib_url = "https://raw.githubusercontent.com/bndr/pipreqs/90102acdbb23c09574d27df8bd1f568d34e0cfd3/pipreqs/stdlib"
|
|
|
|
urlretrieve(mapping_url, mapping_filepath)
|
|
urlretrieve(stdlib_url, stdlib_filepath)
|
|
|
|
from_import_to_package_mapping = {}
|
|
from_package_to_import_mapping = {}
|
|
with open(mapping_filepath, 'r') as f:
|
|
for line in f.readlines():
|
|
import_name, package_name = line.strip().split(":")
|
|
|
|
from_import_to_package_mapping[import_name] = package_name
|
|
from_package_to_import_mapping[package_name] = import_name
|
|
|
|
with open(stdlib_filepath, 'r') as f:
|
|
stdlib = set([l.strip() for l in f.readlines()])
|
|
|
|
return stdlib, from_import_to_package_mapping, from_package_to_import_mapping
|
|
|
|
|
|
|
|
def load_packages_from_requirements(filepath):
|
|
# TODO : Handle when multiple version conditions
|
|
# TODO : Handle greater than (>). If version contains >, should take the greatest available version at the date. Should fit with minor versions ?
|
|
with open(filepath, 'r') as f:
|
|
lines = f.readlines()
|
|
|
|
split_reg = re.compile(r'==|<=|>=|<|>')
|
|
|
|
packages = {}
|
|
|
|
for line in lines:
|
|
splitted = re.split(split_reg, line.strip())
|
|
if len(splitted) > 1:
|
|
version = splitted[-1]
|
|
else:
|
|
version = None
|
|
|
|
packages[splitted[0]] = version
|
|
|
|
return packages
|
|
|
|
|
|
def get_python_filename_at_root():
|
|
return [f[:-3] for f in os.listdir('.') if os.path.isfile(f) and f.endswith('.py')]
|
|
|
|
|
|
def detect_os():
|
|
pass
|
|
|
|
|
|
def get_python_version():
|
|
pass
|