Fix batch operations results

This commit is contained in:
allegroai
2021-05-03 18:07:37 +03:00
parent dc9623e964
commit 6b0c45a861
11 changed files with 368 additions and 278 deletions

View File

@@ -3,13 +3,12 @@ from six import string_types
from apiserver.apimodels import ListField, DictField
from apiserver.apimodels.base import UpdateResponse
from apiserver.apimodels.batch import BatchRequest, BatchResponse
from apiserver.apimodels.batch import BatchRequest
from apiserver.apimodels.metadata import (
MetadataItem,
DeleteMetadata,
AddOrUpdateMetadata,
)
from apiserver.apimodels.tasks import PublishResponse as TaskPublishResponse
class GetFrameworksRequest(models.Base):
@@ -51,10 +50,6 @@ class ModelsDeleteManyRequest(BatchRequest):
force = fields.BoolField(default=False)
class ModelsDeleteManyResponse(BatchResponse):
urls = fields.ListField([str])
class PublishModelRequest(ModelRequest):
force_publish_task = fields.BoolField(default=False)
publish_task = fields.BoolField(default=True)
@@ -62,7 +57,7 @@ class PublishModelRequest(ModelRequest):
class ModelTaskPublishResponse(models.Base):
id = fields.StringField(required=True)
data = fields.EmbeddedField(TaskPublishResponse)
data = fields.EmbeddedField(UpdateResponse)
class PublishModelResponse(UpdateResponse):
@@ -74,10 +69,6 @@ class ModelsPublishManyRequest(BatchRequest):
publish_task = fields.BoolField(default=True)
class ModelsPublishManyResponse(BatchResponse):
published_tasks = fields.ListField([ModelTaskPublishResponse])
class DeleteMetadataRequest(DeleteMetadata):
model = fields.StringField(required=True)