mirror of
https://github.com/kennethreitz/pip-pop.git
synced 2026-06-05 06:56:12 +00:00
74 lines
1.9 KiB
Python
Executable File
74 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""Usage:
|
|
pip-grep [-s] <reqfile> <package>...
|
|
|
|
Options:
|
|
-h --help Show this screen.
|
|
"""
|
|
import os
|
|
from docopt import docopt
|
|
from pip.req import parse_requirements
|
|
from pip.index import PackageFinder
|
|
from pip._vendor.requests import session
|
|
|
|
requests = session()
|
|
|
|
|
|
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.')
|
|
|
|
finder = PackageFinder([], [], session=requests)
|
|
for requirement in parse_requirements(reqfile, finder=finder, session=requests):
|
|
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):
|
|
try:
|
|
r = Requirements(reqfile)
|
|
except ValueError:
|
|
if not silent:
|
|
print('There was a problem loading the given requirement file.')
|
|
exit(os.EX_NOINPUT)
|
|
|
|
for req in r.requirements:
|
|
if req.name in packages:
|
|
if not silent:
|
|
print('Package {} found!'.format(req.name))
|
|
exit(0)
|
|
|
|
if not silent:
|
|
print('Not found.')
|
|
|
|
exit(1)
|
|
|
|
|
|
def main():
|
|
args = docopt(__doc__, version='pip-grep')
|
|
|
|
kwargs = {'reqfile': args['<reqfile>'], 'packages': args['<package>'], 'silent': args['-s']}
|
|
|
|
grep(**kwargs)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|