def create_task(self, type, application, document):
task=Task()
task.type = type
task.application = application
if type == 'createdocument':
task.document = document
elif type == 'classifydocument':
task.classify = document
else:
raise Exception('Not a valid task type.')
task.save()
def create(self, validated_data):
docset = super(DocSetSerializer, self).create(validated_data)
application = Application.objects.filter(id=validated_data['application_id']).first()
# Create a task to finish the work
task=Task()
task.type = 'createdocset'
task.application = application
task.docset = docset
task.save()
return docset
def create(self, validated_data):
classifier = super(ClassifierSerializer, self).create(validated_data)
application = Application.objects.filter(id=validated_data['application_id']).first()
# Create a task to finish the work
task=Task()
task.type = 'createclassifier'
task.application = application
task.classifier = classifier
task.save()
return classifier
