Add universal JSONLogic type, which is essentially a union of all ava…

…ilable operations (#10)
Amsterdam · Sep 5, 2024 · 331529a · 331529a
1 parent be14668
commit 331529a
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 0 deletions.
diff --git a/pydantic_jsonlogic/__init__.py b/pydantic_jsonlogic/__init__.py
@@ -187,3 +187,42 @@ class Some(BaseJSONLogicOperation):
         # Something that produces a truthy or falsy result
         bool | BaseJSONLogicOperation,
     ]
+
+
+JSONLogic = (
+    Add
+    | All
+    | And
+    | Cat
+    | Divide
+    | Equals
+    | Filter
+    | GreaterThan
+    | GreaterThanOrEqual
+    | If
+    | In
+    | LessThan
+    | LessThanOrEqual
+    | Log
+    | Map
+    | Max
+    | Merge
+    | Min
+    | Max
+    | Missing
+    | MissingSome
+    | Modulo
+    | Multiply
+    | None_
+    | Not
+    | NotEquals
+    | NotNot
+    | Or
+    | Reduce
+    | Some
+    | StrictEquals
+    | StrictNotEquals
+    | Substr
+    | Subtract
+    | Var
+)
diff --git a/tests/test_model_integration.py b/tests/test_model_integration.py
@@ -0,0 +1,14 @@
+from pydantic import BaseModel
+
+from pydantic_jsonlogic import All, JSONLogic
+
+
+class SomeModel(BaseModel):
+    json_logic: JSONLogic
+
+
+def test_model_integration() -> None:
+    model = SomeModel.model_validate_json(
+        '{"json_logic": {"all":[{"var":"integers"}, {">=":[{"var":""}, 1]}]}}'
+    )
+    assert isinstance(model.json_logic, All)