mirror of
https://github.com/kennethreitz/pip-pop.git
synced 2026-06-05 23:10:17 +00:00
Kenneth Reitz
132 lines
3.2 KiB
Python
Executable File
132 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""Usage:
|
|
pip-diff (--fresh | --stale) <reqfile1> <reqfile2>
|
|
pip-diff (-h | --help)
|
|
|
|
Options:
|
|
-h --help Show this screen.
|
|
--fresh List newly added packages.
|
|
--stale List removed packages.
|
|
"""
|
|
import os
|
|
from docopt import docopt
|
|
from pkg_resources import parse_requirements
|
|
|
|
|
|
# TODO: ignore lines
|
|
IGNORABLE_LINES = ('#', '-r')
|
|
VERSION_OPERATORS = ('==', '>=', '<=', '>', '<', ',')
|
|
|
|
def split(s):
|
|
for operator in VERSION_OPERATORS:
|
|
s = s.replace(operator, '!')
|
|
|
|
return s.split('!')
|
|
|
|
class Requirements(object):
|
|
def __init__(self, reqfile=None):
|
|
super(Requirements, self).__init__()
|
|
self.path = reqfile
|
|
self.requirements = []
|
|
|
|
if reqfile:
|
|
self.load(reqfile)
|
|
|
|
def __repr__(self):
|
|
return '<Requirements \'{}\'>'.format(self.path)
|
|
|
|
def load(self, reqfile):
|
|
|
|
if not os.path.exists(reqfile):
|
|
raise ValueError('The given requirements file does not exist.')
|
|
|
|
with open(reqfile) as f:
|
|
data = []
|
|
|
|
for line in f:
|
|
line = line.strip()
|
|
|
|
# Skip lines that start with any comment/control charecters.
|
|
if not any([line.startswith(p) for p in IGNORABLE_LINES]):
|
|
data.append(line)
|
|
|
|
for requirement in parse_requirements(data):
|
|
self.requirements.append(requirement)
|
|
|
|
|
|
def diff(self, requirements, ignore_versions=False):
|
|
r1 = self
|
|
r2 = requirements
|
|
results = {'fresh': [], 'stale': []}
|
|
|
|
# Generate fresh packages.
|
|
other_reqs = (
|
|
[r.project_name for r in r1.requirements]
|
|
if ignore_versions else r1.requirements
|
|
)
|
|
|
|
for req in r2.requirements:
|
|
r = req.project_name if ignore_versions else req
|
|
|
|
if r not in other_reqs:
|
|
results['fresh'].append(req)
|
|
|
|
# Generate stale packages.
|
|
other_reqs = (
|
|
[r.project_name for r in r2.requirements]
|
|
if ignore_versions else r2.requirements
|
|
)
|
|
|
|
for req in r1.requirements:
|
|
r = req.project_name if ignore_versions else req
|
|
|
|
if r not in other_reqs:
|
|
results['stale'].append(req)
|
|
|
|
return results
|
|
|
|
|
|
|
|
|
|
|
|
def diff(r1, r2, include_fresh=False, include_stale=False):
|
|
|
|
include_versions = True if include_stale else False
|
|
|
|
try:
|
|
r1 = Requirements(r1)
|
|
r2 = Requirements(r2)
|
|
except ValueError:
|
|
print 'There was a problem loading the given requirements files.'
|
|
exit(os.EX_NOINPUT)
|
|
|
|
results = r1.diff(r2, ignore_versions=True)
|
|
|
|
if include_fresh:
|
|
for line in results['fresh']:
|
|
print line.project_name if include_versions else line
|
|
|
|
if include_stale:
|
|
for line in results['stale']:
|
|
print line.project_name if include_versions else line
|
|
|
|
|
|
|
|
def main():
|
|
args = docopt(__doc__, version='pip-diff')
|
|
|
|
kwargs = {
|
|
'r1': args['<reqfile1>'],
|
|
'r2': args['<reqfile2>'],
|
|
'include_fresh': args['--fresh'],
|
|
'include_stale': args['--stale']
|
|
}
|
|
|
|
diff(**kwargs)
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main() |