add datacollection

Author: clhuang

README.rst

@@ -118,6 +118,23 @@ __ https://docs.scaleapi.com/#create-annotation-task-bounding-box
       objects_to_annotate=["baby cow", "big cow"]
     )
 
+Create datacollection task
+=========================
+
+Check `this`__ for further information.
+
+__ https://docs.scaleapi.com/#create-data-collection-task
+
+.. code-block:: python
+
+    task = client.create_datacollection_task(
+      callback_url='http://www.example.com/callback',
+      instruction='Find the URL for the hiring page for the company with attached website.',
+      attachment_type='website',
+      attachment='http://www.google.com/',
+      fields={ 'hiring_page': 'Hiring Page URL' },
+    )
+
 Retrieve task
 =============
 

scaleapi/__init__.py

@@ -12,7 +12,8 @@
                   'comparison': {'attachments', 'attachment_type',
                                  'fields', 'choices'},
                   'annotation': {'attachment', 'attachment_type', 'instruction',
-                                 'objects_to_annotate', 'with_labels'}}
+                                 'objects_to_annotate', 'with_labels'},
+                  'datacollection': {'attachment', 'attachment_type', 'fields'}}
 SCALE_ENDPOINT = 'https://api.scaleapi.com/v1/'
 
 
@@ -128,3 +129,8 @@ def create_annotation_task(self, **kwargs):
         validate_payload('annotation', kwargs)
         taskdata = self._postrequest('task/annotation', payload=kwargs)
         return Task(taskdata, self)
+
+    def create_datacollection_task(self, **kwargs):
+        validate_payload('datacollection', kwargs)
+        taskdata = self._postrequest('task/datacollection', payload=kwargs)
+        return Task(taskdata, self)

tests/test_client.py

@@ -39,7 +39,7 @@ def test_categorize_fail():
             categories=['public', 'private'])
 
 
-def test_transcrption_ok():
+def test_transcription_ok():
     task = client.create_transcription_task(
         callback_url='http://www.example.com/callback',
         instruction='Transcribe the given fields. Then for each news item on the page, transcribe the information for the row.',
@@ -118,6 +118,22 @@ def test_annotation_fail():
             attachment_type='image')
 
 
+def test_datacollection_ok():
+    task = client.create_datacollection_task(
+        callback_url='http://www.example.com/callback',
+        instruction='Find the URL for the hiring page for the company with attached website.',
+        attachment_type='website',
+        attachment='http://www.google.com/',
+        fields={ 'hiring_page': 'Hiring Page URL' })
+
+
+def test_datacollection_fail():
+    with pytest.raises(scaleapi.ScaleInvalidRequest):
+        client.create_datacollection_task(
+            callback_url='http://www.example.com/callback',
+            attachment_type='website')
+
+
 def test_cancel():
     task = make_a_task()
     # raises a scaleexception, because test tasks complete instantly