feat(snowflake): implement RegexSplit

ibis-project · Dec 19, 2023 · 2c1a726 · 2c1a726
1 parent cfe0329
commit 2c1a726
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/ibis/backends/snowflake/__init__.py b/ibis/backends/snowflake/__init__.py
@@ -49,7 +49,7 @@
             message="You have an incompatible version of 'pyarrow' installed",
             category=UserWarning,
         )
-    from snowflake.sqlalchemy import ARRAY, DOUBLE, OBJECT, URL
+    from snowflake.sqlalchemy import ARRAY, DOUBLE, OBJECT, URL, VARCHAR
 
     from ibis.backends.snowflake.converter import SnowflakePandasData
     from ibis.backends.snowflake.datatypes import SnowflakeType
@@ -113,6 +113,11 @@ class SnowflakeCompiler(AlchemyCompiler):
         "returns": ARRAY,
         "source": """return Array(count).fill(value).flat();""",
     },
+    "ibis_udfs.public.regexp_split": {
+        "inputs": {"value": VARCHAR, "pattern": VARCHAR},
+        "returns": ARRAY,
+        "source": """return value.split(new RegExp(pattern, "g"));""",
+    },
 }
 
 

diff --git a/ibis/backends/snowflake/registry.py b/ibis/backends/snowflake/registry.py
@@ -564,6 +564,7 @@ def _timestamp_range(t, op):
             3,
         ),
         ops.TimestampRange: _timestamp_range,
+        ops.RegexSplit: fixed_arity(sa.func.ibis_udfs.public.regexp_split, 2),
     }
 )
-Original file line number
+Diff line change
@@ Expand Up / @@ -564,6 +564,7 @@ def _timestamp_range(t, op): @@
 ,
             ),
             ops.TimestampRange: _timestamp_range,
+            ops.RegexSplit: fixed_arity(sa.func.ibis_udfs.public.regexp_split, 2),
         }
     )
@@ Expand Down @@