diff options
Diffstat (limited to 'uploader/datautils.py')
-rw-r--r-- | uploader/datautils.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/uploader/datautils.py b/uploader/datautils.py new file mode 100644 index 0000000..b95a9e0 --- /dev/null +++ b/uploader/datautils.py @@ -0,0 +1,21 @@ +"""Generic data utilities: Rename module.""" +import math +from functools import reduce + +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]) |