Merge pull request #153 from wri/batman/workaround_gtc-2986

Workaround GTC-2986 by checking if version is successfully created after get 5XX response from API

src/datapump/clients/data_api.py

@@ -129,14 +129,24 @@ def create_vector_version(
     def create_version(
         self, dataset: str, version: str, payload: Dict[str, Any]
     ) -> Dict[str, Any]:
-
-        uri = f"{GLOBALS.data_api_uri}/dataset/{dataset}/{version}"
-        return self._send_request(ValidMethods.put, uri, payload)["data"]
+        try:
+            uri = f"{GLOBALS.data_api_uri}/dataset/{dataset}/{version}"
+            return self._send_request(ValidMethods.put, uri, payload)["data"]
+        except DataApiResponseError as e:
+            # Workaround for GTC-2986
+            # Getting a 500 response when creating version, but version is still created
+            # causing a 400 response on subsequent retries since we're trying to PUT
+            # an already existing version.
+            # For now, let's just return the version if it exists.
+            # Otherwise, propagate original exception.
+            try:
+                return self.get_version(dataset, version)
+            except DataApiResponseError:
+                raise e
 
     def create_aux_asset(
         self, dataset: str, version: str, payload: Dict[str, Any]
     ) -> Dict[str, Any]:
-
         uri = f"{GLOBALS.data_api_uri}/dataset/{dataset}/{version}/assets"
         return self._send_request(ValidMethods.post, uri, payload)["data"]