clearml/trains/storage/util.py

42 lines
1.3 KiB
Python

from six.moves.urllib.parse import quote, urlparse, urlunparse
import six
import fnmatch
def get_config_object_matcher(**patterns):
unsupported = {k: v for k, v in patterns.items() if not isinstance(v, six.string_types)}
if unsupported:
raise ValueError('Unsupported object matcher (expecting string): %s'
% ', '.join('%s=%s' % (k, v) for k, v in unsupported.items()))
# optimize simple patters
starts_with = {k: v.rstrip('*') for k, v in patterns.items() if '*' not in v.rstrip('*') and '?' not in v}
patterns = {k: v for k, v in patterns.items() if v not in starts_with}
def _matcher(**kwargs):
for key, value in kwargs.items():
if not value:
continue
start = starts_with.get(key)
if start:
if value.startswith(start):
return True
else:
pat = patterns.get(key)
if pat and fnmatch.fnmatch(value, pat):
return True
return _matcher
def quote_url(url):
parsed = urlparse(url)
if parsed.scheme not in ('http', 'https'):
return url
parsed = parsed._replace(path=quote(parsed.path))
return urlunparse(parsed)
def encode_string_to_filename(text):
return quote(text, safe=" ")