|
#!/usr/bin/python3
|
|
import argparse
|
|
import copy
|
|
from distutils.version import LooseVersion
|
|
import sys
|
|
|
|
import apt
|
|
|
|
# Virtual package mapping
|
|
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'}
|
|
|
|
def main():
|
|
|
|
parser = argparse.ArgumentParser(description='Process some integers.')
|
|
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')
|
|
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()
|
|
|
|
# Create a set to hold the names of all packages and their dependencies
|
|
packages = set()
|
|
|
|
while True:
|
|
try:
|
|
# Grab next package
|
|
pkg_name = in_packages.pop()
|
|
if args.verbose:
|
|
sys.stderr.write('Package: {:s}\n'.format(pkg_name))
|
|
sys.stderr.flush()
|
|
|
|
# Check if the package is in the cache, if not check if it is in
|
|
# our list of known virtual packages
|
|
if pkg_name not in cache:
|
|
if pkg_name in virtual_mapping:
|
|
pkg_name = virtual_mapping[pkg_name]
|
|
else:
|
|
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))
|
|
continue
|
|
|
|
# Add this package (with version) to the list of packages
|
|
packages.add(pkg_name)
|
|
|
|
# Get package dependencies
|
|
for pdep in cache[pkg_name].versions[0].dependencies:
|
|
dep_name = pdep[0].name
|
|
if args.verbose:
|
|
sys.stderr.write('\tDependency: {:s}\n'.format(dep_name))
|
|
sys.stderr.flush()
|
|
if dep_name not in packages:
|
|
if dep_name in virtual_mapping:
|
|
dep_name = virtual_mapping[dep_name]
|
|
packages.add(dep_name)
|
|
in_packages.insert(0, dep_name)
|
|
except IndexError:
|
|
break
|
|
|
|
# Print out the gathered package names
|
|
for package in sorted(packages):
|
|
print(package)
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|