aboutsummaryrefslogtreecommitdiff
"""Generic data utilities: Rename module."""
import math
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",
                    order_key: str = "FamilyOrderId") -> list:
    """Order the populations by their families."""
    def __family_order__(item):
        orderval = item[order_key]
        return math.inf if orderval is None else orderval

    def __order__(ordered, current):
        _key = (__family_order__(current), current[family_key])
        return {
            **ordered,
            _key: ordered.get(_key, tuple()) + (current,)
        }

    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