lance-namespace-impls 0.1.2__tar.gz → 0.2.0__tar.gz

{lance_namespace_impls-0.1.2 → lance_namespace_impls-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lance-namespace-impls
-Version: 0.1.2
+Version: 0.2.0
 Summary: Third-party catalog implementations for Lance Namespace
 License: Apache-2.0
 Requires-Python: >=3.10

{lance_namespace_impls-0.1.2 → lance_namespace_impls-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "lance-namespace-impls"
-version = "0.1.2"
+version = "0.2.0"
 description = "Third-party catalog implementations for Lance Namespace"
 readme = "README.md"
 license = { text = "Apache-2.0" }

{lance_namespace_impls-0.1.2 → lance_namespace_impls-0.2.0}/src/lance_namespace_impls/hive2.py

@@ -72,6 +72,8 @@ from lance_namespace_urllib3_client.models import (
     CreateNamespaceResponse,
     DropNamespaceRequest,
     DropNamespaceResponse,
+    DropTableRequest,
+    DropTableResponse,
     ListTablesRequest,
     ListTablesResponse,
     DeclareTableRequest,
@@ -397,6 +399,31 @@ class Hive2Namespace(LanceNamespace):
             logger.error(f"Failed to describe table {request.id}: {e}")
             raise
 
+    def drop_table(self, request: DropTableRequest) -> DropTableResponse:
+        """Drop a table from the Hive Metastore and delete its data."""
+        try:
+            database, table_name = self._normalize_identifier(request.id)
+
+            with self.client as client:
+                table = client.get_table(database, table_name)
+
+                if not table.parameters:
+                    raise ValueError(f"Table {request.id} is not a Lance table")
+                table_type = table.parameters.get(TABLE_TYPE_KEY, "").lower()
+                if table_type != LANCE_TABLE_FORMAT:
+                    raise ValueError(f"Table {request.id} is not a Lance table")
+
+                location = table.sd.location if table.sd else None
+
+                client.drop_table(database, table_name, deleteData=True)
+
+                return DropTableResponse(location=location)
+        except Exception as e:
+            if NoSuchObjectException and isinstance(e, NoSuchObjectException):
+                raise ValueError(f"Table {request.id} does not exist")
+            logger.error(f"Failed to drop table {request.id}: {e}")
+            raise
+
     def deregister_table(
         self, request: DeregisterTableRequest
     ) -> DeregisterTableResponse:
@@ -405,10 +432,8 @@ class Hive2Namespace(LanceNamespace):
             database, table_name = self._normalize_identifier(request.id)
 
             with self.client as client:
-                # Get table to check if it's a Lance table
                 table = client.get_table(database, table_name)
 
-                # Check if it's a Lance table (case insensitive)
                 if not table.parameters:
                     raise ValueError(f"Table {request.id} is not a Lance table")
                 table_type = table.parameters.get(TABLE_TYPE_KEY, "").lower()
@@ -417,7 +442,6 @@ class Hive2Namespace(LanceNamespace):
 
                 location = table.sd.location if table.sd else None
 
-                # Drop the table metadata only (don't delete data)
                 client.drop_table(database, table_name, deleteData=False)
 
                 return DeregisterTableResponse(location=location)

{lance_namespace_impls-0.1.2 → lance_namespace_impls-0.2.0}/src/lance_namespace_impls/hive3.py

@@ -74,6 +74,8 @@ from lance_namespace_urllib3_client.models import (
     CreateNamespaceResponse,
     DropNamespaceRequest,
     DropNamespaceResponse,
+    DropTableRequest,
+    DropTableResponse,
     ListTablesRequest,
     ListTablesResponse,
     DeclareTableRequest,
@@ -477,6 +479,32 @@ class Hive3Namespace(LanceNamespace):
             logger.error(f"Failed to describe table {request.id}: {e}")
             raise
 
+    def drop_table(self, request: DropTableRequest) -> DropTableResponse:
+        """Drop a table and delete its data."""
+        try:
+            catalog, database, table_name = self._normalize_identifier(request.id)
+
+            with self.client as client:
+                table = client.get_table(database, table_name)
+
+                if not table.parameters:
+                    raise ValueError(f"Table {request.id} is not a Lance table")
+                table_type = table.parameters.get(TABLE_TYPE_KEY, "").lower()
+                if table_type != LANCE_TABLE_FORMAT:
+                    raise ValueError(f"Table {request.id} is not a Lance table")
+
+                location = table.sd.location if table.sd else None
+
+                client.drop_table(database, table_name, deleteData=True)
+
+                return DropTableResponse(location=location)
+
+        except Exception as e:
+            if NoSuchObjectException and isinstance(e, NoSuchObjectException):
+                raise ValueError(f"Table {request.id} does not exist")
+            logger.error(f"Failed to drop table {request.id}: {e}")
+            raise
+
     def deregister_table(
         self, request: DeregisterTableRequest
     ) -> DeregisterTableResponse: