Merge pull request #119 from duneanalytics/support-delete-table-endpoint

Add `delete` table endpoint to sdk

Author: waddahAldrobi

.gitignore

@@ -7,3 +7,5 @@ _version.py
 venv/
 tmp/
 .vscode/
+build/
+.DS_Store

dune_client/api/base.py

@@ -209,3 +209,14 @@ def _patch(self, route: str, params: Any) -> Any:
             timeout=self.request_timeout,
         )
         return self._handle_response(response)
+
+    def _delete(self, route: str) -> Any:
+        """Generic interface for the DELETE method of a Dune API request"""
+        url = self._route_url(route)
+        self.logger.debug(f"DELETE received input url={url}")
+        response = self.http.delete(
+            url=url,
+            headers=self.default_headers(),
+            timeout=self.request_timeout,
+        )
+        return self._handle_response(response)

dune_client/api/table.py

@@ -7,7 +7,12 @@
 from typing import List, Dict, IO
 
 from dune_client.api.base import BaseRouter
-from dune_client.models import DuneError, InsertTableResult, CreateTableResult
+from dune_client.models import (
+    DuneError,
+    InsertTableResult,
+    CreateTableResult,
+    DeleteTableResult,
+)
 
 
 class TableAPI(BaseRouter):
@@ -99,3 +104,12 @@ def insert_table(
             data=data,
         )
         return InsertTableResult.from_dict(result_json)
+
+    def delete_table(self, namespace: str, table_name: str) -> DeleteTableResult:
+        """
+        https://docs.dune.com/api-reference/tables/endpoint/delete
+        The delete table endpoint allows you to delete an existing table in Dune.
+        """
+
+        response_json = self._delete(route=f"/table/{namespace}/{table_name}")
+        return DeleteTableResult.from_dict(response_json)

dune_client/models.py

@@ -375,3 +375,12 @@ class InsertTableResult(DataClassJsonMixin):
     """
 
     rows_written: int
+
+
+@dataclass
+class DeleteTableResult(DataClassJsonMixin):
+    """
+    Data type returned by table/delete operation
+    """
+
+    message: str

tests/e2e/test_client.py

@@ -13,6 +13,7 @@
     DuneError,
     InsertTableResult,
     CreateTableResult,
+    DeleteTableResult,
 )
 from dune_client.types import QueryParameter
 from dune_client.client import DuneClient
@@ -226,7 +227,7 @@ def test_get_latest_result_with_query_id(self):
         results = dune.get_latest_result(self.query.query_id).get_rows()
         self.assertGreater(len(results), 0)
 
-    @unittest.skip("This is a plus subscription endpoint.")
+    @unittest.skip("Requires custom namespace and table_name input.")
     def test_upload_csv_success(self):
         client = DuneClient(self.valid_api_key)
         self.assertEqual(
@@ -238,7 +239,7 @@ def test_upload_csv_success(self):
             True,
         )
 
-    @unittest.skip("This is a plus subscription endpoint.")
+    @unittest.skip("Requires custom namespace and table_name input.")
     def test_create_table_success(self):
         # Make sure the table doesn't already exist.
         # You will need to change the namespace to your own.
@@ -268,7 +269,7 @@ def test_create_table_success(self):
             ),
         )
 
-    @unittest.skip("This is a plus subscription endpoint.")
+    @unittest.skip("Requires custom namespace and table_name input.")
     def test_insert_table_csv_success(self):
         # Make sure the table already exists and csv matches table schema.
         # You will need to change the namespace to your own.
@@ -284,7 +285,7 @@ def test_insert_table_csv_success(self):
                 InsertTableResult(rows_written=1),
             )
 
-    @unittest.skip("This is a plus subscription endpoint.")
+    @unittest.skip("Requires custom namespace and table_name input.")
     def test_insert_table_json_success(self):
         # Make sure the table already exists and json matches table schema.
         # You will need to change the namespace to your own.
@@ -300,6 +301,27 @@ def test_insert_table_json_success(self):
                 InsertTableResult(rows_written=1),
             )
 
+    @unittest.skip("Requires custom namespace and table_name input.")
+    def test_delete_table_success(self):
+        # Make sure the table doesn't already exist.
+        # You will need to change the namespace to your own.
+        client = DuneClient(self.valid_api_key)
+
+        namespace = "test"
+        table_name = "dataset_e2e_test"
+
+        self.assertEqual(
+            client.delete_table(
+                namespace=namespace,
+                table_name=table_name,
+            ),
+            DeleteTableResult.from_dict(
+                {
+                    "message": "Table teamwaddah.waddah_test3 successfully deleted",
+                }
+            ),
+        )
+
     def test_download_csv_with_pagination(self):
         # Arrange
         client = DuneClient(self.valid_api_key)