1 # This file is dual licensed under the terms of the Apache License, Version
2 # 2.0, and the BSD License. See the LICENSE file in the root of this repository
3 # for complete details.
8 from typing
import List
, Optional
as TOptional
, Set
10 from pip
._vendor
.pyparsing
import ( # noqa
23 from .markers
import MARKER_EXPR
, Marker
24 from .specifiers
import LegacySpecifier
, Specifier
, SpecifierSet
27 class InvalidRequirement(ValueError):
29 An invalid requirement was found, users should refer to PEP 508.
33 ALPHANUM
= Word(string
.ascii_letters
+ string
.digits
)
35 LBRACKET
= L("[").suppress()
36 RBRACKET
= L("]").suppress()
37 LPAREN
= L("(").suppress()
38 RPAREN
= L(")").suppress()
39 COMMA
= L(",").suppress()
40 SEMICOLON
= L(";").suppress()
41 AT
= L("@").suppress()
43 PUNCTUATION
= Word("-_.")
44 IDENTIFIER_END
= ALPHANUM |
(ZeroOrMore(PUNCTUATION
) + ALPHANUM
)
45 IDENTIFIER
= Combine(ALPHANUM
+ ZeroOrMore(IDENTIFIER_END
))
47 NAME
= IDENTIFIER("name")
50 URI
= Regex(r
"[^ ]+")("url")
53 EXTRAS_LIST
= EXTRA
+ ZeroOrMore(COMMA
+ EXTRA
)
54 EXTRAS
= (LBRACKET
+ Optional(EXTRAS_LIST
) + RBRACKET
)("extras")
56 VERSION_PEP440
= Regex(Specifier
._regex
_str
, re
.VERBOSE | re
.IGNORECASE
)
57 VERSION_LEGACY
= Regex(LegacySpecifier
._regex
_str
, re
.VERBOSE | re
.IGNORECASE
)
59 VERSION_ONE
= VERSION_PEP440 ^ VERSION_LEGACY
60 VERSION_MANY
= Combine(
61 VERSION_ONE
+ ZeroOrMore(COMMA
+ VERSION_ONE
), joinString
=",", adjacent
=False
63 _VERSION_SPEC
= Optional((LPAREN
+ VERSION_MANY
+ RPAREN
) | VERSION_MANY
)
64 _VERSION_SPEC
.setParseAction(lambda s
, l
, t
: t
._raw
_spec
or "")
66 VERSION_SPEC
= originalTextFor(_VERSION_SPEC
)("specifier")
67 VERSION_SPEC
.setParseAction(lambda s
, l
, t
: t
[1])
69 MARKER_EXPR
= originalTextFor(MARKER_EXPR())("marker")
70 MARKER_EXPR
.setParseAction(
71 lambda s
, l
, t
: Marker(s
[t
._original
_start
: t
._original
_end
])
73 MARKER_SEPARATOR
= SEMICOLON
74 MARKER
= MARKER_SEPARATOR
+ MARKER_EXPR
76 VERSION_AND_MARKER
= VERSION_SPEC
+ Optional(MARKER
)
77 URL_AND_MARKER
= URL
+ Optional(MARKER
)
79 NAMED_REQUIREMENT
= NAME
+ Optional(EXTRAS
) + (URL_AND_MARKER | VERSION_AND_MARKER
)
81 REQUIREMENT
= stringStart
+ NAMED_REQUIREMENT
+ stringEnd
82 # pyparsing isn't thread safe during initialization, so we do it eagerly, see
84 REQUIREMENT
.parseString("x[]")
88 """Parse a requirement.
90 Parse a given requirement string into its parts, such as name, specifier,
91 URL, and extras. Raises InvalidRequirement on a badly-formed requirement
95 # TODO: Can we test whether something is contained within a requirement?
96 # If so how do we do that? Do we need to test against the _name_ of
97 # the thing as well as the version? What about the markers?
98 # TODO: Can we normalize the name and extra name?
100 def __init__(self
, requirement_string
: str) -> None:
102 req
= REQUIREMENT
.parseString(requirement_string
)
103 except ParseException
as e
:
104 raise InvalidRequirement(
105 f
'Parse error at "{ requirement_string[e.loc : e.loc + 8]!r}": {e.msg}'
108 self
.name
: str = req
.name
110 parsed_url
= urllib
.parse
.urlparse(req
.url
)
111 if parsed_url
.scheme
== "file":
112 if urllib
.parse
.urlunparse(parsed_url
) != req
.url
:
113 raise InvalidRequirement("Invalid URL given")
114 elif not (parsed_url
.scheme
and parsed_url
.netloc
) or (
115 not parsed_url
.scheme
and not parsed_url
.netloc
117 raise InvalidRequirement(f
"Invalid URL: {req.url}")
118 self
.url
: TOptional
[str] = req
.url
121 self
.extras
: Set
[str] = set(req
.extras
.asList() if req
.extras
else [])
122 self
.specifier
: SpecifierSet
= SpecifierSet(req
.specifier
)
123 self
.marker
: TOptional
[Marker
] = req
.marker
if req
.marker
else None
125 def __str__(self
) -> str:
126 parts
: List
[str] = [self
.name
]
129 formatted_extras
= ",".join(sorted(self
.extras
))
130 parts
.append(f
"[{formatted_extras}]")
133 parts
.append(str(self
.specifier
))
136 parts
.append(f
"@ {self.url}")
141 parts
.append(f
"; {self.marker}")
143 return "".join(parts
)
145 def __repr__(self
) -> str:
146 return f
"<Requirement('{self}')>"