satisfactoscript 0.6.0__tar.gz → 0.6.1__tar.gz

{satisfactoscript-0.6.0 → satisfactoscript-0.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: satisfactoscript
-Version: 0.6.0
+Version: 0.6.1
 Summary: An Enterprise-Ready, Declarative Data Engineering Framework for Databricks Lakehouse.
 Author: julhouba
 Classifier: Programming Language :: Python :: 3

{satisfactoscript-0.6.0 → satisfactoscript-0.6.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "satisfactoscript"
-version = "0.6.0"
+version = "0.6.1"
 description = "An Enterprise-Ready, Declarative Data Engineering Framework for Databricks Lakehouse."
 readme = "README.md"
 authors = [

{satisfactoscript-0.6.0 → satisfactoscript-0.6.1}/src/satisfactoscript/core/core.py

@@ -502,13 +502,40 @@ class SatisfactoEngine:
 
     def _drop_table_if_exists(self, fqn):
         """
-        Drops a table using SQL directly if it exists.
-        
+        Drops a table if it exists, using two strategies in order:
+        1. Spark SQL DROP TABLE IF EXISTS (works natively on Databricks).
+        2. Databricks SDK REST API (works locally when gRPC DDL fails with UserContext).
+
         Args:
-            fqn (str): The Fully Qualified Name of the table.
+            fqn (str): The Fully Qualified Name of the table (backtick-quoted).
         """
         print(f"     -> [Cleanup] Dropping table if exists: {fqn}")
-        self.spark.sql(f"DROP TABLE IF EXISTS {fqn}")
+
+        # Strategy 1: Spark SQL (works natively on Databricks)
+        try:
+            self.spark.sql(f"DROP TABLE IF EXISTS {fqn}")
+            return
+        except Exception:
+            pass
+
+        # Strategy 2: Databricks SDK REST API (works locally when gRPC DDL fails)
+        try:
+            from databricks.sdk import WorkspaceClient
+            host = os.getenv("DATABRICKS_HOST")
+            token = os.getenv("DATABRICKS_TOKEN")
+            if host and token:
+                clean_fqn = fqn.replace("`", "")
+                w = WorkspaceClient(host=host, token=token)
+                try:
+                    w.tables.delete(clean_fqn)
+                except Exception:
+                    # Table does not exist — equivalent to IF EXISTS, not an error
+                    pass
+                return
+        except Exception:
+            pass
+
+        print(f"     -> [Cleanup] WARNING: Could not drop {fqn} via SQL or SDK. Continuing anyway.")
 
     def _write_dataframe(self, df, fqn, label):
         """

{satisfactoscript-0.6.0 → satisfactoscript-0.6.1}/src/satisfactoscript.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: satisfactoscript
-Version: 0.6.0
+Version: 0.6.1
 Summary: An Enterprise-Ready, Declarative Data Engineering Framework for Databricks Lakehouse.
 Author: julhouba
 Classifier: Programming Language :: Python :: 3