root/software/apt-tools/get_deps.py @ 4e515eac
| a0c50279 | david.sorber | #!/usr/bin/python3
|
||
| 982d29ac | david.sorber | import argparse
|
||
| a0c50279 | david.sorber | import copy
|
||
from distutils.version import LooseVersion
|
||||
import sys
|
||||
import apt
|
||||
| 982d29ac | david.sorber | # Virtual package mapping
|
||
| 31c4557a | David Sorber | virtual_mapping = {'awk': 'gawk',
|
||
'perlapi-5.18.1': 'perl-base',
|
||||
'perlapi-5.18.2': 'perl-base',
|
||||
'upstart-job': 'upstart',
|
||||
'python:any': 'python2.7',
|
||||
'python3:any': 'python3.4',
|
||||
'whiptail-provider': 'whiptail'}
|
||||
| a0c50279 | david.sorber | |||
def main():
|
||||
| 982d29ac | david.sorber | parser = argparse.ArgumentParser(description='Process some integers.')
|
||
| 31c4557a | David Sorber | parser.add_argument('-v', '--verbose', action='store_true',
|
||
help='write additional information to stderr')
|
||||
parser.add_argument('input_package_files', action='store',
|
||||
help='a comma separated list of files containing '
|
||||
'package names, one per line')
|
||||
| 982d29ac | david.sorber | args = parser.parse_args()
|
||
# Input packages
|
||||
in_packages = []
|
||||
# Parse input package files
|
||||
for filename in args.input_package_files.split(','):
|
||||
with open(filename) as in_file:
|
||||
for line in in_file:
|
||||
line = line.strip()
|
||||
if line:
|
||||
in_packages.append(line)
|
||||
# Open APT cache
|
||||
cache = apt.Cache(memonly=True)
|
||||
cache.open()
|
||||
| a0c50279 | david.sorber | |||
| 31c4557a | David Sorber | # Create a set to hold the names of all packages and their dependencies
|
||
| a0c50279 | david.sorber | packages = set()
|
||
while True:
|
||||
try:
|
||||
# Grab next package
|
||||
| 31c4557a | David Sorber | pkg_name = in_packages.pop()
|
||
if args.verbose:
|
||||
sys.stderr.write('Package: {:s}\n'.format(pkg_name))
|
||||
sys.stderr.flush()
|
||||
| 982d29ac | david.sorber | |||
| 31c4557a | David Sorber | # Check if the package is in the cache, if not check if it is in
|
||
# our list of known virtual packages
|
||||
| 982d29ac | david.sorber | if pkg_name not in cache:
|
||
if pkg_name in virtual_mapping:
|
||||
pkg_name = virtual_mapping[pkg_name]
|
||||
else:
|
||||
| 31c4557a | David Sorber | sys.stderr.write('ERROR: unkown package "{:s}"; if this is '
|
||
'a virtual package please add it to the '
|
||||
'virtual package mapping\n'.format(pkg_name))
|
||||
| 982d29ac | david.sorber | continue
|
||
| a0c50279 | david.sorber | # Add this package (with version) to the list of packages
|
||
| 31c4557a | David Sorber | packages.add(pkg_name)
|
||
| 982d29ac | david.sorber | |||
# Get package dependencies
|
||||
for pdep in cache[pkg_name].versions[0].dependencies:
|
||||
| a0c50279 | david.sorber | dep_name = pdep[0].name
|
||
| 31c4557a | David Sorber | if args.verbose:
|
||
sys.stderr.write('\tDependency: {:s}\n'.format(dep_name))
|
||||
sys.stderr.flush()
|
||||
| a0c50279 | david.sorber | if dep_name not in packages:
|
||
| 982d29ac | david.sorber | if dep_name in virtual_mapping:
|
||
dep_name = virtual_mapping[dep_name]
|
||||
| a0c50279 | david.sorber | packages.add(dep_name)
|
||
in_packages.insert(0, dep_name)
|
||||
except IndexError:
|
||||
break
|
||||
| 982d29ac | david.sorber | # Print out the gathered package names
|
||
for package in sorted(packages):
|
||||
| a0c50279 | david.sorber | print(package)
|
||
if __name__ == '__main__':
|
||||
sys.exit(main())
|