Skip to content
Snippets Groups Projects
Commit d047f26a authored by Ed Morley's avatar Ed Morley
Browse files

Import upstream pip-pop fix for pip 8 support

Is the fix from kennethreitz/pip-pop#12.
parent 998f996a
No related branches found
No related tags found
No related merge requests found
...@@ -18,6 +18,7 @@ from pip._vendor.requests import session ...@@ -18,6 +18,7 @@ from pip._vendor.requests import session
requests = session() requests = session()
class Requirements(object): class Requirements(object):
def __init__(self, reqfile=None): def __init__(self, reqfile=None):
super(Requirements, self).__init__() super(Requirements, self).__init__()
...@@ -31,13 +32,15 @@ class Requirements(object): ...@@ -31,13 +32,15 @@ class Requirements(object):
return '<Requirements \'{}\'>'.format(self.path) return '<Requirements \'{}\'>'.format(self.path)
def load(self, reqfile): def load(self, reqfile):
if not os.path.exists(reqfile): if not os.path.exists(reqfile):
raise ValueError('The given requirements file does not exist.') raise ValueError('The given requirements file does not exist.')
finder = PackageFinder([], [], session=requests) finder = PackageFinder([], [], session=requests)
for requirement in parse_requirements(reqfile, finder=finder, session=requests): for requirement in parse_requirements(reqfile, finder=finder, session=requests):
if requirement.req: if requirement.req:
if not getattr(requirement.req, 'name', None):
# Prior to pip 8.1.2 the attribute `name` did not exist.
requirement.req.name = requirement.req.project_name
self.requirements.append(requirement.req) self.requirements.append(requirement.req)
...@@ -48,24 +51,24 @@ class Requirements(object): ...@@ -48,24 +51,24 @@ class Requirements(object):
# Generate fresh packages. # Generate fresh packages.
other_reqs = ( other_reqs = (
[r.project_name for r in r1.requirements] [r.name for r in r1.requirements]
if ignore_versions else r1.requirements if ignore_versions else r1.requirements
) )
for req in r2.requirements: for req in r2.requirements:
r = req.project_name if ignore_versions else req r = req.name if ignore_versions else req
if r not in other_reqs and r not in excludes: if r not in other_reqs and r not in excludes:
results['fresh'].append(req) results['fresh'].append(req)
# Generate stale packages. # Generate stale packages.
other_reqs = ( other_reqs = (
[r.project_name for r in r2.requirements] [r.name for r in r2.requirements]
if ignore_versions else r2.requirements if ignore_versions else r2.requirements
) )
for req in r1.requirements: for req in r1.requirements:
r = req.project_name if ignore_versions else req r = req.name if ignore_versions else req
if r not in other_reqs and r not in excludes: if r not in other_reqs and r not in excludes:
results['stale'].append(req) results['stale'].append(req)
...@@ -73,11 +76,7 @@ class Requirements(object): ...@@ -73,11 +76,7 @@ class Requirements(object):
return results return results
def diff(r1, r2, include_fresh=False, include_stale=False, excludes=None): def diff(r1, r2, include_fresh=False, include_stale=False, excludes=None):
include_versions = True if include_stale else False include_versions = True if include_stale else False
excludes = excludes if len(excludes) else [] excludes = excludes if len(excludes) else []
...@@ -92,12 +91,11 @@ def diff(r1, r2, include_fresh=False, include_stale=False, excludes=None): ...@@ -92,12 +91,11 @@ def diff(r1, r2, include_fresh=False, include_stale=False, excludes=None):
if include_fresh: if include_fresh:
for line in results['fresh']: for line in results['fresh']:
print(line.project_name if include_versions else line) print(line.name if include_versions else line)
if include_stale: if include_stale:
for line in results['stale']: for line in results['stale']:
print(line.project_name if include_versions else line) print(line.name if include_versions else line)
def main(): def main():
...@@ -114,6 +112,5 @@ def main(): ...@@ -114,6 +112,5 @@ def main():
diff(**kwargs) diff(**kwargs)
if __name__ == '__main__': if __name__ == '__main__':
main() main()
...@@ -15,6 +15,7 @@ from pip._vendor.requests import session ...@@ -15,6 +15,7 @@ from pip._vendor.requests import session
requests = session() requests = session()
class Requirements(object): class Requirements(object):
def __init__(self, reqfile=None): def __init__(self, reqfile=None):
super(Requirements, self).__init__() super(Requirements, self).__init__()
...@@ -28,38 +29,31 @@ class Requirements(object): ...@@ -28,38 +29,31 @@ class Requirements(object):
return '<Requirements \'{}\'>'.format(self.path) return '<Requirements \'{}\'>'.format(self.path)
def load(self, reqfile): def load(self, reqfile):
if not os.path.exists(reqfile): if not os.path.exists(reqfile):
raise ValueError('The given requirements file does not exist.') raise ValueError('The given requirements file does not exist.')
finder = PackageFinder([], [], session=requests) finder = PackageFinder([], [], session=requests)
for requirement in parse_requirements(reqfile, finder=finder, session=requests): for requirement in parse_requirements(reqfile, finder=finder, session=requests):
self.requirements.append(requirement) if requirement.req:
if not getattr(requirement.req, 'name', None):
# Prior to pip 8.1.2 the attribute `name` did not exist.
requirement.req.name = requirement.req.project_name
self.requirements.append(requirement.req)
def grep(reqfile, packages, silent=False): def grep(reqfile, packages, silent=False):
try: try:
r = Requirements(reqfile) r = Requirements(reqfile)
except ValueError: except ValueError:
if not silent: if not silent:
print('There was a problem loading the given requirement file.') print('There was a problem loading the given requirement file.')
exit(os.EX_NOINPUT) exit(os.EX_NOINPUT)
for requirement in r.requirements: for req in r.requirements:
if req.name in packages:
if requirement.req: if not silent:
print('Package {} found!'.format(req.name))
if requirement.req.project_name in packages: exit(0)
if not silent:
print('Package {} found!'.format(requirement.req.project_name))
exit(0)
if not silent: if not silent:
print('Not found.') print('Not found.')
...@@ -72,10 +66,8 @@ def main(): ...@@ -72,10 +66,8 @@ def main():
kwargs = {'reqfile': args['<reqfile>'], 'packages': args['<package>'], 'silent': args['-s']} kwargs = {'reqfile': args['<reqfile>'], 'packages': args['<package>'], 'silent': args['-s']}
grep(**kwargs) grep(**kwargs)
if __name__ == '__main__': if __name__ == '__main__':
main() main()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment