104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
import hashlib
|
|
import json
|
|
import pkgutil
|
|
from typing import Any, Dict, Iterator, List, MutableMapping, Optional, Tuple, TypeVar
|
|
|
|
|
|
def str2bool(v: Any) -> Optional[bool]:
|
|
if v in ("yes", True, "true", "True", "TRUE", "t", "1"):
|
|
return True
|
|
elif v in ("no", False, "false", "False", "FALSE", "f", "0"):
|
|
return False
|
|
return None
|
|
|
|
|
|
def load_resource(package: str, resource: str) -> Any:
|
|
"""
|
|
Open a file, and return the contents as JSON.
|
|
Usage:
|
|
load_resource(__name__, "resources/file.json")
|
|
"""
|
|
return json.loads(pkgutil.get_data(package, resource)) # type: ignore
|
|
|
|
|
|
def load_resource_as_str(package: str, resource: str) -> str:
|
|
return load_resource_as_bytes(package, resource).decode("utf-8") # type: ignore
|
|
|
|
|
|
def load_resource_as_bytes(package: str, resource: str) -> bytes:
|
|
return pkgutil.get_data(package, resource) # type: ignore
|
|
|
|
|
|
def merge_multiple_dicts(*args: Any) -> Dict[str, Any]:
|
|
result = {}
|
|
for d in args:
|
|
result.update(d)
|
|
return result
|
|
|
|
|
|
RESOURCE_TYPE = TypeVar("RESOURCE_TYPE")
|
|
|
|
|
|
def filter_resources(
|
|
resources: List[RESOURCE_TYPE],
|
|
filters: Any,
|
|
attr_pairs: Tuple[Tuple[str, ...], ...],
|
|
) -> List[RESOURCE_TYPE]:
|
|
"""
|
|
Used to filter resources. Usually in get and describe apis.
|
|
"""
|
|
result = resources.copy()
|
|
for resource in resources:
|
|
for attrs in attr_pairs:
|
|
values = filters.get(attrs[0]) or None
|
|
if values:
|
|
instance = getattr(resource, attrs[1])
|
|
if (len(attrs) <= 2 and instance not in values) or (
|
|
len(attrs) == 3 and instance.get(attrs[2]) not in values
|
|
):
|
|
result.remove(resource)
|
|
break
|
|
return result
|
|
|
|
|
|
def md5_hash(data: Any = None) -> Any:
|
|
"""
|
|
MD5-hashing for non-security usecases.
|
|
Required for Moto to work in FIPS-enabled systems
|
|
"""
|
|
args = (data,) if data else ()
|
|
try:
|
|
return hashlib.md5(*args, usedforsecurity=False) # type: ignore
|
|
except TypeError:
|
|
# The usedforsecurity-parameter is only available as of Python 3.9
|
|
return hashlib.md5(*args)
|
|
|
|
|
|
class LowercaseDict(MutableMapping[str, Any]):
|
|
"""A dictionary that lowercases all keys"""
|
|
|
|
def __init__(self, *args: Any, **kwargs: Any):
|
|
self.store: Dict[str, Any] = dict()
|
|
self.update(dict(*args, **kwargs)) # use the free update to set keys
|
|
|
|
def __getitem__(self, key: str) -> Any:
|
|
return self.store[self._keytransform(key)]
|
|
|
|
def __setitem__(self, key: str, value: Any) -> None:
|
|
self.store[self._keytransform(key)] = value
|
|
|
|
def __delitem__(self, key: str) -> None:
|
|
del self.store[self._keytransform(key)]
|
|
|
|
def __iter__(self) -> Iterator[Any]:
|
|
return iter(self.store)
|
|
|
|
def __len__(self) -> int:
|
|
return len(self.store)
|
|
|
|
def __repr__(self) -> str:
|
|
return str(self.store)
|
|
|
|
def _keytransform(self, key: str) -> str:
|
|
return key.lower()
|