1
# Copyright 2014 Pants project contributors (see CONTRIBUTORS.md).
2
# Licensed under the Apache License, Version 2.0 (see LICENSE).
4
from __future__ import absolute_import
6
from collections import Iterable
8
from pkg_resources import Requirement
10
from .compatibility import string as compatibility_string
12
REQUIRED_ATTRIBUTES = (
20
def quacks_like_req(req):
21
return all(hasattr(req, attr) for attr in REQUIRED_ATTRIBUTES)
24
def maybe_requirement(req):
25
if isinstance(req, Requirement) or quacks_like_req(req):
27
elif isinstance(req, compatibility_string):
28
return Requirement.parse(req)
29
raise ValueError('Unknown requirement %r' % (req,))
32
def maybe_requirement_list(reqs):
33
if isinstance(reqs, (compatibility_string, Requirement)) or quacks_like_req(reqs):
34
return [maybe_requirement(reqs)]
35
elif isinstance(reqs, Iterable):
36
return [maybe_requirement(req) for req in reqs]
37
raise ValueError('Unknown requirement list %r' % (reqs,))
40
def requirement_is_exact(req):
41
return req.specs and len(req.specs) == 1 and req.specs[0][0] == '=='