certbot/tools/merge_requirements.py

#!/usr/bin/env python
"""Merges multiple Python requirements files into one file.

Requirements files specified later take precedence over earlier ones. Only
simple SomeProject==1.2.3 format is currently supported.

"""

from __future__ import print_function

import sys


def read_file(file_path):
    """Reads in a Python requirements file.

    :param str file_path: path to requirements file

    :returns: mapping from a project to its pinned version
    :rtype: dict

    """
    d = {}
    with open(file_path) as f:
        for line in f:
            line = line.strip()
            if line and not line.startswith('#'):
                project, version = line.split('==')
                if not version:
                    raise ValueError("Unexpected syntax '{0}'".format(line))
                d[project] = version
    return d


def print_requirements(requirements):
    """Prints requirements to stdout.

    :param dict requirements: mapping from a project to its pinned version

    """
    print('\n'.join('{0}=={1}'.format(k, v)
          for k, v in sorted(requirements.items())))


def merge_requirements_files(*files):
    """Merges multiple requirements files together and prints the result.

    Requirement files specified later in the list take precedence over earlier
    files.

    :param tuple files: paths to requirements files

    """
    d = {}
    for f in files:
        d.update(read_file(f))
    print_requirements(d)


if __name__ == '__main__':
    merge_requirements_files(*sys.argv[1:])