Text-analytics: new get_analytics get_annotated functions

Author: dariosky

README.md

@@ -33,6 +33,8 @@ api = RPApi(api_key="YOUR_API_KEY")
 To create a dataset you can call the `create_dataset` method of the API with a Dataset instance.
 
 ```python
+from ravenpackapi import Dataset
+
 ds = api.create_dataset(
     Dataset(
         name="New Dataset",

ravenpackapi/upload/models.py

@@ -73,6 +73,7 @@ def save_original(self, filename):
 
     @api_method
     def save_analytics(self, filename, output_format='application/json'):
+        self.wait_for_completion()
         response = self.api.request('%s/files/%s/analytics' % (self.api._UPLOAD_BASE_URL, self.file_id,),
                                     headers=dict(
                                         Accept=output_format,
@@ -83,14 +84,34 @@ def save_analytics(self, filename, output_format='application/json'):
             for chunk in response.iter_content(chunk_size=self.api._CHUNK_SIZE):
                 f.write(chunk)
 
+    @api_method
+    def get_analytics(self, output_format='application/json'):
+        self.wait_for_completion()
+        response = self.api.request('%s/files/%s/analytics' % (self.api._UPLOAD_BASE_URL, self.file_id,),
+                                    headers=dict(
+                                        Accept=output_format,
+                                        **self.api.headers
+                                    ))
+        if output_format == 'application/json':
+            return response.json()
+        else:
+            return response.text
+
     @api_method
     def save_annotated(self, filename):
+        self.wait_for_completion()
         response = self.api.request('%s/files/%s/annotated' % (self.api._UPLOAD_BASE_URL, self.file_id),
                                     stream=True)
         with open(filename, 'wb') as f:
             for chunk in response.iter_content(chunk_size=self.api._CHUNK_SIZE):
                 f.write(chunk)
 
+    @api_method
+    def get_annotated(self):
+        self.wait_for_completion()
+        response = self.api.request('%s/files/%s/annotated' % (self.api._UPLOAD_BASE_URL, self.file_id))
+        return response.text
+
     @api_method
     def delete(self):
         response = self.api.request('%s/files/%s' % (self.api._UPLOAD_BASE_URL, self.file_id),
@@ -114,7 +135,7 @@ def set_metadata(self,
                          method='patch')
 
     def wait_for_completion(self):
-        while self.status not in {"COMPLETED", "DELETED"}:
+        while self.status not in {"COMPLETED", "DELETED", "FAILED"}:
             sleep(1)
             self.get_status()