ifttt func implemented

version bump 0.0.6

Author: znittzel

maz/__init__.py

@@ -349,4 +349,17 @@ def __call__(self, objects: typing.Iterable[typing.Any]) -> typing.Iterable[typi
                 ),
                 key=operator.itemgetter(0),
             )
-        )
+        )
+
+class ifttt:
+
+    def __init__(self, fnif, fnthen, fnelse):
+        self.fnif = fnif
+        self.fnthen = fnthen
+        self.fnelse = fnelse
+
+    def __call__(self, obj: typing.Any) -> typing.Any:
+        if self.fnif(obj):
+            return self.fnthen(obj)
+        else:
+            return self.fnelse(obj)

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = maz
-version = 0.0.5
+version = 0.0.6
 author = ourstudio
 author_email = rikard@ourstudio.se
 description = Functional programming tools.

tests/test_maz.py

@@ -126,4 +126,41 @@ def __eq__(self, o):
         Var("d", 5),
         Var("e", 6),
     ]
+    assert actual == expected
+
+def test_ifttt_function():
+
+    class Var:
+        def __init__(self, id: str, n: int):
+            self.id = id
+            self.n = n
+        
+        def __eq__(self, o):
+            return (self.id == o.id) and (self.n == o.n)
+
+    items = [
+        Var("a", 1),
+        Var("b", 2),
+        Var("c", 3),
+        Var("d", 4),
+        Var("e", 5),
+    ]
+
+    actual = list(
+        map(
+            maz.ifttt(
+                lambda x: x.n > 2,
+                lambda x: Var(x.id, x.n+1),
+                lambda x: Var(x.id, x.n-1)
+            ),
+            items
+        )
+    )
+    expected = [
+        Var("a", 0),
+        Var("b", 1),
+        Var("c", 4),
+        Var("d", 5),
+        Var("e", 6),
+    ]
     assert actual == expected