diff options
Diffstat (limited to 'uploader/datautils.py')
| -rw-r--r-- | uploader/datautils.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/uploader/datautils.py b/uploader/datautils.py index b95a9e0..d132c42 100644 --- a/uploader/datautils.py +++ b/uploader/datautils.py @@ -1,6 +1,15 @@ """Generic data utilities: Rename module.""" import math +import json +import base64 from functools import reduce +from typing import Union, Sequence + +def enumerate_sequence(seq: Sequence[dict], start:int = 1) -> Sequence[dict]: + """Enumerate sequence beginning at 1""" + return tuple({**item, "sequence_number": seqno} + for seqno, item in enumerate(seq, start=start)) + def order_by_family(items: tuple[dict, ...], family_key: str = "Family", @@ -19,3 +28,23 @@ def order_by_family(items: tuple[dict, ...], return sorted(tuple(reduce(__order__, items, {}).items()), key=lambda item: item[0][0]) + + +def safe_int(val: Union[str, int, float]) -> int: + """ + Convert val into an integer: if val cannot be converted, return a zero. + """ + try: + return int(val) + except ValueError: + return 0 + + +def base64_encode_dict(dct: dict, **kwargs) -> bytes: + """Base64 encode a dictionary. Takes the same keywords as `json.dumps` function.""" + return base64.urlsafe_b64encode(json.dumps(dct, **kwargs).encode("utf-8")) + + +def base64_decode_to_dict(value: str, **kwargs) -> dict: + """Base64 encode a dictionary. Takes the same keywords as `json.loads` function.""" + return json.loads(base64.urlsafe_b64decode(value), **kwargs) |
