#! /usr/bin/python3

import apt_pkg
import sys
sys.path.append('/usr/share/botch')
from util import get_fh_out, read_tag_file


def single_verson(inPackages, outPackages, verbose=False):
    apt_pkg.init()

    packages = dict()
    for pkg2 in inPackages:
        pkg1 = packages.get((pkg2['Package'], pkg2['Architecture']))
        if not pkg1 or (apt_pkg.version_compare(pkg2['Version'],
                                                pkg1['Version']) > 0):
            packages[(pkg2['Package'], pkg2['Architecture'])] = pkg2

    with outPackages as outfile:
        for pkg in sorted(packages.keys()):
            packages[pkg].dump(outfile)
            outfile.write(b"\n")

    return True

if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser(
        description=("filters a Packages file so that only the newest " +
                     "version of each package remains"))
    parser.add_argument(
        'inPackages', type=read_tag_file, help='input Packages file')
    parser.add_argument(
        'outPackages', type=get_fh_out, help='output Packages file')
    parser.add_argument('--verbose', action='store_true', help='be verbose')
    args = parser.parse_args()
    ret = single_verson(args.inPackages, args.outPackages, args.verbose)
    exit(not ret)
