This document is for an in-development version of Galaxy. You can alternatively view this page in the latest release if it exists or view the top of the latest release's documentation.

Source code for galaxy.util

Utility functions used systemwide.


import binascii
import codecs
import collections
import errno
import importlib
import itertools
import json
import os
import random
import re
import shlex
import shutil
import smtplib
import stat
import string
import sys
import tempfile
import textwrap
import threading
import time
import unicodedata
import xml.dom.minidom
from datetime import (
from decimal import Decimal
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from hashlib import md5
from os.path import relpath
from pathlib import Path
from typing import (
from urllib.parse import (

from boltons.iterutils import (
from requests.adapters import HTTPAdapter
from requests.packages.urllib3.util.retry import Retry  # type: ignore[import-untyped, unused-ignore]
from typing_extensions import (

from galaxy.util import requests

    import grp
except ImportError:
    # For Pulsar on Windows (which does not use the function that uses grp)
    grp = None  # type: ignore[assignment]
    from lxml import etree

    # lxml.etree.Element is a function that returns a new instance of the
    # lxml.etree._Element class. This class doesn't have a proper __init__()
    # method, so we can add a __new__() constructor that mimicks the
    # xml.etree.ElementTree.Element initialization.
[docs] class Element(etree._Element): def __new__(cls, tag, attrib={}, **extra) -> Self: # noqa: B006 return cast(Self, etree.Element(tag, attrib, **extra)) def __iter__(self) -> Iterator[Self]: # type: ignore[override] return cast(Iterator[Self], super().__iter__())
[docs] def find(self, path: str, namespaces: Optional[Mapping[str, str]] = None) -> Union[Self, None]: ret = super().find(path, namespaces) if ret is not None: return cast(Self, ret) else: return None
[docs] def findall(self, path: str, namespaces: Optional[Mapping[str, str]] = None) -> List[Self]: # type: ignore[override] return cast(List[Self], super().findall(path, namespaces))
[docs] def SubElement(parent: Element, tag: str, attrib: Optional[Dict[str, str]] = None, **extra) -> Element: return cast(Element, etree.SubElement(parent, tag, attrib, **extra))
# lxml.etree.ElementTree is a function that returns a new instance of the # lxml.etree._ElementTree class. This class doesn't have a proper __init__() # method, so we can add a __new__() constructor that mimicks the # xml.etree.ElementTree.ElementTree initialization.
[docs] class ElementTree(etree._ElementTree): def __new__(cls, element=None, file=None) -> Self: return cast(Self, etree.ElementTree(element, file=file))
[docs] def getroot(self) -> Element: return cast(Element, super().getroot())
[docs] def XML(text: Union[str, bytes]) -> Element: return cast(Element, etree.XML(text))
except ImportError: LXML_AVAILABLE = False import xml.etree.ElementTree as etree # type: ignore[no-redef] from xml.etree.ElementTree import ( # type: ignore[assignment] # noqa: F401 Element, ElementTree, XML, ) from .custom_logging import get_logger from .inflection import Inflector from .path import ( # noqa: F401 safe_contains, safe_makedirs, safe_relpath, StrPath, ) from .rst_to_html import rst_to_html # noqa: F401 try: shlex_join = shlex.join # type: ignore[attr-defined, unused-ignore] except AttributeError: # Python < 3.8 def shlex_join(split_command): return " ".join(map(shlex.quote, split_command)) inflector = Inflector() log = get_logger(__name__) _lock = threading.RLock() namedtuple = collections.namedtuple CHUNK_SIZE = 65536 # 64k DATABASE_MAX_STRING_SIZE = 32768 DATABASE_MAX_STRING_SIZE_PRETTY = "32K" DEFAULT_SOCKET_TIMEOUT = 600 gzip_magic = b"\x1f\x8b" bz2_magic = b"BZh" xz_magic = b"\xfd7zXZ\x00" DEFAULT_ENCODING = os.environ.get("GALAXY_DEFAULT_ENCODING", "utf-8") NULL_CHAR = b"\x00" BINARY_CHARS = [NULL_CHAR] FILENAME_VALID_CHARS = ".,^_-()[]0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ" RW_R__R__ = stat.S_IRUSR | stat.S_IWUSR | stat.S_IRGRP | stat.S_IROTH RWXR_XR_X = stat.S_IRWXU | stat.S_IRGRP | stat.S_IXGRP | stat.S_IROTH | stat.S_IXOTH RWXRWXRWX = stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO defaultdict = collections.defaultdict UNKNOWN = "unknown"
[docs]def str_removeprefix(s: str, prefix: str): """ str.removeprefix() equivalent for Python < 3.9 """ if sys.version_info >= (3, 9): return s.removeprefix(prefix) elif s.startswith(prefix): return s[len(prefix) :] else: return s
[docs]def remove_protocol_from_url(url): """Supplied URL may be null, if not ensure http:// or https:// etc... is stripped off. """ if url is None: return url # We have a URL if url.find("://") > 0: new_url = url.split("://")[1] else: new_url = url return new_url.rstrip("/")
[docs]def is_binary(value): """ File is binary if it contains a null-byte by default (e.g. behavior of grep, etc.). This may fail for utf-16 files, but so would ASCII encoding. >>> is_binary( string.printable ) False >>> is_binary( b'\\xce\\x94' ) False >>> is_binary( b'\\x00' ) True """ value = smart_str(value) for binary_char in BINARY_CHARS: if binary_char in value: return True return False
[docs]def is_uuid(value): """ This method returns True if value is a UUID, otherwise False. >>> is_uuid( "123e4567-e89b-12d3-a456-426655440000" ) True >>> is_uuid( "0x3242340298902834" ) False """ uuid_re = re.compile("[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}") if re.match(uuid_re, str(value)): return True else: return False
[docs]def directory_hash_id(id): """ >>> directory_hash_id( 100 ) ['000'] >>> directory_hash_id( "90000" ) ['090'] >>> directory_hash_id("777777777") ['000', '777', '777'] >>> directory_hash_id("135ee48a-4f51-470c-ae2f-ce8bd78799e6") ['1', '3', '5'] """ s = str(id) len_s = len(s) # Shortcut -- ids 0-999 go under ../000/ if len_s < 4: return ["000"] if not is_uuid(s): # Pad with zeros until a multiple of three padded = ((3 - len_s % 3) * "0") + s # Drop the last three digits -- 1000 files per directory padded = padded[:-3] # Break into chunks of three return [padded[i * 3 : (i + 1) * 3] for i in range(len(padded) // 3)] else: # assume it is a UUID return list(iter(s[0:3]))
[docs]def get_charset_from_http_headers(headers, default=None): rval = headers.get("content-type", None) if rval and "charset=" in rval: rval = rval.split("charset=")[-1].split(";")[0].strip() if rval: return rval return default
[docs]def synchronized(func): """This wrapper will serialize access to 'func' to a single thread. Use it as a decorator.""" def caller(*params, **kparams): _lock.acquire(True) # Wait try: return func(*params, **kparams) finally: _lock.release() return caller
[docs]def iter_start_of_line(fh, chunk_size=None): """Iterate over fh and call readline(chunk_size).""" while True: data = fh.readline(chunk_size) if not data: break if not data.endswith("\n"): # Discard the rest of the line fh.readline() yield data
[docs]def file_reader(fp, chunk_size=CHUNK_SIZE): """This generator yields the open file object in chunks (default 64k).""" while True: data = fp.read(chunk_size) if not data: break yield data
[docs]def chunk_iterable(it: Iterable, size: int = 1000) -> Iterator[tuple]: """ Break an iterable into chunks of ``size`` elements. >>> list(chunk_iterable([1, 2, 3, 4, 5, 6, 7], 3)) [(1, 2, 3), (4, 5, 6), (7,)] """ it = iter(it) while True: p = tuple(itertools.islice(it, size)) if not p: break yield p
[docs]def unique_id(KEY_SIZE=128): """ Generates an unique id >>> ids = [ unique_id() for i in range(1000) ] >>> len(set(ids)) 1000 """ random_bits = str(random.getrandbits(KEY_SIZE)).encode("UTF-8") return md5(random_bits).hexdigest()
[docs]def parse_xml( fname: StrPath, strip_whitespace=True, remove_comments=True, schemafname: Union[StrPath, None] = None ) -> ElementTree: """Returns a parsed xml tree""" parser = None schema = None if remove_comments and LXML_AVAILABLE: # If using stdlib etree comments are always removed, # but lxml doesn't do this by default parser = etree.XMLParser(remove_comments=remove_comments) if LXML_AVAILABLE and schemafname: with open(str(schemafname), "rb") as schema_file: schema_root = etree.XML(schema_file.read()) schema = etree.XMLSchema(schema_root) try: tree = cast(ElementTree, etree.parse(str(fname), parser=parser)) root = tree.getroot() if strip_whitespace: for elem in root.iter("*"): if elem.text is not None: elem.text = elem.text.strip() if elem.tail is not None: elem.tail = elem.tail.strip() if schema: schema.assertValid(tree) except OSError as e: if e.errno is None and not os.path.exists(fname): # type: ignore[unreachable] # lxml doesn't set errno e.errno = errno.ENOENT # type: ignore[unreachable] raise except etree.ParseError: log.exception("Error parsing file %s", fname) raise except etree.DocumentInvalid: log.exception("Validation of file %s failed", fname) raise return tree
[docs]def parse_xml_string(xml_string: str, strip_whitespace: bool = True) -> Element: try: elem = XML(xml_string) except ValueError as e: if "strings with encoding declaration are not supported" in unicodify(e): # This happens with lxml for a string that starts with e.g. `<?xml version="1.0" encoding="UTF-8"?>` elem = XML(xml_string.encode("utf-8")) else: raise e if strip_whitespace: for sub_elem in elem.iter("*"): if sub_elem.text is not None: sub_elem.text = sub_elem.text.strip() if sub_elem.tail is not None: sub_elem.tail = sub_elem.tail.strip() return elem
[docs]def parse_xml_string_to_etree(xml_string: str, strip_whitespace: bool = True) -> ElementTree: return ElementTree(parse_xml_string(xml_string=xml_string, strip_whitespace=strip_whitespace))
[docs]def xml_to_string(elem: Optional[Element], pretty: bool = False) -> str: """ Returns a string from an xml tree. """ if elem is None: return "" try: xml_str = etree.tostring(elem, encoding="unicode") except TypeError as e: # we assume this is a comment if hasattr(elem, "text"): return f"<!-- {elem.text} -->\n" else: raise e if xml_str and pretty: pretty_string = xml.dom.minidom.parseString(xml_str).toprettyxml(indent=" ") return "\n".join(line for line in pretty_string.split("\n") if not re.match(r"^[\s\\nb\']*$", line)) return xml_str
[docs]def xml_element_compare(elem1, elem2): if not isinstance(elem1, dict): elem1 = xml_element_to_dict(elem1) if not isinstance(elem2, dict): elem2 = xml_element_to_dict(elem2) return elem1 == elem2
[docs]def xml_element_list_compare(elem_list1, elem_list2): return [xml_element_to_dict(elem) for elem in elem_list1] == [xml_element_to_dict(elem) for elem in elem_list2]
[docs]def xml_element_to_dict(elem): rval = {} if elem.attrib: rval[elem.tag] = {} else: rval[elem.tag] = None sub_elems = list(elem) if sub_elems: sub_elem_dict = {} for sub_sub_elem_dict in map(xml_element_to_dict, sub_elems): for key, value in sub_sub_elem_dict.items(): if key not in sub_elem_dict: sub_elem_dict[key] = [] sub_elem_dict[key].append(value) for key, value in sub_elem_dict.items(): if len(value) == 1: rval[elem.tag][key] = value[0] else: rval[elem.tag][key] = value if elem.attrib: for key, value in elem.attrib.items(): rval[elem.tag][f"@{key}"] = value if elem.text: text = elem.text.strip() if text and sub_elems or elem.attrib: rval[elem.tag]["#text"] = text else: rval[elem.tag] = text return rval
[docs]def pretty_print_xml(elem, level=0): pad = " " i = "\n" + level * pad if len(elem): if not elem.text or not elem.text.strip(): elem.text = i + pad + pad if not elem.tail or not elem.tail.strip(): elem.tail = i for e in elem: pretty_print_xml(e, level + 1) if not elem.tail or not elem.tail.strip(): elem.tail = i else: if level and (not elem.tail or not elem.tail.strip()): elem.tail = i + pad return elem
[docs]def get_file_size(value, default=None): try: # try built-in return os.path.getsize(value) except Exception: try: # try built-in one name attribute return os.path.getsize(value.name) except Exception: try: # try tell() of end of object offset = value.tell() value.seek(0, 2) rval = value.tell() value.seek(offset) return rval except Exception: # return default value return default
[docs]def shrink_stream_by_size( value, size, join_by=b"..", left_larger=True, beginning_on_size_error=False, end_on_size_error=False ): """ Shrinks bytes read from `value` to `size`. `value` needs to implement tell/seek, so files need to be opened in binary mode. Returns unicode text with invalid characters replaced. """ rval = b"" join_by = smart_str(join_by) if get_file_size(value) > size: start = value.tell() len_join_by = len(join_by) min_size = len_join_by + 2 if size < min_size: if beginning_on_size_error: rval = value.read(size) value.seek(start) return rval elif end_on_size_error: value.seek(-size, 2) rval = value.read(size) value.seek(start) return rval raise ValueError( "With the provided join_by value (%s), the minimum size value is %i." % (join_by, min_size) ) left_index = right_index = int((size - len_join_by) / 2) if left_index + right_index + len_join_by < size: if left_larger: left_index += 1 else: right_index += 1 rval = value.read(left_index) + join_by value.seek(-right_index, 2) rval += value.read(right_index) else: while True: data = value.read(CHUNK_SIZE) if not data: break rval += data return unicodify(rval)
[docs]def shrink_and_unicodify(stream): stream = unicodify(stream, strip_null=True) or "" if len(stream) > DATABASE_MAX_STRING_SIZE: stream = shrink_string_by_size( stream, DATABASE_MAX_STRING_SIZE, join_by="\n..\n", left_larger=True, beginning_on_size_error=True ) return stream
[docs]def shrink_string_by_size( value, size, join_by="..", left_larger=True, beginning_on_size_error=False, end_on_size_error=False ): if len(value) > size: len_join_by = len(join_by) min_size = len_join_by + 2 if size < min_size: if beginning_on_size_error: return value[:size] elif end_on_size_error: return value[-size:] raise ValueError( "With the provided join_by value (%s), the minimum size value is %i." % (join_by, min_size) ) left_index = right_index = int((size - len_join_by) / 2) if left_index + right_index + len_join_by < size: if left_larger: left_index += 1 else: right_index += 1 value = f"{value[:left_index]}{join_by}{value[-right_index:]}" return value
[docs]def pretty_print_time_interval(time=False, precise=False, utc=False): """ Get a datetime object or a int() Epoch timestamp and return a pretty string like 'an hour ago', 'Yesterday', '3 months ago', 'just now', etc credit: http://stackoverflow.com/questions/1551382/user-friendly-time-format-in-python """ if utc: now = datetime.utcnow() else: now = datetime.now() if isinstance(time, (int, float)): diff = now - datetime.fromtimestamp(time) elif isinstance(time, datetime): diff = now - time elif isinstance(time, str): try: time = datetime.strptime(time, "%Y-%m-%dT%H:%M:%S.%f") except ValueError: # MySQL may not support microseconds precision time = datetime.strptime(time, "%Y-%m-%dT%H:%M:%S") diff = now - time else: diff = now - now second_diff = diff.seconds day_diff = diff.days if day_diff < 0: return "" if precise: if day_diff == 0: if second_diff < 10: return "just now" if second_diff < 60: return str(second_diff) + " seconds ago" if second_diff < 120: return "a minute ago" if second_diff < 3600: return str(second_diff / 60) + " minutes ago" if second_diff < 7200: return "an hour ago" if second_diff < 86400: return str(second_diff / 3600) + " hours ago" if day_diff == 1: return "yesterday" if day_diff < 7: return str(day_diff) + " days ago" if day_diff < 31: return str(day_diff / 7) + " weeks ago" if day_diff < 365: return str(day_diff / 30) + " months ago" return str(day_diff / 365) + " years ago" else: if day_diff == 0: return "today" if day_diff == 1: return "yesterday" if day_diff < 7: return "less than a week" if day_diff < 31: return "less than a month" if day_diff < 365: return "less than a year" return "a few years ago"
# characters that are valid valid_chars = set(string.ascii_letters + string.digits + " -=_.()/+*^,:?!") # characters that are allowed but need to be escaped mapped_chars = { ">": "__gt__", "<": "__lt__", "'": "__sq__", '"': "__dq__", "[": "__ob__", "]": "__cb__", "{": "__oc__", "}": "__cc__", "@": "__at__", "\n": "__cn__", "\r": "__cr__", "\t": "__tc__", "#": "__pd__", }
[docs]def restore_text(text, character_map=mapped_chars): """Restores sanitized text""" if not text: return text for key, value in character_map.items(): text = text.replace(value, key) return text
[docs]def sanitize_text(text, valid_characters=valid_chars, character_map=mapped_chars, invalid_character="X"): """ Restricts the characters that are allowed in text; accepts both strings and lists of strings; non-string entities will be cast to strings. """ if isinstance(text, list): return [ sanitize_text( x, valid_characters=valid_characters, character_map=character_map, invalid_character=invalid_character ) for x in text ] if not isinstance(text, str): text = smart_str(text) return _sanitize_text_helper(text, valid_characters=valid_characters, character_map=character_map)
def _sanitize_text_helper(text, valid_characters=valid_chars, character_map=mapped_chars, invalid_character="X"): """Restricts the characters that are allowed in a string""" out = [] for c in text: if c in valid_characters: out.append(c) elif c in character_map: out.append(character_map[c]) else: out.append(invalid_character) # makes debugging easier return "".join(out)
[docs]def sanitize_lists_to_string(values, valid_characters=valid_chars, character_map=mapped_chars, invalid_character="X"): if isinstance(values, list): rval = [] for value in values: rval.append( sanitize_lists_to_string( value, valid_characters=valid_characters, character_map=character_map, invalid_character=invalid_character, ) ) values = ",".join(rval) else: values = sanitize_text( values, valid_characters=valid_characters, character_map=character_map, invalid_character=invalid_character ) return values
[docs]def sanitize_param(value, valid_characters=valid_chars, character_map=mapped_chars, invalid_character="X"): """Clean incoming parameters (strings or lists)""" if isinstance(value, str): return sanitize_text( value, valid_characters=valid_characters, character_map=character_map, invalid_character=invalid_character ) elif isinstance(value, list): return [ sanitize_text( x, valid_characters=valid_characters, character_map=character_map, invalid_character=invalid_character ) for x in value ] else: raise Exception(f"Unknown parameter type ({type(value)})")
valid_filename_chars = set(string.ascii_letters + string.digits + "_.") invalid_filenames = ["", ".", ".."]
[docs]def sanitize_for_filename(text, default=None): """ Restricts the characters that are allowed in a filename portion; Returns default value or a unique id string if result is not a valid name. Method is overly aggressive to minimize possible complications, but a maximum length is not considered. """ out = [] for c in text: if c in valid_filename_chars: out.append(c) else: out.append("_") out = "".join(out) if out in invalid_filenames: if default is None: return sanitize_for_filename(str(unique_id())) return default return out
[docs]def find_instance_nested(item, instances): """ Recursively find instances from lists, dicts, tuples. `instances` should be a tuple of valid instances. Returns a dictionary, where keys are the deepest key at which an instance has been found, and the value is the matched instance. """ matches = {} def visit(path, key, value): if isinstance(value, instances): if key not in matches: matches[key] = value return key, value def enter(path, key, value): if isinstance(value, instances): return None, False return default_enter(path, key, value) remap(item, visit, reraise_visit=False, enter=enter) return matches
[docs]def mask_password_from_url(url): """ Masks out passwords from connection urls like the database connection in galaxy.ini >>> mask_password_from_url( 'sqlite+postgresql://user:password@localhost/' ) 'sqlite+postgresql://user:********@localhost/' >>> mask_password_from_url( 'amqp://user:amqp@localhost' ) 'amqp://user:********@localhost' >>> mask_password_from_url( 'amqp://localhost') 'amqp://localhost' """ split = urlsplit(url) if split.password: if url.count(split.password) == 1: url = url.replace(split.password, "********") else: # This can manipulate the input other than just masking password, # so the previous string replace method is preferred when the # password doesn't appear twice in the url split = split._replace( netloc=split.netloc.replace(f"{split.username}:{split.password}", f"{split.username}:********") ) url = urlunsplit(split) return url
[docs]def ready_name_for_url(raw_name): """General method to convert a string (i.e. object name) to a URL-ready slug. >>> ready_name_for_url( "My Cool Object" ) 'My-Cool-Object' >>> ready_name_for_url( "!My Cool Object!" ) 'My-Cool-Object' >>> ready_name_for_url( "Hello₩◎ґʟⅾ" ) 'Hello' """ # Replace whitespace with '-' slug_base = re.sub(r"\s+", "-", raw_name) # Remove all non-alphanumeric characters. slug_base = re.sub(r"[^a-zA-Z0-9\-]", "", slug_base) # Remove trailing '-'. if slug_base.endswith("-"): slug_base = slug_base[:-1] return slug_base
[docs]def which(file: str) -> Optional[str]: # http://stackoverflow.com/questions/5226958/which-equivalent-function-in-python for path in os.environ["PATH"].split(":"): if os.path.exists(path + "/" + file): return path + "/" + file return None
[docs]def in_directory(file, directory, local_path_module=os.path): """ Return true, if the common prefix of both is equal to directory e.g. /a/b/c/d.rst and directory is /a/b, the common prefix is /a/b. This function isn't used exclusively for security checks, but if it is used for such checks it is assumed that ``directory`` is a "trusted" path - supplied by Galaxy or by the admin and ``file`` is something generated by a tool, configuration, external web server, or user supplied input. local_path_module is used by Pulsar to check Windows paths while running on a POSIX-like system. """ if local_path_module != os.path: _safe_contains = importlib.import_module(f"galaxy.util.path.{local_path_module.__name__}").safe_contains else: directory = os.path.realpath(directory) _safe_contains = safe_contains return _safe_contains(directory, file)
[docs]def merge_sorted_iterables(operator, *iterables): """ >>> operator = lambda x: x >>> list( merge_sorted_iterables( operator, [1,2,3], [4,5] ) ) [1, 2, 3, 4, 5] >>> list( merge_sorted_iterables( operator, [4, 5], [1,2,3] ) ) [1, 2, 3, 4, 5] >>> list( merge_sorted_iterables( operator, [1, 4, 5], [2], [3] ) ) [1, 2, 3, 4, 5] """ first_iterable = iterables[0] if len(iterables) == 1: yield from first_iterable else: yield from __merge_two_sorted_iterables( operator, iter(first_iterable), merge_sorted_iterables(operator, *iterables[1:]) )
def __merge_two_sorted_iterables(operator, iterable1, iterable2): unset = object() continue_merge = True next_1 = unset next_2 = unset while continue_merge: try: if next_1 is unset: next_1 = next(iterable1) if next_2 is unset: next_2 = next(iterable2) if operator(next_2) < operator(next_1): yield next_2 next_2 = unset else: yield next_1 next_1 = unset except StopIteration: continue_merge = False if next_1 is not unset: yield next_1 if next_2 is not unset: yield next_2 yield from iterable1 yield from iterable2
[docs]class Params: """ Stores and 'sanitizes' parameters. Alphanumeric characters and the non-alphanumeric ones that are deemed safe are let to pass through (see L{valid_chars}). Some non-safe characters are escaped to safe forms for example C{>} becomes C{__lt__} (see L{mapped_chars}). All other characters are replaced with C{X}. Operates on string or list values only (HTTP parameters). >>> values = { 'status':'on', 'symbols':[ 'alpha', '<>', '$rm&#!' ] } >>> par = Params(values) >>> par.status 'on' >>> par.value == None # missing attributes return None True >>> par.get('price', 0) 0 >>> par.symbols # replaces unknown symbols with X ['alpha', '__lt____gt__', 'XrmX__pd__!'] >>> sorted(par.flatten()) # flattening to a list [('status', 'on'), ('symbols', 'XrmX__pd__!'), ('symbols', '__lt____gt__'), ('symbols', 'alpha')] """ # is NEVER_SANITIZE required now that sanitizing for tool parameters can be controlled on a per parameter basis and occurs via InputValueWrappers? NEVER_SANITIZE = ["file_data", "url_paste", "URL", "filesystem_paths"]
[docs] def __init__(self, params, sanitize=True): if sanitize: for key, value in params.items(): # sanitize check both ungrouped and grouped parameters by # name. Anything relying on NEVER_SANITIZE should be # changed to not require this and NEVER_SANITIZE should be # removed. if ( value is not None and key not in self.NEVER_SANITIZE and True not in [key.endswith(f"|{nonsanitize_parameter}") for nonsanitize_parameter in self.NEVER_SANITIZE] ): self.__dict__[key] = sanitize_param(value) else: self.__dict__[key] = value else: self.__dict__.update(params)
[docs] def flatten(self): """ Creates a tuple list from a dict with a tuple/value pair for every value that is a list """ flat = [] for key, value in self.__dict__.items(): if isinstance(value, list): for v in value: flat.append((key, v)) else: flat.append((key, value)) return flat
def __getattr__(self, name): """This is here to ensure that we get None for non existing parameters""" return None
[docs] def get(self, key, default): return self.__dict__.get(key, default)
def __str__(self): return f"{self.__dict__}" def __len__(self): return len(self.__dict__) def __iter__(self): return iter(self.__dict__)
[docs] def update(self, values): self.__dict__.update(values)
[docs]def xml_text(root, name=None): """Returns the text inside an element""" if name is not None: # Try attribute first val = root.get(name) if val: return val # Then try as element elem = root.find(name) else: elem = root if elem is not None and elem.text: text = "".join(elem.text.splitlines()) return text.strip() # No luck, return empty string return ""
[docs]def parse_resource_parameters(resource_param_file): """Code shared between jobs and workflows for reading resource parameter configuration files. TODO: Allow YAML in addition to XML. """ resource_parameters = {} if os.path.exists(resource_param_file): resource_definitions = parse_xml(resource_param_file) resource_definitions_root = resource_definitions.getroot() for parameter_elem in resource_definitions_root.findall("param"): name = parameter_elem.get("name") resource_parameters[name] = etree.tostring(parameter_elem, encoding="unicode") return resource_parameters
# asbool implementation pulled from PasteDeploy truthy = frozenset({"true", "yes", "on", "y", "t", "1"}) falsy = frozenset({"false", "no", "off", "n", "f", "0"})
[docs]def asbool(obj): if isinstance(obj, str): obj = obj.strip().lower() if obj in truthy: return True elif obj in falsy: return False else: raise ValueError(f"String is not true/false: {obj!r}") return bool(obj)
[docs]def string_as_bool(string: Any) -> bool: if str(string).lower() in ("true", "yes", "on", "1"): return True else: return False
[docs]def string_as_bool_or_none(string): """ Returns True, None or False based on the argument: True if passed True, 'True', 'Yes', or 'On' None if passed None or 'None' False otherwise Note: string comparison is case-insensitive so lowecase versions of those function equivalently. """ string = str(string).lower() if string in ("true", "yes", "on"): return True elif string in ["none", "null"]: return None else: return False
ItemType = TypeVar("ItemType") @overload def listify(item: Union[None, Literal[False]], do_strip: bool = False) -> List: ... @overload def listify(item: str, do_strip: bool = False) -> List[str]: ... @overload def listify(item: Union[List[ItemType], Tuple[ItemType, ...]], do_strip: bool = False) -> List[ItemType]: ... # Unfortunately we cannot use ItemType .. -> List[ItemType] in the next overload # because then that would also match Union types. @overload def listify(item: Any, do_strip: bool = False) -> List: ...
[docs]def listify(item: Any, do_strip: bool = False) -> List: """ Make a single item a single item list. If *item* is a string, it is split on comma (``,``) characters to produce the list. Optionally, if *do_strip* is true, any extra whitespace around the split items is stripped. If *item* is a list it is returned unchanged. If *item* is a tuple, it is converted to a list and returned. If *item* evaluates to False, an empty list is returned. :type item: object :param item: object to make a list from :type do_strip: bool :param do_strip: strip whitespaces from around split items, if set to ``True`` :rtype: list :returns: The input as a list """ if not item: return [] elif isinstance(item, (list, tuple)): return list(item) elif isinstance(item, str) and item.count(","): if do_strip: return [token.strip() for token in item.split(",")] else: return item.split(",") else: return [item]
[docs]def commaify(amount): orig = amount new = re.sub(r"^(-?\d+)(\d{3})", r"\g<1>,\g<2>", amount) if orig == new: return new else: return commaify(new)
@overload def unicodify( # type: ignore[overload-overlap] value: Literal[None], encoding: str = DEFAULT_ENCODING, error: str = "replace", strip_null: bool = False, log_exception: bool = True, ) -> None: ... @overload def unicodify( value: Any, encoding: str = DEFAULT_ENCODING, error: str = "replace", strip_null: bool = False, log_exception: bool = True, ) -> str: ...
[docs]def unicodify( value: Any, encoding: str = DEFAULT_ENCODING, error: str = "replace", strip_null: bool = False, log_exception: bool = True, ) -> Optional[str]: """ Returns a Unicode string or None. >>> assert unicodify(None) is None >>> assert unicodify('simple string') == 'simple string' >>> assert unicodify(3) == '3' >>> assert unicodify(bytearray([115, 116, 114, 196, 169, 195, 177, 103])) == 'strĩñg' >>> assert unicodify(Exception('strĩñg')) == 'strĩñg' >>> assert unicodify('cómplǐcḁtëd strĩñg') == 'cómplǐcḁtëd strĩñg' >>> s = 'cómplǐcḁtëd strĩñg'; assert unicodify(s) == s >>> s = 'lâtín strìñg'; assert unicodify(s.encode('latin-1'), 'latin-1') == s >>> s = 'lâtín strìñg'; assert unicodify(s.encode('latin-1')) == 'l\ufffdt\ufffdn str\ufffd\ufffdg' >>> s = 'lâtín strìñg'; assert unicodify(s.encode('latin-1'), error='ignore') == 'ltn strg' """ if value is None: return value try: if isinstance(value, bytearray): value = bytes(value) elif not isinstance(value, (str, bytes)): value = str(value) # Now value is an instance of bytes or str if not isinstance(value, str): value = str(value, encoding, error) except Exception as e: if log_exception: msg = f"Value '{repr(value)}' could not be coerced to Unicode: {type(e).__name__}('{e}')" log.exception(msg) raise if strip_null: return value.replace("\0", "") return value
[docs]def filesystem_safe_string( s, max_len=255, truncation_chars="..", strip_leading_dot=True, invalid_chars=("/",), replacement_char="_" ): """ Strip unicode null chars, truncate at 255 characters. Optionally replace additional ``invalid_chars`` with `replacement_char` . Defaults are probably only safe on linux / osx. Needs further escaping if used in shell commands """ sanitized_string = unicodify(s, strip_null=True) if strip_leading_dot: sanitized_string = sanitized_string.lstrip(".") for invalid_char in invalid_chars: sanitized_string = sanitized_string.replace(invalid_char, replacement_char) if len(sanitized_string) > max_len: sanitized_string = sanitized_string[: max_len - len(truncation_chars)] sanitized_string = f"{sanitized_string}{truncation_chars}" return sanitized_string
[docs]def smart_str(s, encoding=DEFAULT_ENCODING, strings_only=False, errors="strict"): """ Returns a bytestring version of 's', encoded as specified in 'encoding'. If strings_only is True, don't convert (some) non-string-like objects. Adapted from an older, simpler version of django.utils.encoding.smart_str. >>> assert smart_str(None) == b'None' >>> assert smart_str(None, strings_only=True) is None >>> assert smart_str(3) == b'3' >>> assert smart_str(3, strings_only=True) == 3 >>> s = b'a bytes string'; assert smart_str(s) == s >>> s = bytearray(b'a bytes string'); assert smart_str(s) == s >>> assert smart_str('a simple unicode string') == b'a simple unicode string' >>> assert smart_str('à strange ünicode ڃtring') == b'\\xc3\\xa0 strange \\xc3\\xbcnicode \\xda\\x83tring' >>> assert smart_str(b'\\xc3\\xa0n \\xc3\\xabncoded utf-8 string', encoding='latin-1') == b'\\xe0n \\xebncoded utf-8 string' >>> assert smart_str(bytearray(b'\\xc3\\xa0n \\xc3\\xabncoded utf-8 string'), encoding='latin-1') == b'\\xe0n \\xebncoded utf-8 string' """ if strings_only and isinstance(s, (type(None), int)): return s if not isinstance(s, (str, bytes, bytearray)): s = str(s) # Now s is an instance of str, bytes or bytearray if not isinstance(s, (bytes, bytearray)): return s.encode(encoding, errors) elif s and encoding != DEFAULT_ENCODING: return s.decode(DEFAULT_ENCODING, errors).encode(encoding, errors) else: return s
[docs]def strip_control_characters(s): """Strip unicode control characters from a string.""" return "".join(c for c in unicodify(s) if unicodedata.category(c) != "Cc")
[docs]def object_to_string(obj): return binascii.hexlify(obj)
[docs]def string_to_object(s): return binascii.unhexlify(s)
[docs]def clean_multiline_string(multiline_string, sep="\n"): """ Dedent, split, remove first and last empty lines, rejoin. """ multiline_string = textwrap.dedent(multiline_string) string_list = multiline_string.split(sep) if not string_list[0]: string_list = string_list[1:] if not string_list[-1]: string_list = string_list[:-1] return "\n".join(string_list) + "\n"
[docs]class ParamsWithSpecs(collections.defaultdict): """ """
[docs] def __init__(self, specs=None, params=None): self.specs = specs or {} self.params = params or {} for name, value in self.params.items(): if name not in self.specs: self._param_unknown_error(name) if "map" in self.specs[name]: try: self.params[name] = self.specs[name]["map"](value) except Exception: self._param_map_error(name, value) if "valid" in self.specs[name]: if not self.specs[name]["valid"](value): self._param_vaildation_error(name, value) self.update(self.params)
def __missing__(self, name): return self.specs[name]["default"] def __getattr__(self, name): return self[name] def _param_unknown_error(self, name): raise NotImplementedError() def _param_map_error(self, name, value): raise NotImplementedError() def _param_vaildation_error(self, name, value): raise NotImplementedError()
[docs]def compare_urls(url1, url2, compare_scheme=True, compare_hostname=True, compare_path=True): url1 = urlparse(url1) url2 = urlparse(url2) if compare_scheme and url1.scheme and url2.scheme and url1.scheme != url2.scheme: return False if compare_hostname and url1.hostname and url2.hostname and url1.hostname != url2.hostname: return False if compare_path and url1.path and url2.path and url1.path != url2.path: return False return True
[docs]def read_build_sites(filename, check_builds=True): """read db names to ucsc mappings from file, this file should probably be merged with the one above""" build_sites = [] try: for line in open(filename): try: if line[0:1] == "#": continue fields = line.replace("\r", "").replace("\n", "").split("\t") site_name = fields[0] site = fields[1] if check_builds: site_builds = fields[2].split(",") site_dict = {"name": site_name, "url": site, "builds": site_builds} else: site_dict = {"name": site_name, "url": site} build_sites.append(site_dict) except Exception: continue except Exception: log.error("ERROR: Unable to read builds for site file %s", filename) return build_sites
[docs]def stringify_dictionary_keys(in_dict): # returns a new dictionary # changes unicode keys into strings, only works on top level (does not recurse) # unicode keys are not valid for expansion into keyword arguments on method calls out_dict = {} for key, value in in_dict.items(): out_dict[str(key)] = value return out_dict
[docs]def mkstemp_ln(src, prefix="mkstemp_ln_"): """ From tempfile._mkstemp_inner, generate a hard link in the same dir with a random name. Created so we can persist the underlying file of a NamedTemporaryFile upon its closure. """ dir = os.path.dirname(src) names = tempfile._get_candidate_names() for _ in range(tempfile.TMP_MAX): name = next(names) file = os.path.join(dir, prefix + name) try: os.link(src, file) return os.path.abspath(file) except OSError as e: if e.errno == errno.EEXIST: continue # try again raise raise OSError(errno.EEXIST, "No usable temporary file name found")
[docs]def umask_fix_perms(path, umask, unmasked_perms, gid=None): """ umask-friendly permissions fixing """ perms = unmasked_perms & ~umask try: st = os.stat(path) except OSError: log.exception("Unable to set permissions or group on %s", path) return # fix modes if stat.S_IMODE(st.st_mode) != perms: try: os.chmod(path, perms) except Exception as e: log.warning( "Unable to honor umask (%s) for %s, tried to set: %s but mode remains %s, error was: %s", oct(umask), path, oct(perms), oct(stat.S_IMODE(st.st_mode)), e, ) # fix group if gid is not None and st.st_gid != gid: try: os.chown(path, -1, gid) except Exception as e: try: desired_group = grp.getgrgid(gid) current_group = grp.getgrgid(st.st_gid) except Exception: desired_group = gid current_group = st.st_gid log.warning( "Unable to honor primary group (%s) for %s, group remains %s, error was: %s", desired_group, path, current_group, e, )
[docs]def docstring_trim(docstring): """Trimming python doc strings. Taken from: http://www.python.org/dev/peps/pep-0257/""" if not docstring: return "" # Convert tabs to spaces (following the normal Python rules) # and split into a list of lines: lines = docstring.expandtabs().splitlines() # Determine minimum indentation (first line doesn't count): indent = sys.maxsize for line in lines[1:]: stripped = line.lstrip() if stripped: indent = min(indent, len(line) - len(stripped)) # Remove indentation (first line is special): trimmed = [lines[0].strip()] if indent < sys.maxsize: for line in lines[1:]: trimmed.append(line[indent:].rstrip()) # Strip off trailing and leading blank lines: while trimmed and not trimmed[-1]: trimmed.pop() while trimmed and not trimmed[0]: trimmed.pop(0) # Return a single string: return "\n".join(trimmed)
[docs]def metric_prefix(number: Union[int, float], base: int) -> Tuple[float, str]: """ >>> metric_prefix(100, 1000) (100.0, '') >>> metric_prefix(999, 1000) (999.0, '') >>> metric_prefix(1000, 1000) (1.0, 'K') >>> metric_prefix(1001, 1000) (1.001, 'K') >>> metric_prefix(1000000, 1000) (1.0, 'M') >>> metric_prefix(1000**10, 1000) (1.0, 'Q') >>> metric_prefix(1000**11, 1000) (1000.0, 'Q') """ prefixes = ["", "K", "M", "G", "T", "P", "E", "Z", "Y", "R", "Q"] if number < 0: number = abs(number) sign = -1 else: sign = 1 for prefix in prefixes: if number < base: return sign * float(number), prefix number /= base else: return sign * float(number) * base, prefix
[docs]def shorten_with_metric_prefix(amount: int) -> str: """ >>> shorten_with_metric_prefix(23000) '23K' >>> shorten_with_metric_prefix(2300000) '2.3M' >>> shorten_with_metric_prefix(23000000) '23M' >>> shorten_with_metric_prefix(1) '1' >>> shorten_with_metric_prefix(0) '0' >>> shorten_with_metric_prefix(100) '100' >>> shorten_with_metric_prefix(-100) '-100' """ m, prefix = metric_prefix(amount, 1000) m_str = str(int(m)) if m.is_integer() else f"{m:.1f}" exp = f"{m_str}{prefix}" if len(exp) <= len(str(amount)): return exp else: return str(amount)
[docs]def nice_size(size: Union[float, int, str, Decimal]) -> str: """ Returns a readably formatted string with the size >>> nice_size(100) '100 bytes' >>> nice_size(10000) '9.8 KB' >>> nice_size(1000000) '976.6 KB' >>> nice_size(100000000) '95.4 MB' """ try: size = float(size) except ValueError: return "??? bytes" size, prefix = metric_prefix(size, 1024) if prefix == "": return "%d bytes" % size else: return f"{size:.1f} {prefix}B"
[docs]def size_to_bytes(size): """ Returns a number of bytes (as integer) if given a reasonably formatted string with the size >>> size_to_bytes('1024') 1024 >>> size_to_bytes('1.0') 1 >>> size_to_bytes('10 bytes') 10 >>> size_to_bytes('4k') 4096 >>> size_to_bytes('2.2 TB') 2418925581107 >>> size_to_bytes('.01 TB') 10995116277 >>> size_to_bytes('1.b') 1 >>> size_to_bytes('1.2E2k') 122880 """ # The following number regexp is based on https://stackoverflow.com/questions/385558/extract-float-double-value/385597#385597 size_re = re.compile(r"(?P<number>(\d+(\.\d*)?|\.\d+)(e[+-]?\d+)?)\s*(?P<multiple>[eptgmk]?(b|bytes?)?)?$") size_match = size_re.match(size.lower()) if size_match is None: raise ValueError(f"Could not parse string '{size}'") number = float(size_match.group("number")) multiple = size_match.group("multiple") if multiple == "" or multiple.startswith("b"): return int(number) elif multiple.startswith("k"): return int(number * 1024) elif multiple.startswith("m"): return int(number * 1024**2) elif multiple.startswith("g"): return int(number * 1024**3) elif multiple.startswith("t"): return int(number * 1024**4) elif multiple.startswith("p"): return int(number * 1024**5) elif multiple.startswith("e"): return int(number * 1024**6) else: raise ValueError(f"Unknown multiplier '{multiple}' in '{size}'")
[docs]def send_mail(frm, to, subject, body, config, html=None, reply_to=None): """ Sends an email. :type frm: str :param frm: from address :type to: str :param to: to address :type subject: str :param subject: Subject line :type body: str :param body: Body text (should be plain text) :type config: object :param config: Galaxy configuration object :type html: str :param html: Alternative HTML representation of the body content. If provided will convert the message to a MIMEMultipart. (Default None) :type reply_to: str :param reply_to: Reply-to address (Default None) """ smtp_server = config.smtp_server if smtp_server and isinstance(smtp_server, str) and smtp_server.startswith("mock_emails_to_path://"): path = config.smtp_server[len("mock_emails_to_path://") :] email_dict = { "from": frm, "to": to, "subject": subject, "body": body, "html": html, "reply_to": reply_to, } email_json = json.to_json_string(email_dict) with open(path, "w") as f: f.write(email_json) return to = listify(to) if html: msg = MIMEMultipart("alternative") else: msg = MIMEText(body, "plain", "utf-8") msg["To"] = ", ".join(to) msg["From"] = frm msg["Subject"] = subject if reply_to: msg["Reply-To"] = reply_to if config.smtp_server is None: log.error("Mail is not configured for this Galaxy instance.") log.info(msg) return if html: mp_text = MIMEText(body, "plain", "utf-8") mp_html = MIMEText(html, "html", "utf-8") msg.attach(mp_text) msg.attach(mp_html) smtp_ssl = asbool(getattr(config, "smtp_ssl", False)) if smtp_ssl: s = smtplib.SMTP_SSL(config.smtp_server) else: s = smtplib.SMTP(config.smtp_server) if not smtp_ssl: try: s.starttls() log.debug("Initiated SSL/TLS connection to SMTP server: %s", config.smtp_server) except RuntimeError as e: log.warning("SSL/TLS support is not available to your Python interpreter: %s", unicodify(e)) except smtplib.SMTPHeloError as e: log.error("The server didn't reply properly to the HELO greeting: %s", unicodify(e)) s.close() raise except smtplib.SMTPException as e: log.warning("The server does not support the STARTTLS extension: %s", unicodify(e)) if config.smtp_username and config.smtp_password: try: s.login(config.smtp_username, config.smtp_password) except smtplib.SMTPHeloError as e: log.error("The server didn't reply properly to the HELO greeting: %s", unicodify(e)) s.close() raise except smtplib.SMTPAuthenticationError as e: log.error("The server didn't accept the username/password combination: %s", unicodify(e)) s.close() raise except smtplib.SMTPException as e: log.error("No suitable authentication method was found: %s", unicodify(e)) s.close() raise s.sendmail(frm, to, msg.as_string()) s.quit()
[docs]def move_merge(source, target): # when using shutil and moving a directory, if the target exists, # then the directory is placed inside of it # if the target doesn't exist, then the target is made into the directory # this makes it so that the target is always the target, and if it exists, # the source contents are moved into the target if os.path.isdir(source) and os.path.exists(target) and os.path.isdir(target): for name in os.listdir(source): move_merge(os.path.join(source, name), os.path.join(target, name)) else: return shutil.move(source, target)
[docs]def safe_str_cmp(a, b): """safely compare two strings in a timing-attack-resistant manner""" if len(a) != len(b): return False rv = 0 for x, y in zip(a, b): rv |= ord(x) ^ ord(y) return rv == 0
# Don't use this directly, prefer method version that "works" with packaged Galaxy. galaxy_root_path = Path(__file__).parent.parent.parent.parent
[docs]def galaxy_directory() -> str: if in_packages(): # This will work only when running pytest from <galaxy_root>/packages/<package_name>/ cwd = Path.cwd() path = cwd.parent.parent else: path = galaxy_root_path return os.path.abspath(path)
[docs]def in_packages() -> bool: galaxy_lib_path = Path(__file__).parent.parent.parent return galaxy_lib_path.name != "lib"
[docs]def config_directories_from_setting(directories_setting, galaxy_root=galaxy_root_path): """ Parse the ``directories_setting`` into a list of relative or absolute filesystem paths that will be searched to discover plugins. :type galaxy_root: string :param galaxy_root: the root path of this galaxy installation :type directories_setting: string (default: None) :param directories_setting: the filesystem path (or paths) to search for plugins. Can be CSV string of paths. Will be treated as absolute if a path starts with '/', relative otherwise. :rtype: list of strings :returns: list of filesystem paths """ directories = [] if not directories_setting: return directories for directory in listify(directories_setting): directory = directory.strip() if not directory.startswith("/"): directory = os.path.join(galaxy_root, directory) if not os.path.exists(directory): log.warning("directory not found: %s", directory) continue directories.append(directory) return directories
[docs]def parse_int(value, min_val=None, max_val=None, default=None, allow_none=False): try: value = int(value) if min_val is not None and value < min_val: return min_val if max_val is not None and value > max_val: return max_val return value except ValueError: if allow_none: if default is None or value == "None": return None if default: return default else: raise
[docs]def parse_non_hex_float(s): r""" Parse string `s` into a float but throw a `ValueError` if the string is in the otherwise acceptable format `\d+e\d+` (e.g. 40000000000000e5.) This can be passed into `json.loads` to prevent a hex string in the above format from being incorrectly parsed as a float in scientific notation. >>> parse_non_hex_float( '123.4' ) 123.4 >>> parse_non_hex_float( '2.45e+3' ) 2450.0 >>> parse_non_hex_float( '2.45e-3' ) 0.00245 >>> parse_non_hex_float( '40000000000000e5' ) Traceback (most recent call last): ... ValueError: could not convert string to float: 40000000000000e5 """ f = float(s) # successfully parsed as float if here - check for format in original string if "e" in s and not ("+" in s or "-" in s): raise ValueError("could not convert string to float: " + s) return f
[docs]def build_url(base_url, port=80, scheme="http", pathspec=None, params=None, doseq=False): if params is None: params = {} if pathspec is None: pathspec = [] parsed_url = urlparse(base_url) if scheme != "http": parsed_url.scheme = scheme assert parsed_url.scheme in ("http", "https", "ftp"), f"Invalid URL scheme: {parsed_url.scheme}" if port != 80: url = "%s://%s:%d/%s" % (parsed_url.scheme, parsed_url.netloc.rstrip("/"), int(port), parsed_url.path) else: url = f"{parsed_url.scheme}://{parsed_url.netloc.rstrip('/')}/{parsed_url.path.lstrip('/')}" if len(pathspec) > 0: url = f"{url.rstrip('/')}/{'/'.join(pathspec)}" if parsed_url.query: for query_parameter in parsed_url.query.split("&"): key, value = query_parameter.split("=") params[key] = value if params: url += f"?{urlencode(params, doseq=doseq)}" return url
[docs]def url_get(base_url, auth=None, pathspec=None, params=None, max_retries=5, backoff_factor=1): """Make contact with the uri provided and return any contents.""" full_url = build_url(base_url, pathspec=pathspec, params=params) s = requests.Session() retries = Retry(total=max_retries, backoff_factor=backoff_factor, status_forcelist=[429]) s.mount(base_url, HTTPAdapter(max_retries=retries)) response = s.get(full_url, auth=auth) response.raise_for_status() return response.text
[docs]def is_url(uri, allow_list=None): """ Check if uri is (most likely) an URL, more precisely the function checks if uri starts with a scheme from the allow list (defaults to "http://", "https://", "ftp://") >>> is_url('https://zenodo.org/record/4104428/files/UCSC-hg38-chr22-Coding-Exons.bed') True >>> is_url('file:///some/path') False >>> is_url('/some/path') False """ if allow_list is None: allow_list = ("http://", "https://", "ftp://") return any(uri.startswith(scheme) for scheme in allow_list)
[docs]def check_github_api_response_rate_limit(response): if response.status_code == 403 and "API rate limit exceeded" in response.json()["message"]: # It can take tens of minutes before the rate limit window resets message = "GitHub API rate limit exceeded." rate_limit_reset_UTC_timestamp = response.headers.get("X-RateLimit-Reset") if rate_limit_reset_UTC_timestamp: rate_limit_reset_datetime = datetime.fromtimestamp(int(rate_limit_reset_UTC_timestamp), tz=timezone.utc) message += f" The rate limit window will reset at {rate_limit_reset_datetime.isoformat()}." raise Exception(message)
[docs]def download_to_file(url, dest_file_path, timeout=30, chunk_size=2**20): """Download a URL to a file in chunks.""" with requests.get(url, timeout=timeout, stream=True) as r, open(dest_file_path, "wb") as f: for chunk in r.iter_content(chunk_size): if chunk: f.write(chunk)
[docs]def stream_to_open_named_file( stream, fd, filename, source_encoding=None, source_error="strict", target_encoding=None, target_error="strict" ): """Writes a stream to the provided file descriptor, returns the file name. Closes file descriptor""" # signature and behavor is somewhat odd, due to backwards compatibility, but this can/should be done better CHUNK_SIZE = 1048576 try: codecs.lookup(target_encoding) except Exception: target_encoding = DEFAULT_ENCODING # utf-8 use_source_encoding = source_encoding is not None try: while True: chunk = stream.read(CHUNK_SIZE) if not chunk: break if use_source_encoding: # If a source encoding is given we use it to convert to the target encoding try: if not isinstance(chunk, str): chunk = chunk.decode(source_encoding, source_error) os.write(fd, chunk.encode(target_encoding, target_error)) except UnicodeDecodeError: use_source_encoding = False os.write(fd, chunk) else: # Compressed files must be encoded after they are uncompressed in the upload utility, # while binary files should not be encoded at all. if isinstance(chunk, str): chunk = chunk.encode(target_encoding, target_error) os.write(fd, chunk) finally: os.close(fd) return filename
[docs]class classproperty:
[docs] def __init__(self, f): self.f = f
def __get__(self, obj, owner): return self.f(owner)
[docs]class ExecutionTimer:
[docs] def __init__(self): self.begin = time.time()
def __str__(self): return f"({self.elapsed * 1000:0.3f} ms)" @property def elapsed(self): return time.time() - self.begin
[docs]class StructuredExecutionTimer:
[docs] def __init__(self, timer_id, template, **tags): self.begin = time.time() self.timer_id = timer_id self.template = template self.tags = tags
def __str__(self): return self.to_str()
[docs] def to_str(self, **kwd): if kwd: message = string.Template(self.template).safe_substitute(kwd) else: message = self.template log_message = message + f" ({self.elapsed * 1000:0.3f} ms)" return log_message
@property def elapsed(self): return time.time() - self.begin
[docs]def enum_values(enum_class): """ Return a list of member values of enumeration enum_class. Values are in member definition order. """ return [value.value for value in enum_class.__members__.values()]
[docs]def hex_to_lowercase_alphanum(hex_string: str) -> str: """ Convert a hexadecimal string encoding into a lowercase 36-base alphanumeric string using the characters a-z and 0-9 """ import numpy as np return np.base_repr(int(hex_string, 16), 36).lower()
[docs]def lowercase_alphanum_to_hex(lowercase_alphanum: str) -> str: """ Convert a lowercase 36-base alphanumeric string encoding using the characters a-z and 0-9 to a hexadecimal string """ import numpy as np return np.base_repr(int(lowercase_alphanum, 36), 16).lower()