pytrilogy 0.0.2.29__tar.gz → 0.0.2.30__tar.gz

{pytrilogy-0.0.2.29/pytrilogy.egg-info → pytrilogy-0.0.2.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytrilogy
-Version: 0.0.2.29
+Version: 0.0.2.30
 Summary: Declarative, typed query language that compiles to SQL.
 Home-page: 
 Author: 

{pytrilogy-0.0.2.29 → pytrilogy-0.0.2.30/pytrilogy.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytrilogy
-Version: 0.0.2.29
+Version: 0.0.2.30
 Summary: Declarative, typed query language that compiles to SQL.
 Home-page: 
 Author: 

{pytrilogy-0.0.2.29 → pytrilogy-0.0.2.30}/trilogy/__init__.py

@@ -4,6 +4,6 @@ from trilogy.executor import Executor
 from trilogy.parser import parse
 from trilogy.constants import CONFIG
 
-__version__ = "0.0.2.29"
+__version__ = "0.0.2.30"
 
 __all__ = ["parse", "Executor", "Dialects", "Environment", "CONFIG"]

{pytrilogy-0.0.2.29 → pytrilogy-0.0.2.30}/trilogy/parsing/render.py

@@ -386,13 +386,31 @@ class Renderer:
 
     @to_string.register
     def _(self, arg: "Function"):
-        inputs = ",".join(self.to_string(c) for c in arg.arguments)
+        args = [self.to_string(c) for c in arg.arguments]
+
+        if arg.operator == FunctionType.SUBTRACT:
+            return f"{args[0]} - {args[1]}"
+        if arg.operator == FunctionType.ADD:
+            return f"{args[0]} + {args[1]}"
+        if arg.operator == FunctionType.MULTIPLY:
+            return f"{args[0]} * {args[1]}"
+        if arg.operator == FunctionType.DIVIDE:
+            return f"{args[0]} / {args[1]}"
+        if arg.operator == FunctionType.MOD:
+            return f"{args[0]} % {args[1]}"
+
+        inputs = ",".join(args)
+
         if arg.operator == FunctionType.CONSTANT:
             return f"{inputs}"
         if arg.operator == FunctionType.CAST:
             return f"CAST({self.to_string(arg.arguments[0])} AS {self.to_string(arg.arguments[1])})"
         if arg.operator == FunctionType.INDEX_ACCESS:
             return f"{self.to_string(arg.arguments[0])}[{self.to_string(arg.arguments[1])}]"
+
+        if arg.operator == FunctionType.CASE:
+            inputs = "\n".join(args)
+            return f"CASE {inputs}\nEND"
         return f"{arg.operator.value}({inputs})"
 
     @to_string.register