pydataframer-databricks 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

pydataframer_databricks/connectors.py

@@ -203,42 +203,37 @@ class DatabricksConnector:
                 raise ValueError(f"Unsupported file_type: {file_type}. Supported: CSV, JSON, JSONL for SINGLE_FILE datasets")
             
             with self.get_connection() as connection:
-                cursor = connection.cursor()
-
-                columns_sql = ", ".join(
-                    f"`{col}` STRING" for col in pandas_df.columns
-                )
-
-                try:
-                    cursor.execute(f"""
-                        CREATE OR REPLACE TABLE {table_name} (
-                            {columns_sql}
-                        )
-                    """)
-                except Exception as e:
-                    error_msg = f"Failed to create table `{table_name}`"
-                    print(f"{error_msg}: {str(e)}")
-                    print("Verify table name format (catalog.schema.table), permissions, and warehouse is running")
-                    cursor.close()
-                    raise RuntimeError(f"{error_msg}: {str(e)}") from e
-
-                insert_sql = f"""
-                    INSERT INTO {table_name}
-                    VALUES ({", ".join(["?"] * len(pandas_df.columns))})
-                """
-
-                try:
-                    cursor.executemany(
-                        insert_sql,
-                        pandas_df.values.tolist()
+                with connection.cursor() as cursor:
+                    columns_sql = ", ".join(
+                        f"`{col}` STRING" for col in pandas_df.columns
                     )
-                except Exception as e:
-                    error_msg = f"Failed to insert data into table `{table_name}`"
-                    print(f"{error_msg}: {str(e)} | Rows attempted: {len(pandas_df)}")
-                    cursor.close()
-                    raise RuntimeError(f"{error_msg}: {str(e)}") from e
 
-                cursor.close()
+                    try:
+                        cursor.execute(f"""
+                            CREATE OR REPLACE TABLE {table_name} (
+                                {columns_sql}
+                            )
+                        """)
+                    except Exception as e:
+                        error_msg = f"Failed to create table `{table_name}`"
+                        print(f"{error_msg}: {str(e)}")
+                        print("Verify table name format (catalog.schema.table), permissions, and warehouse is running")
+                        raise RuntimeError(f"{error_msg}: {str(e)}") from e
+
+                    insert_sql = f"""
+                        INSERT INTO {table_name}
+                        VALUES ({", ".join(["?"] * len(pandas_df.columns))})
+                    """
+
+                    try:
+                        cursor.executemany(
+                            insert_sql,
+                            pandas_df.values.tolist()
+                        )
+                    except Exception as e:
+                        error_msg = f"Failed to insert data into table `{table_name}`"
+                        print(f"{error_msg}: {str(e)} | Rows attempted: {len(pandas_df)}")
+                        raise RuntimeError(f"{error_msg}: {str(e)}") from e
 
             print(f"✅ Table `{table_name}` saved successfully using Databricks SQL")
         

{pydataframer_databricks-0.1.1.dist-info → pydataframer_databricks-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydataframer-databricks
-Version: 0.1.1
+Version: 0.1.2
 Summary: Databricks connector for Dataframer
 Author-email: Dataframer <info@dataframer.ai>
 License: MIT

pydataframer_databricks-0.1.2.dist-info/RECORD

@@ -0,0 +1,5 @@
+pydataframer_databricks/__init__.py,sha256=piRrFtKpGGc2ctFDnMNblp5Whp6froRKXNeYkHnrw_o,214
+pydataframer_databricks/connectors.py,sha256=ZzRBuv-RGJeKoGaAtUTov_52yq2Tq_YKfOs5vFSQ9Zw,9389
+pydataframer_databricks-0.1.2.dist-info/METADATA,sha256=VN0LL1dH5UL0f9eWJnf15gSA-kvyVoXemJGkgbRa6BI,1113
+pydataframer_databricks-0.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+pydataframer_databricks-0.1.2.dist-info/RECORD,,

pydataframer_databricks-0.1.1.dist-info/RECORD

@@ -1,5 +0,0 @@
-pydataframer_databricks/__init__.py,sha256=piRrFtKpGGc2ctFDnMNblp5Whp6froRKXNeYkHnrw_o,214
-pydataframer_databricks/connectors.py,sha256=efzhUrls_ns0_rwSjXi4n8w_Adcez7cJt5VJT7qHc2Q,9377
-pydataframer_databricks-0.1.1.dist-info/METADATA,sha256=7g23DdCJraqZQkWWemWlA3s8jmGM2nCmoDmWNZi6560,1113
-pydataframer_databricks-0.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-pydataframer_databricks-0.1.1.dist-info/RECORD,,