mirror of
https://github.com/clearml/clearml-server
synced 2025-01-31 19:06:55 +00:00
81 lines
1.9 KiB
Python
81 lines
1.9 KiB
Python
from jsonmodels import validators
|
|
from jsonmodels.fields import StringField, IntField, BoolField, FloatField
|
|
from jsonmodels.models import Base
|
|
|
|
from apiserver.apimodels import ListField, DictField
|
|
from apiserver.apimodels.metadata import (
|
|
MetadataItem,
|
|
DeleteMetadata,
|
|
AddOrUpdateMetadata,
|
|
)
|
|
|
|
|
|
class GetDefaultResp(Base):
|
|
id = StringField(required=True)
|
|
name = StringField(required=True)
|
|
|
|
|
|
class CreateRequest(Base):
|
|
name = StringField(required=True)
|
|
tags = ListField(items_types=[str])
|
|
system_tags = ListField(items_types=[str])
|
|
metadata = DictField(value_types=[MetadataItem])
|
|
|
|
|
|
class QueueRequest(Base):
|
|
queue = StringField(required=True)
|
|
|
|
|
|
class GetNextTaskRequest(QueueRequest):
|
|
queue = StringField(required=True)
|
|
get_task_info = BoolField(default=False)
|
|
|
|
|
|
class DeleteRequest(QueueRequest):
|
|
force = BoolField(default=False)
|
|
|
|
|
|
class UpdateRequest(QueueRequest):
|
|
name = StringField()
|
|
tags = ListField(items_types=[str])
|
|
system_tags = ListField(items_types=[str])
|
|
metadata = DictField(value_types=[MetadataItem])
|
|
|
|
|
|
class TaskRequest(QueueRequest):
|
|
task = StringField(required=True)
|
|
|
|
|
|
class MoveTaskRequest(TaskRequest):
|
|
count = IntField(default=1)
|
|
|
|
|
|
class MoveTaskResponse(Base):
|
|
position = IntField()
|
|
|
|
|
|
class GetMetricsRequest(Base):
|
|
queue_ids = ListField([str])
|
|
from_date = FloatField(required=True, validators=validators.Min(0))
|
|
to_date = FloatField(required=True, validators=validators.Min(0))
|
|
interval = IntField(required=True, validators=validators.Min(1))
|
|
|
|
|
|
class QueueMetrics(Base):
|
|
queue = StringField()
|
|
dates = ListField(int)
|
|
avg_waiting_times = ListField([float, int])
|
|
queue_lengths = ListField(int)
|
|
|
|
|
|
class GetMetricsResponse(Base):
|
|
queues = ListField(QueueMetrics)
|
|
|
|
|
|
class DeleteMetadataRequest(DeleteMetadata):
|
|
queue = StringField(required=True)
|
|
|
|
|
|
class AddOrUpdateMetadataRequest(AddOrUpdateMetadata):
|
|
queue = StringField(required=True)
|