Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[DO NOT MERGE] Tooling for auditing dependencies #2054

Draft
wants to merge 2 commits into
base: master
Choose a base branch
from
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -36,4 +36,7 @@ commands.log
.vscode/launch.json
.vscode/settings.json

pip_list.txt
sorted_pip_list.txt

.pytest_cache/
111 changes: 111 additions & 0 deletions tools/report_deps_used.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,111 @@
# !python

import argparse
from configure_logging import configure_logging
import logging
import os
import pprint
from subprocess import check_call

pp = pprint.PrettyPrinter(indent=4, width=100)

default_python_cmd = ['python']
ni_owned_modules = [
'hightime',
'nidcpower',
'nidigital',
'nidmm',
'nifake',
'nifgen',
'nimodinst',
'niscope',
'nise',
'niswitch',
'nitclk',
]


class CustomFormatter(argparse.ArgumentDefaultsHelpFormatter, argparse.RawDescriptionHelpFormatter):
'''We want the description to use the raw formatting but have the parameters be formatted as before

from stackoverflow:
https://stackoverflow.com/questions/18462610/argumentparser-epilog-and-description-formatting-in-conjunction-with-argumentdef
'''
pass


def main():
# Setup the required arguments for this script
usage = """Log and process list of python deps used in the project"""

parser = argparse.ArgumentParser(description=usage, formatter_class=CustomFormatter)

report_group = parser.add_argument_group("Reporting Configuration")
report_group.add_argument("--clean", action="store_true", default=False, help="Clean the dep list file")
report_group.add_argument("--log", action="store_true", default=False, help="Log the deps installed to the tox env.")
report_group.add_argument("--env_name", action="store", default=None, help="The name of the tox env used")
report_group.add_argument("--report", action="store_true", default=False, help="Report all unique deps and versions used.")
report_group.add_argument("--python-cmd", action="store", default=None, help=f"Command to use for invoking python. Default: {default_python_cmd}")
report_group.add_argument("--dir", action="store", default=None, help="Working directory to change to before running commands")

verbosity_group = parser.add_argument_group("Verbosity, Logging & Debugging")
verbosity_group.add_argument("-v", "--verbose", action="count", default=0, help="Verbose output")
verbosity_group.add_argument("--preview", action="store_true", default=False, help="Show what would happen when running with given parameters")
verbosity_group.add_argument("--log-file", action="store", default=None, help="Send logging to listed file instead of stdout")
args = parser.parse_args()

if args.verbose > 1:
configure_logging(logging.DEBUG, args.log_file)
elif args.verbose == 1:
configure_logging(logging.INFO, args.log_file)
else:
configure_logging(logging.WARNING, args.log_file)

logging.info(pp.pformat(args))

python_cmd = [args.python_cmd] if args.python_cmd is not None else default_python_cmd
dep_list_file = 'pip_list.txt'

passthrough_params = ['-v' for i in range(args.verbose)]
if args.log_file:
passthrough_params.append('--log-file').append(args.log_file)

if args.dir:
os.chdir(args.dir)

if args.clean:
logging.info('Cleaning Logged Deps')
with open(dep_list_file, 'w') as f:
f.write('')

if args.log:
logging.info('Logging Deps')
env_label_cmd = ['echo', args.env_name]
pip_list_cmd = python_cmd + ['-m', 'pip', 'list', '--format=freeze']
logging.info(pp.pformat(env_label_cmd))
logging.info(pp.pformat(pip_list_cmd))
with open(dep_list_file, 'a') as f:
check_call(env_label_cmd, stdout=f)
check_call(pip_list_cmd, stdout=f)

if args.report:
logging.info('Reporting Deps')
with open(dep_list_file, 'r') as f:
deps = f.readlines()

deps = [dep.strip() for dep in deps if "==" in dep]
deps = list(set(deps))

with open(f'sorted_{dep_list_file}', 'w') as f:
for dep in sorted(deps):
if dep.split('==')[0] in ni_owned_modules:
continue
logging.info(dep)
f.write(f'{dep}\n')




if __name__ == '__main__':
main()

9 changes: 9 additions & 0 deletions tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -128,6 +128,15 @@ commands =
pkg: python -m twine check generated/nise/dist/*
pkg: python -m twine check generated/niswitch/dist/*
# pkg: check-manifest --ignore tox.ini,tests*,.github,.github/*,CONTRIBUTING.rst,docs,docs/*
# Document the list of installed packages for our release process
clean: python tools/report_deps_used.py --clean
buid_test_env: python tools/report_deps_used.py --log --env_name {envname}
codegen: python tools/report_deps_used.py --log --env_name {envname}
installers: python tools/report_deps_used.py --log --env_name {envname}
test: python tools/report_deps_used.py --log --env_name {envname}
flake8: python tools/report_deps_used.py --log --env_name {envname}
docs: python ../tools/report_deps_used.py --log --env_name {envname} --dir ..
pkg: python tools/report_deps_used.py --log --env_name {envname}

deps =
test: pytest
Expand Down
Loading