118 lines
3.3 KiB
Python
118 lines
3.3 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2012 OpenStack, LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
r"""
|
|
A simple script to update the requirements files from a global set of
|
|
allowable requirements.
|
|
|
|
The script can be called like this:
|
|
|
|
$> python update.py ../myproj
|
|
|
|
Any requirements listed in the target files will have their versions
|
|
updated to match the global requirements. Requirements not in the global
|
|
files will be dropped.
|
|
"""
|
|
|
|
import os
|
|
import os.path
|
|
import sys
|
|
|
|
from pip import req
|
|
|
|
|
|
def _parse_pip(pip):
|
|
|
|
install_require = req.InstallRequirement.from_line(pip)
|
|
if install_require.editable:
|
|
return pip
|
|
elif install_require.url:
|
|
return pip
|
|
else:
|
|
return install_require.req.key
|
|
|
|
|
|
def _parse_reqs(filename):
|
|
|
|
reqs = dict()
|
|
|
|
pip_requires = open(filename, "r").readlines()
|
|
for pip in pip_requires:
|
|
pip = pip.strip()
|
|
if pip.startswith("#") or len(pip) == 0:
|
|
continue
|
|
reqs[_parse_pip(pip)] = pip
|
|
return reqs
|
|
|
|
|
|
def _copy_requires(source_path, dest_dir):
|
|
"""Copy requirements files."""
|
|
|
|
target_map = {
|
|
'requirements.txt': ('requirements.txt', 'tools/pip-requires'),
|
|
'test-requirements.txt': (
|
|
'test-requirements.txt', 'tools/test-requires'),
|
|
}
|
|
for dest in target_map[source_path]:
|
|
dest_path = os.path.join(dest_dir, dest)
|
|
if os.path.exists(dest_path):
|
|
break
|
|
|
|
# Catch the fall through
|
|
if not os.path.exists(dest_path):
|
|
# This can happen, we try all paths
|
|
return
|
|
|
|
source_reqs = _parse_reqs(source_path)
|
|
|
|
with open(dest_path, 'r') as dest_reqs_file:
|
|
dest_reqs = dest_reqs_file.readlines()
|
|
|
|
print "Syncing %s" % source_path
|
|
|
|
with open(dest_path, 'w') as new_reqs:
|
|
for old_require in dest_reqs:
|
|
# copy comments
|
|
if old_require[0] == '#':
|
|
new_reqs.write(old_require)
|
|
continue
|
|
old_require = old_require.rstrip().lower()
|
|
if not old_require:
|
|
continue
|
|
old_pip = _parse_pip(old_require)
|
|
|
|
# Special cases:
|
|
# projects need to align pep8 version on their own time
|
|
if "pep8" in old_pip:
|
|
new_reqs.write("%s\n" % old_require)
|
|
continue
|
|
# versions of our stuff from tarballs.openstack.org are ok
|
|
if "http://tarballs.openstack.org/" in old_pip:
|
|
new_reqs.write("%s\n" % old_require)
|
|
continue
|
|
|
|
if old_pip in source_reqs:
|
|
new_reqs.write("%s\n" % source_reqs[old_pip])
|
|
|
|
|
|
def main(argv):
|
|
for req in ('requirements.txt', 'test-requirements.txt'):
|
|
_copy_requires(req, argv[0])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv[1:])
|