thestage 0.5.46__py3-none-any.whl → 0.5.49__py3-none-any.whl

thestage/services/clients/thestage_api/dtos/project_controller/project_start_inference_simulator_response.py

@@ -1,10 +1,10 @@
-from pydantic import Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
-
-
-class ProjectStartInferenceSimulatorResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    inferenceSimulator: InferenceSimulatorDto = Field(None, alias='inferenceSimulator')
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
+
+
+class ProjectStartInferenceSimulatorResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    inferenceSimulator: InferenceSimulatorDto = Field(None, alias='inferenceSimulator')

thestage/services/clients/thestage_api/dtos/project_response.py

@@ -1,32 +1,32 @@
-from typing import Optional, List
-
-from pydantic import Field, BaseModel, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.clients.thestage_api.dtos.selfhosted_instance_response import SelfHostedInstanceDto
-from thestage.services.clients.thestage_api.dtos.instance_rented_response import InstanceRentedDto
-
-
-class ProjectDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    id: Optional[int] = Field(None, alias='id')
-    client_id: Optional[int] = Field(None, alias='сlientId')
-    ssh_key_deploy_id: Optional[int] = Field(None, alias='sshKeyDeployId')
-    name: Optional[str] = Field(None, alias='name')
-    slug: Optional[str] = Field(None, alias='slug')
-    description: Optional[str] = Field(None, alias='description')
-    github_username: Optional[str] = Field(None, alias='githubCollaboratorUsername')
-    last_commit_hash: Optional[str] = Field(None, alias='lastCommitHash')
-    last_commit_description: Optional[str] = Field(None, alias='lastCommitDescription')
-    git_repository_url: Optional[str] = Field(None, alias='gitRepositoryUrl')
-    git_repository_name: Optional[str] = Field(None, alias='gitRepositoryName')
-    task_count: Optional[int] = Field(None, alias='taskCount')
-    favourite_task_count: Optional[int] = Field(None, alias='favouriteTaskCount')
-    last_task_run_date: Optional[str] = Field(None, alias='lastTaskRunDate')
-    created_at: Optional[str] = Field(None, alias='createdAt')
-    updated_at: Optional[str] = Field(None, alias='updatedAt')
-
-
-class ProjectViewResponse(TheStageBaseResponse):
-    project: Optional[ProjectDto] = Field(None, alias='project')
+from typing import Optional, List
+
+from pydantic import Field, BaseModel, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.clients.thestage_api.dtos.selfhosted_instance_response import SelfHostedInstanceDto
+from thestage.services.clients.thestage_api.dtos.instance_rented_response import InstanceRentedDto
+
+
+class ProjectDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    id: Optional[int] = Field(None, alias='id')
+    client_id: Optional[int] = Field(None, alias='сlientId')
+    ssh_key_deploy_id: Optional[int] = Field(None, alias='sshKeyDeployId')
+    name: Optional[str] = Field(None, alias='name')
+    slug: Optional[str] = Field(None, alias='slug')
+    description: Optional[str] = Field(None, alias='description')
+    github_username: Optional[str] = Field(None, alias='githubCollaboratorUsername')
+    last_commit_hash: Optional[str] = Field(None, alias='lastCommitHash')
+    last_commit_description: Optional[str] = Field(None, alias='lastCommitDescription')
+    git_repository_url: Optional[str] = Field(None, alias='gitRepositoryUrl')
+    git_repository_name: Optional[str] = Field(None, alias='gitRepositoryName')
+    task_count: Optional[int] = Field(None, alias='taskCount')
+    favourite_task_count: Optional[int] = Field(None, alias='favouriteTaskCount')
+    last_task_run_date: Optional[str] = Field(None, alias='lastTaskRunDate')
+    created_at: Optional[str] = Field(None, alias='createdAt')
+    updated_at: Optional[str] = Field(None, alias='updatedAt')
+
+
+class ProjectViewResponse(TheStageBaseResponse):
+    project: Optional[ProjectDto] = Field(None, alias='project')

thestage/services/clients/thestage_api/dtos/selfhosted_instance_response.py

@@ -1,56 +1,56 @@
-from datetime import datetime
-from typing import Optional, List, Dict
-
-from pydantic import Field, BaseModel, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBasePaginatedResponse
-from thestage.services.clients.thestage_api.dtos.frontend_status import FrontendStatusDto
-from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
-from thestage.services.clients.thestage_api.dtos.pagination_data import PaginationData
-from thestage.services.clients.thestage_api.dtos.instance_detected_gpus import InstanceDetectedGpusDto
-from thestage.services.clients.thestage_api.dtos.enums.instance_type import InstanceTypeEnumDto
-from thestage.services.clients.thestage_api.dtos.enums.selfhosted_status import SelfhostedBusinessStatus
-from thestage.services.clients.thestage_api.dtos.enums.cpu_type import InstanceCpuType
-
-
-class SelfHostedInstanceDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    instance_type: Optional[InstanceTypeEnumDto] = Field(InstanceTypeEnumDto.UNKNOWN, alias='instanceType')
-    id: Optional[int] = Field(None, alias='id')
-    ram_size_gb: Optional[int] = Field(None, alias='ramSizeGb')
-    client_id: Optional[int] = Field(None, alias='clientId')
-    slug: Optional[str] = Field(None, alias='slug')
-    title: Optional[str] = Field(None, alias='title')
-    cpu_type: Optional[InstanceCpuType] = Field(InstanceCpuType.UNKNOWN, alias='cpuType')
-    cpu_name: Optional[str] = Field(None, alias='cpuName')
-    cpu_cores: Optional[int] = Field(None, alias='cpuCores')
-    gpu_count: Optional[int] = Field(None, alias='gpuCount')
-    detected_gpus: Optional[InstanceDetectedGpusDto] = Field(None, alias='detectedGpus')
-    hardware_profile_id: Optional[str] = Field(None, alias='hardwareProfileId')
-    disk_size_gb: Optional[int] = Field(None, alias='diskSizeGb')
-
-    frontend_status: Optional[FrontendStatusDto] = Field(None, alias='frontendStatus')
-    business_status: SelfhostedBusinessStatus = Field(SelfhostedBusinessStatus.UNKNOWN, alias='businessStatus')
-
-    updated_at: Optional[datetime] = Field(None, alias='updatedAt')
-    created_at: Optional[datetime] = Field(None, alias='createdAt')
-    ip_address: Optional[str] = Field(None, alias='ipAddress')
-    # not present now
-    host_username: Optional[str] = Field(None, alias='hostUsername')
-
-
-class SelfHostedInstanceListResponse(TheStageBasePaginatedResponse):
-    paginated_list: Optional[PaginatedEntityList[SelfHostedInstanceDto]] = Field(None, alias='selfHostedInstanceList')
-
-
-class SelfHostedRentedItemResponse(TheStageBasePaginatedResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    selfhosted_instance: Optional[SelfHostedInstanceDto] = Field(None, alias='selfhostedInstance')
-
-
-class SelfHostedRentedRentedBusinessStatusMapperResponse(TheStageBasePaginatedResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    selfhosted_instance_business_status_map: Dict[str, str] = Field(default=dict, alias='selfhostedInstanceBusinessStatusMap')
+from datetime import datetime
+from typing import Optional, List, Dict
+
+from pydantic import Field, BaseModel, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBasePaginatedResponse
+from thestage.services.clients.thestage_api.dtos.frontend_status import FrontendStatusDto
+from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
+from thestage.services.clients.thestage_api.dtos.pagination_data import PaginationData
+from thestage.services.clients.thestage_api.dtos.instance_detected_gpus import InstanceDetectedGpusDto
+from thestage.services.clients.thestage_api.dtos.enums.instance_type import InstanceTypeEnumDto
+from thestage.services.clients.thestage_api.dtos.enums.selfhosted_status import SelfhostedBusinessStatus
+from thestage.services.clients.thestage_api.dtos.enums.cpu_type import InstanceCpuType
+
+
+class SelfHostedInstanceDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    instance_type: Optional[InstanceTypeEnumDto] = Field(InstanceTypeEnumDto.UNKNOWN, alias='instanceType')
+    id: Optional[int] = Field(None, alias='id')
+    ram_size_gb: Optional[int] = Field(None, alias='ramSizeGb')
+    client_id: Optional[int] = Field(None, alias='clientId')
+    slug: Optional[str] = Field(None, alias='slug')
+    title: Optional[str] = Field(None, alias='title')
+    cpu_type: Optional[InstanceCpuType] = Field(InstanceCpuType.UNKNOWN, alias='cpuType')
+    cpu_name: Optional[str] = Field(None, alias='cpuName')
+    cpu_cores: Optional[int] = Field(None, alias='cpuCores')
+    gpu_count: Optional[int] = Field(None, alias='gpuCount')
+    detected_gpus: Optional[InstanceDetectedGpusDto] = Field(None, alias='detectedGpus')
+    hardware_profile_id: Optional[str] = Field(None, alias='hardwareProfileId')
+    disk_size_gb: Optional[int] = Field(None, alias='diskSizeGb')
+
+    frontend_status: Optional[FrontendStatusDto] = Field(None, alias='frontendStatus')
+    business_status: SelfhostedBusinessStatus = Field(SelfhostedBusinessStatus.UNKNOWN, alias='businessStatus')
+
+    updated_at: Optional[datetime] = Field(None, alias='updatedAt')
+    created_at: Optional[datetime] = Field(None, alias='createdAt')
+    ip_address: Optional[str] = Field(None, alias='ipAddress')
+    # not present now
+    host_username: Optional[str] = Field(None, alias='hostUsername')
+
+
+class SelfHostedInstanceListResponse(TheStageBasePaginatedResponse):
+    paginated_list: Optional[PaginatedEntityList[SelfHostedInstanceDto]] = Field(None, alias='selfHostedInstanceList')
+
+
+class SelfHostedRentedItemResponse(TheStageBasePaginatedResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    selfhosted_instance: Optional[SelfHostedInstanceDto] = Field(None, alias='selfhostedInstance')
+
+
+class SelfHostedRentedRentedBusinessStatusMapperResponse(TheStageBasePaginatedResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    selfhosted_instance_business_status_map: Dict[str, str] = Field(default={}, alias='selfhostedInstanceBusinessStatusMap')

thestage/services/clients/thestage_api/dtos/sftp_path_helper.py

@@ -1,13 +1,13 @@
-from typing import Optional, List
-
-from pydantic import Field
-
-from thestage.entities.file_item import FileItemEntity
-
-
-class SftpFileItemEntity(FileItemEntity):
-
-    instance_path: Optional[str] = Field(None)
-    container_path: Optional[str] = Field(None)
-    children: List['SftpFileItemEntity'] = Field(default=[])
-    dest_path: Optional[str] = Field(None)
+from typing import Optional, List
+
+from pydantic import Field
+
+from thestage.entities.file_item import FileItemEntity
+
+
+class SftpFileItemEntity(FileItemEntity):
+
+    instance_path: Optional[str] = Field(None)
+    container_path: Optional[str] = Field(None)
+    children: List['SftpFileItemEntity'] = Field(default=[])
+    dest_path: Optional[str] = Field(None)

thestage/services/clients/thestage_api/dtos/ssh_key_controller/add_ssh_key_to_user_request.py

@@ -1,8 +1,8 @@
-from pydantic import Field, ConfigDict, BaseModel
-
-
-class AddSshKeyToUserRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    sshKey: str = Field(None, alias='sshKey')
-    note: str = Field(None, alias='note')
+from pydantic import Field, ConfigDict, BaseModel
+
+
+class AddSshKeyToUserRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    sshKey: str = Field(None, alias='sshKey')
+    note: str = Field(None, alias='note')

thestage/services/clients/thestage_api/dtos/ssh_key_controller/add_ssh_key_to_user_response.py

@@ -1,12 +1,12 @@
-from typing import Optional
-
-from pydantic import Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.task.dto.task_dto import TaskDto
-
-
-class AddSshKeyToUserResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
+from typing import Optional
+
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.task.dto.task_dto import TaskDto
+
+
+class AddSshKeyToUserResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
     sshKeyPairId: Optional[int] = Field(None, alias='sshKeyPairId')

thestage/services/clients/thestage_api/dtos/ssh_key_controller/add_ssh_public_key_to_instance_request.py

@@ -1,8 +1,8 @@
-from pydantic import Field, ConfigDict, BaseModel
-
-
-class AddSshPublicKeyToInstanceRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    instanceRentedId: int = Field(None, alias='instanceRentedId')
-    sshPublicKeyId: int = Field(None, alias='sshPublicKeyId')
+from pydantic import Field, ConfigDict, BaseModel
+
+
+class AddSshPublicKeyToInstanceRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    instanceRentedId: int = Field(None, alias='instanceRentedId')
+    sshPublicKeyId: int = Field(None, alias='sshPublicKeyId')

thestage/services/clients/thestage_api/dtos/ssh_key_controller/add_ssh_public_key_to_instance_response.py

@@ -1,11 +1,11 @@
-from typing import Optional
-
-from pydantic import Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.task.dto.task_dto import TaskDto
-
-
-class AddSshPublicKeyToInstanceResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
+from typing import Optional
+
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.task.dto.task_dto import TaskDto
+
+
+class AddSshPublicKeyToInstanceResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+

thestage/services/clients/thestage_api/dtos/ssh_key_controller/is_user_has_public_ssh_key_request.py

@@ -1,7 +1,7 @@
-from pydantic import Field, ConfigDict, BaseModel
-
-
-class IsUserHasSshPublicKeyRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    sshKey: str = Field(None, alias='sshKey')
+from pydantic import Field, ConfigDict, BaseModel
+
+
+class IsUserHasSshPublicKeyRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    sshKey: str = Field(None, alias='sshKey')

thestage/services/clients/thestage_api/dtos/ssh_key_controller/is_user_has_public_ssh_key_response.py

@@ -1,12 +1,12 @@
-from typing import Optional
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-
-
-class IsUserHasSshPublicKeyResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    isUserHasPublicKey: bool = Field(None, alias='isUserHasPublicKey')
-    sshKeyPairId: Optional[int] = Field(None, alias='sshKeyPairId')
+from typing import Optional
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+
+
+class IsUserHasSshPublicKeyResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    isUserHasPublicKey: bool = Field(None, alias='isUserHasPublicKey')
+    sshKeyPairId: Optional[int] = Field(None, alias='sshKeyPairId')

thestage/services/clients/thestage_api/dtos/task_controller/task_list_for_project_request.py

@@ -1,10 +1,10 @@
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
-
-
-class TaskListForProjectRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
-    projectSlug: str = Field(None, alias='projectSlug')
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+
+
+class TaskListForProjectRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
+    projectSlug: str = Field(None, alias='projectSlug')

thestage/services/clients/thestage_api/dtos/task_controller/task_list_for_project_response.py

@@ -1,12 +1,12 @@
-
-from pydantic import Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
-from thestage.services.task.dto.task_dto import TaskDto
-
-
-class TaskListForProjectResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    tasks: PaginatedEntityList[TaskDto] = Field(None, alias='tasks')
+
+from pydantic import Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
+from thestage.services.task.dto.task_dto import TaskDto
+
+
+class TaskListForProjectResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    tasks: PaginatedEntityList[TaskDto] = Field(None, alias='tasks')

thestage/services/clients/thestage_api/dtos/task_controller/task_status_localized_map_response.py

@@ -1,9 +1,9 @@
-from typing import Dict
-
-from pydantic import Field
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-
-
-class TaskStatusLocalizedMapResponse(TheStageBaseResponse):
-    taskStatusMap: Dict[str, str] = Field(default=dict, alias='taskStatusMap')
+from typing import Dict
+
+from pydantic import Field
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+
+
+class TaskStatusLocalizedMapResponse(TheStageBaseResponse):
+    taskStatusMap: Dict[str, str] = Field(default={}, alias='taskStatusMap')

thestage/services/clients/thestage_api/dtos/task_controller/task_view_response.py

@@ -1,12 +1,12 @@
-from typing import Optional
-
-from pydantic import Field
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.task.dto.task_dto import TaskDto
-
-
-class TaskViewResponse(TheStageBaseResponse):
-    task: Optional[TaskDto] = Field(None, alias='task')
-    task_view_url: Optional[str] = Field(None, alias='websiteTaskViewUrl')
-    task_output_url: Optional[str] = Field(None, alias='websiteTaskOutputUrl')
+from typing import Optional
+
+from pydantic import Field
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.task.dto.task_dto import TaskDto
+
+
+class TaskViewResponse(TheStageBaseResponse):
+    task: Optional[TaskDto] = Field(None, alias='task')
+    task_view_url: Optional[str] = Field(None, alias='websiteTaskViewUrl')
+    task_output_url: Optional[str] = Field(None, alias='websiteTaskOutputUrl')

thestage/services/clients/thestage_api/dtos/user_controller/user_profile.py

@@ -1,12 +1,12 @@
-from typing import Optional
-
-from pydantic import BaseModel, Field
-
-
-class UserProfile(BaseModel):
-    email: Optional[str] = Field(None, alias='email')
-
-
-
-class UserProfileResponse(BaseModel):
-    userProfile: Optional[UserProfile] = Field(None, alias='userProfile')
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class UserProfile(BaseModel):
+    email: Optional[str] = Field(None, alias='email')
+
+
+
+class UserProfileResponse(BaseModel):
+    userProfile: Optional[UserProfile] = Field(None, alias='userProfile')

thestage/services/clients/thestage_api/dtos/validate_token_response.py

@@ -0,0 +1,11 @@
+from typing import Optional, List, Dict
+
+from pydantic import Field, BaseModel, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse, TheStageBasePaginatedResponse
+
+
+class ValidateTokenResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    allowedCliCommands: Optional[Dict[str, bool]] = Field(default={}, alias='allowedCliCommands')