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