Spaces:
Sleeping
Sleeping
File size: 2,842 Bytes
04ffec9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
import os
from typing import List, Callable, Tuple
import numpy as np
import hashlib
import re
MODULE_PATH = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
BASE_PATH = os.path.dirname(MODULE_PATH)
# Adapted from argparse.Namespace
class Context(dict):
__getattr__ = dict.get
__setattr__ = dict.__setitem__
__delattr__ = dict.__delitem__
def __init__(self, **kwargs):
for name in kwargs:
setattr(self, name, kwargs[name])
def __eq__(self, other):
if not isinstance(other, Context):
return NotImplemented
return vars(self) == vars(other)
def __contains__(self, key):
return key in self.keys()
def __repr__(self):
type_name = type(self).__name__
arg_strings = []
star_args = {}
for arg in self._get_args():
arg_strings.append(repr(arg))
for name, value in self._get_kwargs():
if name.isidentifier():
arg_strings.append('%s=%r' % (name, value))
else:
star_args[name] = value
if star_args:
arg_strings.append('**%s' % repr(star_args))
return '%s(%s)' % (type_name, ', '.join(arg_strings))
def _get_kwargs(self):
return list(self.items())
def _get_args(self):
return []
def repeating_sequence(s: str):
"""Extracts repeating sequence from string. Example: 'abcabca' -> 'abc'."""
for i in range(1, len(s) // 2 + 1):
seq = s[:i]
if seq * (len(s) // len(seq)) + seq[:len(s) % len(seq)] == s:
return seq
return s
def replace_prefix(s: str, old: str, new: str):
if s.startswith(old):
s = new + s[len(old):]
return s
def chunks(lst, n):
"""Yield successive n-sized chunks from lst."""
for i in range(0, len(lst), n):
yield lst[i:i + n]
def get_digest(file_path: str) -> str:
h = hashlib.sha256()
BUF_SIZE = 65536
with open(file_path, 'rb') as file:
while True:
# Reading is buffered, so we can read smaller chunks.
chunk = file.read(BUF_SIZE)
if not chunk:
break
h.update(chunk)
return h.hexdigest()
def get_filename_from_url(url: str, default: str = '') -> str:
m = re.search(r'/([^/?]+)[^/]*$', url)
if m:
return m.group(1)
return default
def prompt_yes_no(query: str, default: bool = None) -> bool:
s = '%s (%s/%s): ' % (query, 'Y' if default == True else 'y', 'N' if default == False else 'n')
while True:
inp = input(s).lower()
if inp in ('yes', 'y'):
return True
elif inp in ('no', 'n'):
return False
elif default != None:
return default
if inp:
print('Error: Please answer with "y" or "n"')
|