Merge pull request #13 from flojoy-ai/jeff/iteration-and-functions

feat: functions, sequences, conditionals, editable constant

blocks/flojoy/control/slider.py

@@ -2,5 +2,5 @@
 
 
 @ui_input
-def slider(x):
+def slider(x: int) -> int:
     return x

blocks/flojoy/control/toggle.py

@@ -0,0 +1,6 @@
+from captain.decorators import ui_input
+
+
+@ui_input
+def toggle(x: bool):
+    return x

blocks/flojoy/logic/clock.py

@@ -0,0 +1,5 @@
+import reactivex as rx
+
+
+def clock():
+    return rx.interval(1 / 60)

blocks/flojoy/logic/conditional.py

@@ -0,0 +1,14 @@
+from typing import TypedDict
+from captain.types.builtins import Ignore
+
+
+class Output(TypedDict):
+    true: Ignore | None
+    false: Ignore | None
+
+
+def conditional(b: bool) -> Output:
+    return Output(
+        true=None if b else Ignore(),
+        false=Ignore() if b else None,
+    )

blocks/flojoy/logic/false.py

@@ -0,0 +1,2 @@
+def false() -> bool:
+    return False

blocks/flojoy/logic/sequence.py

@@ -0,0 +1,5 @@
+import reactivex as rx
+
+
+def sequence(start: int = 0, stop: int | None = None, step: int | None = None):
+    return rx.range(start=start, stop=stop, step=step)

blocks/flojoy/logic/true.py

@@ -0,0 +1,2 @@
+def true() -> bool:
+    return True

blocks/flojoy/math/arithmetic/add.py

@@ -1,2 +1,2 @@
-def add(x, y):
+def add(x: int, y: int) -> int:
     return x + y

blocks/flojoy/math/arithmetic/subtract.py

@@ -1,2 +1,2 @@
-def subtract(x, y):
+def subtract(x: int, y: int) -> int:
     return x - y

blocks/flojoy/math/constant.py

@@ -1,2 +1,2 @@
-def constant():
-    return 2
+def constant(val: int) -> int:
+    return val

blocks/flojoy/math/rand.py

@@ -0,0 +1,6 @@
+import random
+from typing import Any
+
+
+def rand(x: Any = None) -> int:
+    return random.randint(0, 100)