You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
2.5 KiB
113 lines
2.5 KiB
from contextlib import contextmanager
|
|
from functools import wraps
|
|
from typing import Any, Dict, Mapping, Union
|
|
|
|
try:
|
|
import hiredis # noqa
|
|
|
|
# Only support Hiredis >= 1.0:
|
|
HIREDIS_AVAILABLE = not hiredis.__version__.startswith("0.")
|
|
HIREDIS_PACK_AVAILABLE = hasattr(hiredis, "pack_command")
|
|
except ImportError:
|
|
HIREDIS_AVAILABLE = False
|
|
HIREDIS_PACK_AVAILABLE = False
|
|
|
|
try:
|
|
import cryptography # noqa
|
|
|
|
CRYPTOGRAPHY_AVAILABLE = True
|
|
except ImportError:
|
|
CRYPTOGRAPHY_AVAILABLE = False
|
|
|
|
|
|
def from_url(url, **kwargs):
|
|
"""
|
|
Returns an active Redis client generated from the given database URL.
|
|
|
|
Will attempt to extract the database id from the path url fragment, if
|
|
none is provided.
|
|
"""
|
|
from redis.client import Redis
|
|
|
|
return Redis.from_url(url, **kwargs)
|
|
|
|
|
|
@contextmanager
|
|
def pipeline(redis_obj):
|
|
p = redis_obj.pipeline()
|
|
yield p
|
|
p.execute()
|
|
|
|
|
|
def str_if_bytes(value: Union[str, bytes]) -> str:
|
|
return (
|
|
value.decode("utf-8", errors="replace") if isinstance(value, bytes) else value
|
|
)
|
|
|
|
|
|
def safe_str(value):
|
|
return str(str_if_bytes(value))
|
|
|
|
|
|
def dict_merge(*dicts: Mapping[str, Any]) -> Dict[str, Any]:
|
|
"""
|
|
Merge all provided dicts into 1 dict.
|
|
*dicts : `dict`
|
|
dictionaries to merge
|
|
"""
|
|
merged = {}
|
|
|
|
for d in dicts:
|
|
merged.update(d)
|
|
|
|
return merged
|
|
|
|
|
|
def list_keys_to_dict(key_list, callback):
|
|
return dict.fromkeys(key_list, callback)
|
|
|
|
|
|
def merge_result(command, res):
|
|
"""
|
|
Merge all items in `res` into a list.
|
|
|
|
This command is used when sending a command to multiple nodes
|
|
and the result from each node should be merged into a single list.
|
|
|
|
res : 'dict'
|
|
"""
|
|
result = set()
|
|
|
|
for v in res.values():
|
|
for value in v:
|
|
result.add(value)
|
|
|
|
return list(result)
|
|
|
|
|
|
def warn_deprecated(name, reason="", version="", stacklevel=2):
|
|
import warnings
|
|
|
|
msg = f"Call to deprecated {name}."
|
|
if reason:
|
|
msg += f" ({reason})"
|
|
if version:
|
|
msg += f" -- Deprecated since version {version}."
|
|
warnings.warn(msg, category=DeprecationWarning, stacklevel=stacklevel)
|
|
|
|
|
|
def deprecated_function(reason="", version="", name=None):
|
|
"""
|
|
Decorator to mark a function as deprecated.
|
|
"""
|
|
|
|
def decorator(func):
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs):
|
|
warn_deprecated(name or func.__name__, reason, version, stacklevel=3)
|
|
return func(*args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
return decorator
|