<reponame>colinRawlings/osparc-simcore
from typing import List

from models_library.services import PropertyName
from pydantic import BaseModel, Field

from ..schemas.dynamic_services import RunningDynamicServiceDetails, ServiceDetails


class RetrieveDataIn(BaseModel):
    port_keys: List[PropertyName] = Field(
        ..., description="The port keys to retrieve data from"
    )


class RetrieveDataOut(BaseModel):
    size_bytes: int = Field(
        ..., description="The amount of data transferred by the retrieve call"
    )


class RetrieveDataOutEnveloped(BaseModel):
    data: RetrieveDataOut


DynamicServiceCreate = ServiceDetails
DynamicServiceOut = RunningDynamicServiceDetails

from typing import List

from models_library.services import PropertyName
from pydantic import BaseModel, Field

from ..schemas.dynamic_services import RunningDynamicServiceDetails, ServiceDetails


class RetrieveDataIn(BaseModel):
    port_keys: List[PropertyName] = Field(
        ..., description="The port keys to retrieve data from"
    )


class RetrieveDataOut(BaseModel):
    size_bytes: int = Field(
        ..., description="The amount of data transferred by the retrieve call"
    )


class RetrieveDataOutEnveloped(BaseModel):
    data: RetrieveDataOut


DynamicServiceCreate = ServiceDetails
DynamicServiceOut = RunningDynamicServiceDetails