Text-analytics: retry policy for PUT operations

Author: dariosky

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+
+## v1.0.38 (2020-09-04)
+* Text-Analytics API additional functions: `get_analytics` and `get_annotated`
+
 ## v1.0.36 (2020-05-14)
 * Text-Analytics API endpoint updated: folders & richer metadata
 * Extended error handling to support Feed disconnection problems

ravenpackapi/core.py

@@ -15,7 +15,7 @@
 from ravenpackapi.utils.date_formats import as_datetime_str
 
 _VALID_METHODS = ('get', 'post', 'put', 'delete', 'patch')
-VERSION = '1.0.37'
+VERSION = '1.0.38'
 
 logger = logging.getLogger("ravenpack.core")
 

ravenpackapi/upload/module.py

@@ -1,10 +1,15 @@
+import logging
 import mimetypes
 import os
+from time import sleep
 
+from ravenpackapi.exceptions import APIException
 from ravenpackapi.upload.models import (File, FILE_FIELDS,
                                         Folder, FOLDER_FIELDS)
 from ravenpackapi.utils.date_formats import as_datetime_str
 
+logger = logging.getLogger("ravenpack.upload")
+
 
 class UploadApi(object):
     def __init__(self, api):
@@ -90,15 +95,27 @@ def file(self, name_or_file_handler,
         file_id = promise['file_id']
         location = promise['Location']
 
-        self.api.request(
-            endpoint=location,
-            method='put',
-            data=fh,
-            headers={
-                "x-amz-server-side-encryption": "AES256",
-                "Content-Type": content_type,
-            }
-        )
+        attempt = 3
+        while attempt:
+            try:
+                fh.seek(0)
+                self.api.request(
+                    endpoint=location,
+                    method='put',
+                    data=fh,
+                    headers={
+                        "x-amz-server-side-encryption": "AES256",
+                        "Content-Type": content_type,
+                    }
+                )
+            except APIException as e:
+                attempt -= 1
+                if attempt == 0:
+                    raise e  # raise the exception
+                logger.warning("Error with PUT file operation - retring")
+                sleep(1)
+            else:
+                break
 
         if close_file:  # we opened the handler, so let's close it
             fh.close()

setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-VERSION = '1.0.37'
+VERSION = '1.0.38'
 
 with open('README.rst') as readme_file:
     readme = readme_file.read()