adds ability to add custom corpora

Author: Joshua B. Smith

test/test_speech_to_text_v1.py

@@ -1,5 +1,5 @@
 # coding=utf-8
-import os,json
+import os,json,pytest
 import responses
 import watson_developer_cloud
 
@@ -98,3 +98,45 @@ def test_custom_model():
 
     assert len(responses.calls) == 5
 
+@responses.activate
+def test_custom_corpora():
+
+    corpora_url = 'https://stream.watsonplatform.net/speech-to-text/api/v1/customizations/{0}/corpora'
+    get_corpora_url = '{0}/{1}'.format(corpora_url.format('customid'),'corpus')
+
+    responses.add(responses.GET, corpora_url.format('customid'),
+                  body='{"get response": "yep"}', status=200,
+                  content_type='application/json')
+
+    responses.add(responses.POST, get_corpora_url,
+                 body='{"get response": "yep"}',
+                 status=200,
+                 content_type='application/json')
+
+    responses.add(responses.GET, get_corpora_url,
+                 body='{"get response": "yep"}',
+                 status=200,
+                 content_type='application/json')
+
+    responses.add(responses.DELETE, get_corpora_url,
+                  body='{"get response": "yep"}',
+                  status=200,
+                  content_type='application/json')
+
+    speech_to_text = watson_developer_cloud.SpeechToTextV1(
+        username="username", password="password")
+
+    speech_to_text.list_corpora(customization_id='customid')
+
+    file_path = './resources/speech_to_text/corpus-short-1.txt'
+    with open(file_path) as f:
+        speech_to_text.add_corpora(customization_id='customid',
+                                   corpus_name="corpus",file=f)
+
+    speech_to_text.get_corpus(customization_id='customid',
+                              corpus_name='corpus')
+
+    speech_to_text.delete_corpus(customization_id='customid',
+                                 corpus_name='corpus')
+
+    assert len(responses.calls) == 4

watson_developer_cloud/speech_to_text_v1.py

@@ -18,6 +18,7 @@
 
 from .watson_developer_cloud_service import WatsonDeveloperCloudService
 import json
+from collections import namedtuple
 
 
 class SpeechToTextV1(WatsonDeveloperCloudService):
@@ -96,3 +97,44 @@ def delete_custom_model(self, modelid):
         return self.request(method='DELETE',
                             url='/v1/customizations/{0}'.format(modelid),
                             accept_json=True)
+
+    def list_corpora(self, customization_id):
+        url = '/v1/customizations/{0}/corpora'
+        return self.request(method='GET',
+                            url=url.format(customization_id),
+                            accept_json=True)
+
+    def add_corpora(self,
+                    customization_id,
+                    corpus_name,
+                    file,
+                    allow_overwrite=None):
+
+        url = '/v1/customizations/{0}/corpora/{1}'
+
+        if allow_overwrite is None:
+            allow_overwrite = False
+
+        headers = {'Content-Type': 'application/octet-stream'}
+
+        return self.request(method='GET',
+                            url=url.format(customization_id,
+                                           corpus_name),
+                            headers=headers,
+                            data=file,
+                            params={'allow_overwrite': allow_overwrite},
+                            accept_json=True)
+
+    def get_corpus(self, customization_id, corpus_name):
+        url = '/v1/customizations/{0}/corpora/{1}'
+        return self.request(method='GET',
+                            url=url.format(customization_id,
+                                           corpus_name),
+                            accept_json=True)
+
+    def delete_corpus(self, customization_id, corpus_name):
+        url = '/v1/customizations/{0}/corpora/{1}'
+        return self.request(method='DELETE',
+                            url=url.format(customization_id,
+                                           corpus_name),
+                            accept_json=True)