fabricks 3.0.14__tar.gz → 3.0.15__tar.gz

{fabricks-3.0.14 → fabricks-3.0.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricks
-Version: 3.0.14
+Version: 3.0.15
 Author-email: BMS DWH Team <bi_support@bmsuisse.ch>
 Requires-Python: <4,>=3.9
 Requires-Dist: azure-data-tables<13,>=12.5.0

{fabricks-3.0.14 → fabricks-3.0.15}/fabricks/context/helpers.py

@@ -15,7 +15,7 @@ def get_config_from_file():
             break
         if path == path.parent:
             break
-        
+
         path = path.parent
 
     if (path / "fabricksconfig.json").exists():

{fabricks-3.0.14 → fabricks-3.0.15}/fabricks/core/udfs.py

@@ -11,7 +11,7 @@ from fabricks.context.log import DEFAULT_LOGGER
 UDFS: dict[str, Callable] = {}
 
 
-def register_all_udfs(extension: Optional[str] = None):
+def register_all_udfs(extension: Optional[str] = None, override: bool = False):
     """
     Register all user-defined functions (UDFs).
     """
@@ -20,7 +20,7 @@ def register_all_udfs(extension: Optional[str] = None):
     for udf in get_udfs(extension=extension):
         split = udf.split(".")
         try:
-            register_udf(udf=split[0], extension=split[1])
+            register_udf(udf=split[0], extension=split[1], override=override)
         except Exception as e:
             DEFAULT_LOGGER.exception(f"could not register udf {udf}", exc_info=e)
 
@@ -57,7 +57,12 @@ def is_registered(udf: str, spark: Optional[SparkSession] = None) -> bool:
     return not df.isEmpty()
 
 
-def register_udf(udf: str, extension: Optional[str] = None, spark: Optional[SparkSession] = None):
+def register_udf(
+    udf: str,
+    extension: Optional[str] = None,
+    spark: Optional[SparkSession] = None,
+    override: bool = False,
+):
     """
     Register a user-defined function (UDF).
     """
@@ -65,8 +70,11 @@ def register_udf(udf: str, extension: Optional[str] = None, spark: Optional[Spar
         spark = SPARK
     assert spark is not None
 
-    if not is_registered(udf, spark):
-        DEFAULT_LOGGER.debug(f"register udf {udf}")
+    if not is_registered(udf, spark) or override:
+        if override:
+            DEFAULT_LOGGER.debug(f"override udf {udf}")
+        else:
+            DEFAULT_LOGGER.debug(f"register udf {udf}")
 
         if extension is None:
             extension = get_extension(udf)

{fabricks-3.0.14 → fabricks-3.0.15}/fabricks/deploy/udfs.py

@@ -4,10 +4,10 @@ from fabricks.core.udfs import register_all_udfs
 from fabricks.utils.sqlglot import fix as fix_sql
 
 
-def deploy_udfs():
+def deploy_udfs(override: bool = True):
     DEFAULT_LOGGER.info("create or replace udfs")
 
-    register_all_udfs(extension="sql")
+    register_all_udfs(extension="sql", override=override)
     create_or_replace_udf_job_id()
 
 

{fabricks-3.0.14 → fabricks-3.0.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fabricks"
-version = "3.0.14"
+version = "3.0.15"
 description = ""
 authors = [
     { name = "BMS DWH Team", email = "bi_support@bmsuisse.ch" },

{fabricks-3.0.14 → fabricks-3.0.15}/uv.lock

@@ -645,7 +645,7 @@ wheels = [
 
 [[package]]
 name = "fabricks"
-version = "3.0.14"
+version = "3.0.15"
 source = { editable = "." }
 dependencies = [
     { name = "azure-data-tables" },