Ref/sqlalchemy session

CHANGELOG.md

@@ -37,6 +37,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Add a standalone flag in Streamlit to mark the page as independent.
 - Add secrets directory mount for loading secret env vars.
 - Remove components recursively
+- Add metadata batched db updates
 
 #### Bug Fixes
 

plugins/ibis/superduper_ibis/__init__.py

@@ -1,6 +1,6 @@
 from .data_backend import IbisDataBackend as DataBackend
 from .query import IbisQuery
 
-__version__ = "0.4.7"
+__version__ = "0.4.8"
 
 __all__ = ["IbisQuery", "DataBackend"]

plugins/ibis/superduper_ibis/data_backend.py

@@ -216,11 +216,11 @@ def drop_table_or_collection(self, name: str):
         :param name: Table name to drop.
         """
         try:
-            return self.db.databackend.conn.drop_table(name)
+            return self.conn.drop_table(name)
         except Exception as e:
             msg = "Object found is of type 'VIEW'"
             if msg in str(e):
-                return self.db.databackend.conn.drop_view(name)
+                return self.conn.drop_view(name)
             raise
 
     def create_output_dest(
@@ -300,7 +300,7 @@ def drop(self, force: bool = False):
 
         for table in self.conn.list_tables():
             logging.info(f"Dropping table: {table}")
-            self.conn.drop_table(table)
+            self.drop_table_or_collection(table)
 
     def get_table_or_collection(self, identifier):
         """Get a table or collection from the database.

plugins/sqlalchemy/plugin_test/test_metadata.py

@@ -11,6 +11,7 @@
 @pytest.fixture
 def metadata():
     store = SQLAlchemyMetadata(DATABASE_URL)
+    store._batched = False
     yield store
     store.drop(force=True)
 

plugins/sqlalchemy/superduper_sqlalchemy/__init__.py

@@ -1,5 +1,5 @@
 from .metadata import SQLAlchemyMetadata as MetaDataStore
 
-__version__ = "0.4.5"
+__version__ = "0.4.6"
 
 __all__ = ['MetaDataStore']