Merge branch 'TA-1188_use_status_endpoint' into 'master'

Switching get_status method to use new status endpoint which should be faster

See merge request python/ravenpack-api!2

Author: Dario Varotto

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## v1.0.56 (2021-11-04)
+
+Internal changes to speed up get_status and wait_for_completion file methods
+
 ## v1.0.53 (2021-08-26)
 
 Product aware instance: allows to access edge just instanciating your api with

ravenpackapi/core.py

@@ -15,7 +15,7 @@
 from ravenpackapi.utils.dynamic_sessions import DynamicSession
 
 _VALID_METHODS = ('get', 'post', 'put', 'delete', 'patch')
-VERSION = '1.0.55'
+VERSION = '1.0.56'
 
 logger = logging.getLogger("ravenpack.core")
 

ravenpackapi/upload/models.py

@@ -1,6 +1,6 @@
 from time import sleep
 
-from ravenpackapi.exceptions import api_method
+from ravenpackapi.exceptions import api_method, APIException
 from ravenpackapi.upload.upload_utils import retry_on_too_early
 
 FILE_FIELDS = (
@@ -46,7 +46,24 @@ def __str__(self):
 
     @api_method
     def get_status(self):
-        self.get_metadata(force_refresh=True)
+        response = retry_on_too_early(
+            self.api.request,
+            '%s/files/%s/status' % (self.api._UPLOAD_BASE_URL, self.file_id)
+            )
+        api_status = response.json()
+        response.raise_for_status()
+
+        self.status = None
+        if api_status.get('error'):
+            raise APIException(
+                'Got an error {status}: body was \'{error_message}\''.format(
+                    status=response.status_code,
+                    error_message=api_status['error']
+                ), response=response
+            )
+        if api_status.get('status'):
+            self.status = api_status['status']
+
         return self.status
 
     @api_method

setup.py

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