mirror of
https://github.com/clearml/clearml-agent
synced 2025-01-31 09:06:52 +00:00
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
import re
|
|
|
|
# Copied from pip
|
|
# https://github.com/pypa/pip/blob/281eb61b09d87765d7c2b92f6982b3fe76ccb0af/pip/index.py#L947
|
|
HASH_ALGORITHMS = set(['sha1', 'sha224', 'sha384', 'sha256', 'sha512', 'md5'])
|
|
|
|
extras_require_search = re.compile(
|
|
r'(?P<name>.+)\[(?P<extras>[^\]]+)\]').search
|
|
|
|
|
|
def parse_fragment(fragment_string):
|
|
"""Takes a fragment string nd returns a dict of the components"""
|
|
fragment_string = fragment_string.lstrip('#')
|
|
|
|
try:
|
|
return dict(
|
|
key_value_string.split('=')
|
|
for key_value_string in fragment_string.split('&')
|
|
)
|
|
except ValueError:
|
|
raise ValueError(
|
|
'Invalid fragment string {fragment_string}'.format(
|
|
fragment_string=fragment_string
|
|
)
|
|
)
|
|
|
|
|
|
def get_hash_info(d):
|
|
"""Returns the first matching hashlib name and value from a dict"""
|
|
for key in d.keys():
|
|
if key.lower() in HASH_ALGORITHMS:
|
|
return key, d[key]
|
|
|
|
return None, None
|
|
|
|
|
|
def parse_extras_require(egg):
|
|
if egg is not None:
|
|
match = extras_require_search(egg)
|
|
if match is not None:
|
|
name = match.group('name')
|
|
extras = match.group('extras')
|
|
return name, [extra.strip() for extra in extras.split(',')]
|
|
return egg, []
|