fabricks 3.0.3__tar.gz → 3.0.4__tar.gz

{fabricks-3.0.3 → fabricks-3.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricks
-Version: 3.0.3
+Version: 3.0.4
 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.3 → fabricks-3.0.4}/fabricks/cdc/base/configurator.py

@@ -37,6 +37,10 @@ class Configurator(ABC):
     def is_registered(self):
         return self.table.is_registered
 
+    @property
+    def qualified_name(self):
+        return f"{self.database}_{'_'.join(self.levels)}"
+
     @abstractmethod
     def get_query(self, src: Union[DataFrame, Table, str], **kwargs):
         raise NotImplementedError()

{fabricks-3.0.3 → fabricks-3.0.4}/fabricks/cdc/base/merger.py

@@ -101,7 +101,7 @@ class Merger(Processor):
             self.create_table(src, **kwargs)
 
         df = self.get_data(src, **kwargs)
-        global_temp_view = f"{self.database}_{'_'.join(self.levels)}__merge"
+        global_temp_view = f"{self.qualified_name}__merge"
         view = create_or_replace_global_temp_view(global_temp_view, df, uuid=kwargs.get("uuid", False))
 
         merge = self.get_merge_query(view, **kwargs)

{fabricks-3.0.3 → fabricks-3.0.4}/fabricks/cdc/base/processor.py

@@ -16,7 +16,7 @@ from fabricks.utils.sqlglot import fix as fix_sql
 class Processor(Generator):
     def get_data(self, src: Union[DataFrame, Table, str], **kwargs) -> DataFrame:
         if isinstance(src, (DataFrame, CDataFrame)):
-            name = f"{self.database}_{'_'.join(self.levels)}__data"
+            name = f"{self.qualified_name}__data"
             global_temp_view = create_or_replace_global_temp_view(name, src, uuid=kwargs.get("uuid", False))
             src = f"select * from {global_temp_view}"
 
@@ -347,8 +347,11 @@ class Processor(Generator):
         df = self.get_data(src, **kwargs)
         df = self.reorder_columns(df)
 
+        name = f"{self.qualified_name}__append"
+        create_or_replace_global_temp_view(name, df, uuid=kwargs.get("uuid", False))
+
         DEFAULT_LOGGER.debug("append", extra={"job": self})
-        df.write.format("delta").mode("append").save(self.table.delta_path.string)
+        self.spark.sql(f"insert into table {self.table} by name select * from global_temp.{name}")
 
     def overwrite(
         self,
@@ -366,9 +369,11 @@ class Processor(Generator):
             if kwargs.get("update_where"):
                 dynamic = True
 
-        writer = df.write.format("delta").mode("overwrite")
         if dynamic:
-            writer.option("partitionOverwriteMode", "dynamic")
+            self.spark.sql("set spark.sql.sources.partitionOverwriteMode = dynamic")
+
+        name = f"{self.qualified_name}__overwrite"
+        create_or_replace_global_temp_view(name, df, uuid=kwargs.get("uuid", False))
 
-        DEFAULT_LOGGER.info("overwrite", extra={"job": self})
-        writer.save(self.table.delta_path.string)
+        DEFAULT_LOGGER.debug("overwrite", extra={"job": self})
+        self.spark.sql(f"insert overwrite table {self.table} by name select * from global_temp.{name}")

{fabricks-3.0.3 → fabricks-3.0.4}/fabricks/core/jobs/base/_types.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Any, List, Literal, Optional, TypedDict, Union
+from typing import List, Literal, Optional, TypedDict, Union
 
 from pydantic import BaseModel, ConfigDict, model_validator
 from pyspark.sql.types import StringType, StructField, StructType
@@ -33,8 +33,8 @@ Origins = Literal["parser", "job"]
 
 
 class SparkOptions(TypedDict):
-    sql: Optional[dict[Any, Any]]
-    conf: Optional[dict[Any, Any]]
+    sql: Optional[dict[str, str]]
+    conf: Optional[dict[str, str]]
 
 
 class TableOptions(TypedDict):
@@ -45,17 +45,17 @@ class TableOptions(TypedDict):
     cluster_by: Optional[List[str]]
     powerbi: Optional[bool]
     bloomfilter_by: Optional[List[str]]
-    constraints: Optional[dict[Any, Any]]
-    properties: Optional[dict[Any, Any]]
+    constraints: Optional[dict[str, str]]
+    properties: Optional[dict[str, str]]
     comment: Optional[str]
-    calculated_columns: Optional[dict[Any, Any]]
+    calculated_columns: Optional[dict[str, str]]
     retention_days: Optional[int]
 
 
 class _InvokeOptions(TypedDict):
     notebook: str
     timeout: int
-    arguments: Optional[dict[Any, Any]]
+    arguments: Optional[dict[str, str]]
 
 
 class InvokerOptions(TypedDict):
@@ -66,7 +66,7 @@ class InvokerOptions(TypedDict):
 
 class ExtenderOptions(TypedDict):
     extender: str
-    arguments: Optional[dict[Any, Any]]
+    arguments: Optional[dict[str, str]]
 
 
 class CheckOptions(TypedDict):
@@ -90,7 +90,7 @@ class BronzeOptions(TypedDict):
     filter_where: Optional[str]
     # extra
     encrypted_columns: Optional[List[str]]
-    calculated_columns: Optional[dict[Any, Any]]
+    calculated_columns: Optional[dict[str, str]]
     operation: Optional[Operations]
     timeout: Optional[int]
 
@@ -106,7 +106,7 @@ class SilverOptions(TypedDict):
     deduplicate: Optional[bool]
     stream: Optional[bool]
     # else
-    order_duplicate_by: Optional[dict[Any, Any]]
+    order_duplicate_by: Optional[dict[str, str]]
     timeout: Optional[int]
 
 
@@ -141,7 +141,7 @@ class BaseJobConf:
 
 @dataclass
 class JobConfBronze(BaseJobConf):
-    step: str
+    step: TBronze
     options: BronzeOptions
     table_options: Optional[TableOptions] = None
     parser_options: Optional[ParserOptions] = None
@@ -155,7 +155,7 @@ class JobConfBronze(BaseJobConf):
 
 @dataclass
 class JobConfSilver(BaseJobConf):
-    step: str
+    step: TSilver
     options: SilverOptions
     table_options: Optional[TableOptions] = None
     check_options: Optional[CheckOptions] = None
@@ -168,7 +168,7 @@ class JobConfSilver(BaseJobConf):
 
 @dataclass
 class JobConfGold(BaseJobConf):
-    step: str
+    step: TGold
     options: Optional[GoldOptions]
     table_options: Optional[TableOptions] = None
     check_options: Optional[CheckOptions] = None

{fabricks-3.0.3 → fabricks-3.0.4}/pyproject.toml

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

{fabricks-3.0.3 → fabricks-3.0.4}/uv.lock

@@ -627,7 +627,7 @@ wheels = [
 
 [[package]]
 name = "fabricks"
-version = "3.0.3"
+version = "3.0.4"
 source = { editable = "." }
 dependencies = [
     { name = "azure-data-tables" },