relib 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl

relib/__init__.py

@@ -1,4 +1,5 @@
-from .utils import distinct, find, transpose_dict, make_combinations_by_dict, merge_dicts, intersect, ensure_tuple, omit, pick, flatten, transpose, deepen_dict, group
+from .utils import list_split, drop_none, distinct, find, transpose_dict, make_combinations_by_dict, \
+  merge_dicts, intersect, ensure_tuple, omit, dict_by, tuple_by, flatten, transpose, deepen_dict, group
 from .raypipe import raypipe
 from .hashing import hash
 from .measure_duration import measure_duration

relib/utils.py

@@ -1,20 +1,43 @@
-def distinct(items):
+from typing import TypeVar, Union, Sequence, Generator, Any
+
+T = TypeVar('T')
+U = TypeVar('U')
+
+def list_split(l: list[T], sep: T) -> list[list[T]]:
+  l = [sep, *l, sep]
+  split_at = [i for i, x in enumerate(l) if x is sep]
+  ranges = list(zip(split_at[0:-1], split_at[1:]))
+  result: Any = [
+    l[start + 1:end]
+    for start, end in ranges
+  ]
+  return result
+
+def drop_none(l: list[Union[T, None]]) -> list[T]:
+  return [x for x in l if x is not None]
+
+def distinct(items: list[T]) -> list[T]:
   return list(set(items))
 
-def find(iterable):
+def find(iterable: Union[Sequence[T], Generator[T, None, None]]) -> Union[T, None]:
   return next(iterable, None)
 
 def transpose_dict(des):
-  keys = list(des.keys())
-  length = len(des[keys[0]])
-
-  return [
-    {
-      key: des[key][i]
+  if isinstance(des, list):
+    keys = list(des[0].keys()) if des else []
+    length = len(des)
+    return {
+      key: [des[i][key] for i in range(length)]
       for key in keys
     }
-    for i in range(length)
-  ]
+  elif isinstance(des, dict):
+    keys = list(des.keys())
+    length = len(des[keys[0]]) if keys else 0
+    return [
+      {key: des[key][i] for key in keys}
+      for i in range(length)
+    ]
+  raise ValueError('transpose_dict only accepts dict or list')
 
 def make_combinations_by_dict(des, keys=None, pairs=[]):
   keys = sorted(des.keys()) if keys == None else keys
@@ -23,38 +46,40 @@ def make_combinations_by_dict(des, keys=None, pairs=[]):
   key = keys[0]
   remaining_keys = keys[1:]
   new_pairs = [(key, val) for val in des[key]]
-  return flatten(
-    [make_combinations_by_dict(des, remaining_keys, [pair] + pairs) for pair in new_pairs]
-  )
+  return flatten([
+    make_combinations_by_dict(des, remaining_keys, [pair] + pairs)
+    for pair in new_pairs
+  ])
 
-def merge_dicts(*dicts):
+def merge_dicts(*dicts: dict[T, U]) -> dict[T, U]:
   result = {}
   for dictionary in dicts:
     result.update(dictionary)
   return result
 
-def intersect(*lists):
+def intersect(*lists: list[T]) -> list[T]:
   return set.intersection(*map(set, lists))
 
-def ensure_tuple(value):
+def ensure_tuple(value: Union[T, tuple[T, ...]]) -> tuple[T, ...]:
   if isinstance(value, tuple):
     return value
   return (value,)
 
-def omit(d, keys):
+def omit(d: dict[T, U], keys: Sequence[T]) -> dict[T, U]:
   if keys:
     d = dict(d)
     for key in keys:
       del d[key]
   return d
 
-def pick(d, keys):
+def dict_by(keys: Sequence[T], values: Sequence[U]) -> dict[T, U]:
+  return dict(zip(keys, values))
+
+def tuple_by(d: dict[T, U], keys: Sequence[T]) -> tuple[U, ...]:
   return tuple(d[key] for key in keys)
 
-def flatten(l, iterations=1):
-  if iterations == 0:
-    return l
-  return flatten([value for inner_list in l for value in inner_list], iterations - 1)
+def flatten(l: list[list[T]]) -> list[T]:
+  return [value for inner_list in l for value in inner_list]
 
 def transpose(tuples, default_num_returns=0):
   result = tuple(zip(*tuples))
@@ -73,7 +98,7 @@ def deepen_dict(d):
     curr[head] = value
   return result
 
-def group(pairs):
+def group(pairs: Union[Sequence[tuple[T, U]], Generator[tuple[T, U], None, None]]) -> dict[T, list[U]]:
   values_by_key = {}
   for key, value in pairs:
     if key not in values_by_key:

{relib-0.2.1.dist-info → relib-0.2.3.dist-info}/LICENSE.txt

@@ -1,4 +1,4 @@
-Copyright 2018 Hampus Hallman
+Copyright 2023 Hampus Hallman
 
 Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
 

{relib-0.2.1.dist-info → relib-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: relib
-Version: 0.2.1
+Version: 0.2.3
 Home-page: https://github.com/Reddan/relib
 Author: Hampus Hallman
 Author-email: me@hampushallman.com

relib-0.2.3.dist-info/RECORD

@@ -0,0 +1,10 @@
+relib/__init__.py,sha256=OmN7DRgA2rYN0XMamE4P3rJPGJIFpnkFQoDXn-GK6sU,309
+relib/hashing.py,sha256=6iAPRiJI_4jaSooZRFJnqK2limXqTmErzcwpd050LAA,8943
+relib/measure_duration.py,sha256=jJa5Kh5FxaBysS__nkwqcnTt8Uc2niLucXfTzFE0j-E,555
+relib/raypipe.py,sha256=ynEoXs1dnD-360_uQC8v89xjiilt3knpocXpFaQ3plA,1905
+relib/utils.py,sha256=hJ0A3N-H3PaPmT06wWa003FGMlzHMoUOs-Wja49369o,3073
+relib-0.2.3.dist-info/LICENSE.txt,sha256=t9LfkVbmcvZjP0x3Sq-jR38UfTNbNtRQvc0Q8HWmLak,1054
+relib-0.2.3.dist-info/METADATA,sha256=9Q6IJAOtehmp-dQR6l3ppMl8MA1-XFJqE5LFSn84Xk4,260
+relib-0.2.3.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+relib-0.2.3.dist-info/top_level.txt,sha256=Yc96FwkbRYj4AQVatga8uK4hH9ATKI9XIyEH_1ba6KQ,6
+relib-0.2.3.dist-info/RECORD,,

{relib-0.2.1.dist-info → relib-0.2.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.37.1)
+Generator: bdist_wheel (0.38.4)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

relib-0.2.1.dist-info/RECORD

@@ -1,10 +0,0 @@
-relib/__init__.py,sha256=pzWwFP1HnqS-gPaER62RZuDyQuUOlbH0XiYQV87jsqU,269
-relib/hashing.py,sha256=6iAPRiJI_4jaSooZRFJnqK2limXqTmErzcwpd050LAA,8943
-relib/measure_duration.py,sha256=jJa5Kh5FxaBysS__nkwqcnTt8Uc2niLucXfTzFE0j-E,555
-relib/raypipe.py,sha256=ynEoXs1dnD-360_uQC8v89xjiilt3knpocXpFaQ3plA,1905
-relib/utils.py,sha256=CiYE2QU1ybmY2pk9AoNOuhcpMkc0y-PAGihFWzFDqyQ,1930
-relib-0.2.1.dist-info/LICENSE.txt,sha256=2c7g4mni-RUemFGkk6GnoFwknh-leF04BF_J_3gp4sg,1054
-relib-0.2.1.dist-info/METADATA,sha256=jEimdiPgyg6p219YAmhH6e3IpMU8SvNHeaDV_VJKDsE,260
-relib-0.2.1.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-relib-0.2.1.dist-info/top_level.txt,sha256=Yc96FwkbRYj4AQVatga8uK4hH9ATKI9XIyEH_1ba6KQ,6
-relib-0.2.1.dist-info/RECORD,,