handle generator function

mo_sql_parsing/formatting.py

@@ -178,6 +178,8 @@ def dispatch(self, json, prec=100):
                 return self._literal(json, prec)
             elif "substring" in json:
                 return self._substring(json, prec)
+            elif "generator" in json:
+                return self._generator(json, prec)
             elif "group_concat" in json:
                 return self._group_concat(json, prec)
             elif "value" in json:
@@ -362,6 +364,11 @@ def _substring(self, json, prec):
             params = ", ".join(self.dispatch(p) for p in json["substring"])
             return f"SUBSTRING({params})"
 
+    def _generator(self, json, prec):
+        # TODO: replace with function-using-keywords
+        rowcount = self.dispatch(json["rowcount"])
+        return f"GENERATOR(ROWCOUNT=>{rowcount})"
+
     def _group_concat(self, json, prec):
         acc = ["group_concat(", self.dispatch(json["group_concat"]), " "]
         if "orderby" in json.keys():

mo_sql_parsing/sql_parser.py

@@ -305,7 +305,7 @@ def matching(type):
         one_param = (
             # KEYWORD PARAMETERS?
             # https://docs.snowflake.com/en/sql-reference/functions/generator.html
-            Group(ident + Literal("=>").suppress() + Group(expression))("kwargs")
+            Group(ident/ (lambda t: t[0].lower()) + Literal("=>").suppress() + Group(expression))("kwargs")
             / to_kwarg
         ) | Group(expression)("params")