requirements-fix-names.py

#!/usr/bin/env python

from __future__ import print_function

import os
import packaging.utils
import packaging.requirements
import sys

try:
    gr = sys.argv[1]
except IndexError:
    gr = '~/projects/openstack/openstack/requirements/global-requirements.txt'

with open(os.path.expanduser(gr)) as gr:
    for line in gr:
        if line.startswith('#') or line == '\n':
            print(line, end='')
        else:
            if '#' in line:
                (line, _, license) = line.partition('#')
            requirement = line.rstrip().lstrip()
            r = packaging.requirements.Requirement(requirement)
            r.name = packaging.utils.canonicalize_name(r.name)
            print("%s  #%s" %(r, license), end='')