File tree Expand file tree Collapse file tree 2 files changed +11
-2
lines changed Expand file tree Collapse file tree 2 files changed +11
-2
lines changed Original file line number Diff line number Diff line change 16
16
17
17
18
18
def validate_payload (task_type , kwargs ):
19
- allowed_fields = DEFAULT_FIELDS + ALLOWED_FIELDS [task_type ]
19
+ allowed_fields = DEFAULT_FIELDS | ALLOWED_FIELDS [task_type ]
20
20
for k in kwargs :
21
21
if k not in allowed_fields :
22
22
raise ScaleInvalidRequest ('Illegal parameter %s for task_type %s'
@@ -88,7 +88,7 @@ def cancel_task(self, task_id):
88
88
89
89
def tasks (self ):
90
90
"""Returns a list of all your tasks."""
91
- return [Task (json , self ) for json in self ._getrequest ('tasks' )]
91
+ return [Task (json , self ) for json in self ._getrequest ('tasks' )[ 'docs' ] ]
92
92
93
93
def create_categorization_task (self , ** kwargs ):
94
94
validate_payload ('categorization' , kwargs )
Original file line number Diff line number Diff line change @@ -11,5 +11,14 @@ def __getattr__(self, name):
11
11
raise AttributeError ("'%s' object has no attribute %s"
12
12
% (type (self ).__name__ , name ))
13
13
14
+ def __str__ (self ):
15
+ return 'Task(id=%s)' % self .id
16
+
17
+ def __repr__ (self ):
18
+ return 'Task(%s)' % self .param_dict
19
+
20
+ def refresh (self ):
21
+ self .param_dict = self .client ._getrequest ('task/%s' % self .id )
22
+
14
23
def cancel (self ):
15
24
self .client .cancel_task (self .id )
You can’t perform that action at this time.
0 commit comments