raise DuneError if insert failed

duneanalytics · May 24, 2024 · 886b796 · 886b796
1 parent 02712d5
commit 886b796
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 12 deletions.
diff --git a/dune_client/api/table.py b/dune_client/api/table.py
@@ -104,11 +104,12 @@ def insert_table(
             headers={"Content-Type": content_type},
             data=data,
         )
-        if "error" in result_json:
-            return InsertTableErrorResult.from_dict(result_json)
-        else:
-            return InsertTableResult.from_dict(result_json)
 
+            # return InsertTableErrorResult.from_dict(result_json)
+        try:
+            return InsertTableResult.from_dict(result_json)
+        except KeyError as err:
+            raise DuneError(result_json, "ResultsResponse", err) from err
 
     def delete_table(self, namespace: str, table_name: str) -> DeleteTableResult:
         """

diff --git a/dune_client/models.py b/dune_client/models.py
@@ -377,14 +377,6 @@ class InsertTableResult(DataClassJsonMixin):
     rows_written: int
     bytes_written: int
 
-@dataclass
-class InsertTableErrorResult(DataClassJsonMixin):
-    """
-    Data type returned by table/insert operation error
-    """
-
-    error: str
-
 
 @dataclass
 class DeleteTableResult(DataClassJsonMixin):