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

thestage/services/clients/thestage_api/core/api_client_core.py

@@ -1,108 +1,108 @@
-import json
-from typing import Optional
-
-import requests
-
-from thestage.config import THESTAGE_API_URL
-from thestage.exceptions.auth_exception import AuthException
-from thestage import __version__
-from thestage.services.clients.thestage_api.core.http_client_exception import HttpClientException
-from thestage.services.clients.thestage_api.dtos.validate_token_response import ValidateTokenResponse
-
-
-class TheStageApiClientCore:
-    def __init__(self, url: Optional[str] = None):
-        self.__api_url = url
-
-
-    def validate_token(self, token: str) -> ValidateTokenResponse:
-        data = {
-            "userApiToken": token,
-            "cliVersion": __version__,
-        }
-
-        response = self._request(
-            method='POST',
-            url='/user-api/v1/validate-token',
-            data=data,
-        )
-        result = ValidateTokenResponse.model_validate(response) if response else None
-        return result
-
-
-    def _get_host(self, ) -> str:
-        return self.__api_url or THESTAGE_API_URL
-
-    def _request(
-            self,
-            method: str,
-            url: str,
-            data: dict = None,
-            query_params: dict = None,
-            headers: Optional[dict] = None,
-            token: Optional[str] = None,
-    ):
-        if not data:
-            data = {}
-
-        host = self._get_host()
-        url = f'{host}{url}'
-
-        request_headers = {
-            'Content-Type': 'application/json',
-        }
-
-        if token:
-            request_headers['Authorization'] = f"Bearer {token}"
-
-        if headers:
-            request_headers.update(headers)
-        response = requests.request(
-            method=method,
-            url=url,
-            json=data,
-            params=query_params,
-            headers=request_headers,
-            timeout=90,
-
-        )
-        return self._parse_api_response(response)
-
-
-    @staticmethod
-    def _parse_api_response(raw_response):
-        content_type = raw_response.headers.get('content-type')
-        message_error = None
-        if content_type == 'application/json':
-            try:
-                result = raw_response.json()
-                message_error = result.get('message', None)
-            except json.JSONDecodeError:
-                raise HttpClientException(
-                    message=f"Failed to parse server response",
-                    status_code=raw_response.status_code,
-                )
-        else:
-            result = raw_response.content.text()
-
-        if raw_response.status_code == 401:
-            raise AuthException(
-                message=f"Unauthorized",
-            )
-        elif raw_response.status_code == 403:
-            raise HttpClientException(
-                message=f"{message_error if message_error else 'Forbidden'} ({raw_response.status_code})",
-                status_code=raw_response.status_code,
-            )
-        elif raw_response.status_code >= 400:
-            raise HttpClientException(
-                message=f"{message_error if message_error else 'Request error'} ({raw_response.status_code})",
-                status_code=raw_response.status_code,
-            )
-        elif raw_response.status_code < 200 or raw_response.status_code > 300:
-            raise HttpClientException(
-                message=f"{message_error if message_error else 'Request error'} ({raw_response.status_code})",
-                status_code=raw_response.status_code,
-            )
-
-        return result
+import json
+from typing import Optional
+
+import requests
+
+from thestage.config import THESTAGE_API_URL
+from thestage.exceptions.auth_exception import AuthException
+from thestage import __version__
+from thestage.services.clients.thestage_api.core.http_client_exception import HttpClientException
+from thestage.services.clients.thestage_api.dtos.validate_token_response import ValidateTokenResponse
+
+
+class TheStageApiClientCore:
+    def __init__(self, url: Optional[str] = None):
+        self.__api_url = url
+
+
+    def validate_token(self, token: str) -> ValidateTokenResponse:
+        data = {
+            "userApiToken": token,
+            "cliVersion": __version__,
+        }
+
+        response = self._request(
+            method='POST',
+            url='/user-api/v1/validate-token',
+            data=data,
+        )
+        result = ValidateTokenResponse.model_validate(response) if response else None
+        return result
+
+
+    def _get_host(self, ) -> str:
+        return self.__api_url or THESTAGE_API_URL
+
+    def _request(
+            self,
+            method: str,
+            url: str,
+            data: dict = None,
+            query_params: dict = None,
+            headers: Optional[dict] = None,
+            token: Optional[str] = None,
+    ):
+        if not data:
+            data = {}
+
+        host = self._get_host()
+        url = f'{host}{url}'
+
+        request_headers = {
+            'Content-Type': 'application/json',
+        }
+
+        if token:
+            request_headers['Authorization'] = f"Bearer {token}"
+
+        if headers:
+            request_headers.update(headers)
+        response = requests.request(
+            method=method,
+            url=url,
+            json=data,
+            params=query_params,
+            headers=request_headers,
+            timeout=90,
+
+        )
+        return self._parse_api_response(response)
+
+
+    @staticmethod
+    def _parse_api_response(raw_response):
+        content_type = raw_response.headers.get('content-type')
+        message_error = None
+        if content_type == 'application/json':
+            try:
+                result = raw_response.json()
+                message_error = result.get('message', None)
+            except json.JSONDecodeError:
+                raise HttpClientException(
+                    message=f"Failed to parse server response",
+                    status_code=raw_response.status_code,
+                )
+        else:
+            result = raw_response.content.text()
+
+        if raw_response.status_code == 401:
+            raise AuthException(
+                message=f"Unauthorized",
+            )
+        elif raw_response.status_code == 403:
+            raise HttpClientException(
+                message=f"{message_error if message_error else 'Forbidden'} ({raw_response.status_code})",
+                status_code=raw_response.status_code,
+            )
+        elif raw_response.status_code >= 400:
+            raise HttpClientException(
+                message=f"{message_error if message_error else 'Request error'} ({raw_response.status_code})",
+                status_code=raw_response.status_code,
+            )
+        elif raw_response.status_code < 200 or raw_response.status_code > 300:
+            raise HttpClientException(
+                message=f"{message_error if message_error else 'Request error'} ({raw_response.status_code})",
+                status_code=raw_response.status_code,
+            )
+
+        return result

thestage/services/clients/thestage_api/core/http_client_exception.py

@@ -1,12 +1,12 @@
-from thestage.exceptions.base_exception import BaseAbstractException
-
-
-class HttpClientException(BaseAbstractException):
-    _status_code: int = 0
-
-    def __init__(self, message: str, status_code: int, ):
-        super(HttpClientException, self).__init__(message=message)
-        self._status_code = status_code
-
-    def get_status_code(self) -> int:
-        return self._status_code
+from thestage.exceptions.base_exception import BaseAbstractException
+
+
+class HttpClientException(BaseAbstractException):
+    _status_code: int = 0
+
+    def __init__(self, message: str, status_code: int, ):
+        super(HttpClientException, self).__init__(message=message)
+        self._status_code = status_code
+
+    def get_status_code(self) -> int:
+        return self._status_code

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

@@ -1,13 +1,13 @@
-from typing import Optional, List
-
-from pydantic import BaseModel, Field
-
-
-class TheStageBaseResponse(BaseModel):
-    message: Optional[str] = Field(None, alias='message')
-    is_success: Optional[bool] = Field(None, alias='isSuccess')
-
-
-# TODO: add Generic then back doing that
-class TheStageBasePaginatedResponse(TheStageBaseResponse):
-    fields_with_error: List[str] = Field(None, alias='fieldsWithError')
+from typing import Optional, List
+
+from pydantic import BaseModel, Field
+
+
+class TheStageBaseResponse(BaseModel):
+    message: Optional[str] = Field(None, alias='message')
+    is_success: Optional[bool] = Field(None, alias='isSuccess')
+
+
+# TODO: add Generic then back doing that
+class TheStageBasePaginatedResponse(TheStageBaseResponse):
+    fields_with_error: List[str] = Field(None, alias='fieldsWithError')

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

@@ -1,19 +1,19 @@
-from typing import Optional
-
-from pydantic import BaseModel, Field, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.enums.location_region import LocationRegionEnumDto
-from thestage.services.clients.thestage_api.dtos.enums.provider_name import ProviderNameEnumDto
-
-
-class CloudProviderRegionDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    id: Optional[int] = Field(None, alias='id')
-    region_key: Optional[str] = Field(None, alias='regionKey')
-    provider_name: ProviderNameEnumDto = Field(ProviderNameEnumDto.UNKNOWN, alias='providerName')
-    location_region: LocationRegionEnumDto = Field(LocationRegionEnumDto.UNKNOWN, alias='locationRegion')
-    location_country: Optional[str] = Field(None, alias='locationCountry')
-    location_detail: Optional[str] = Field(None, alias='locationDetail')
-    is_active_for_storages: Optional[bool] = Field(False, alias='isActiveForStorages')
-    is_active_for_instances: Optional[bool] = Field(False, alias='isActiveForInstances')
+from typing import Optional
+
+from pydantic import BaseModel, Field, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.enums.location_region import LocationRegionEnumDto
+from thestage.services.clients.thestage_api.dtos.enums.provider_name import ProviderNameEnumDto
+
+
+class CloudProviderRegionDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    id: Optional[int] = Field(None, alias='id')
+    region_key: Optional[str] = Field(None, alias='regionKey')
+    provider_name: ProviderNameEnumDto = Field(ProviderNameEnumDto.UNKNOWN, alias='providerName')
+    location_region: LocationRegionEnumDto = Field(LocationRegionEnumDto.UNKNOWN, alias='locationRegion')
+    location_country: Optional[str] = Field(None, alias='locationCountry')
+    location_detail: Optional[str] = Field(None, alias='locationDetail')
+    is_active_for_storages: Optional[bool] = Field(False, alias='isActiveForStorages')
+    is_active_for_instances: Optional[bool] = Field(False, alias='isActiveForInstances')

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

@@ -1,11 +1,11 @@
-from typing import Optional, List
-
-from pydantic import Field, BaseModel, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.enums.container_pending_action import DockerContainerAction
-
-class DockerContainerActionRequestDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    container_id: Optional[int] = Field(None, alias='dockerContainerId')
-    action: DockerContainerAction = Field(None, alias='action')
+from typing import Optional, List
+
+from pydantic import Field, BaseModel, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.enums.container_pending_action import DockerContainerAction
+
+class DockerContainerActionRequestDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    container_id: Optional[int] = Field(None, alias='dockerContainerId')
+    action: DockerContainerAction = Field(None, alias='action')

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

@@ -1,67 +1,67 @@
-from typing import Optional, List, Dict
-
-from pydantic import Field, BaseModel, ConfigDict
-
-from thestage.services.clients.thestage_api.dtos.docker_container_assigned_device import DockerContainerAssignedDeviceDto
-from thestage.services.clients.thestage_api.dtos.docker_container_mapping import DockerContainerMappingDto
-from thestage.services.clients.thestage_api.dtos.enums.container_pending_action import DockerContainerAction
-from thestage.services.clients.thestage_api.dtos.enums.container_status import DockerContainerStatus
-from thestage.services.clients.thestage_api.dtos.frontend_status import FrontendStatusDto
-from thestage.services.clients.thestage_api.dtos.installed_service import DockerContainerInstalledServicesDto
-from thestage.services.clients.thestage_api.dtos.instance_rented_response import InstanceRentedDto
-from thestage.services.clients.thestage_api.dtos.project_response import ProjectDto
-from thestage.services.clients.thestage_api.dtos.selfhosted_instance_response import SelfHostedInstanceDto
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse, TheStageBasePaginatedResponse
-from thestage.services.clients.thestage_api.dtos.pagination_data import PaginationData
-
-
-class DockerContainerDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    id: Optional[int] = Field(None, alias='id')
-    instance_rented_id: Optional[int] = Field(None, alias='instanceRentedId')
-    selfhosted_instance_id: Optional[int] = Field(None, alias='selfhostedInstanceId')
-    instance_rented: Optional[InstanceRentedDto] = Field(None, alias='instanceRented')
-    selfhosted_instance: Optional[SelfHostedInstanceDto] = Field(None, alias='selfhostedInstance')
-
-    project_id: Optional[int] = Field(None, alias='projectId')
-    project: Optional[ProjectDto] = Field(None, alias='project')
-
-    is_default: Optional[bool] = Field(None, alias='isDefault')
-    system_name: Optional[str] = Field(None, alias='systemName')
-    title: Optional[str] = Field(None, alias='title')
-    slug: Optional[str] = Field(None, alias='slug')
-    docker_image: Optional[str] = Field(None, alias='dockerImage')
-    assigned_devices: Optional[DockerContainerAssignedDeviceDto] = Field(None, alias='assignedDevices')
-    mappings: Optional[DockerContainerMappingDto] = Field(None, alias='mappings')
-
-    frontend_status: Optional[FrontendStatusDto] = Field(None, alias='frontendStatus')
-
-    pending_action: Optional[DockerContainerAction] = Field(None, alias='pendingAction')
-    installed_services: Optional[DockerContainerInstalledServicesDto] = Field(None, alias='installedServices')
-    created_at: Optional[str] = Field(None, alias='createdAt')
-    updated_at: Optional[str] = Field(None, alias='updatedAt')
-
-
-class DockerContainerPaginated(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    entities: List[DockerContainerDto] = Field(default_factory=list, alias='entities')
-    current_page: Optional[int] = Field(None, alias='currentPage')
-    last_page: Optional[bool] = Field(None, alias='lastPage')
-    total_pages: Optional[int] = Field(None, alias='totalPages')
-    pagination_data: Optional[PaginationData] = Field(None, alias='paginationData')
-
-#
-# class DockerContainerListResponse(TheStageBasePaginatedResponse):
-#     paginated_list: Optional[DockerContainerPaginated] = Field(None, alias='paginatedList')
-
-
-class DockerContainerViewResponse(TheStageBaseResponse):
-    docker_container: Optional[DockerContainerDto] = Field(None, alias='dockerContainer')
-
-
-class ContainerBusinessStatusMapperResponse(TheStageBasePaginatedResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    docker_container_status_map: Dict[str, str] = Field(default={}, alias='dockerContainerStatusMap')
+from typing import Optional, List, Dict
+
+from pydantic import Field, BaseModel, ConfigDict
+
+from thestage.services.clients.thestage_api.dtos.docker_container_assigned_device import DockerContainerAssignedDeviceDto
+from thestage.services.clients.thestage_api.dtos.docker_container_mapping import DockerContainerMappingDto
+from thestage.services.clients.thestage_api.dtos.enums.container_pending_action import DockerContainerAction
+from thestage.services.clients.thestage_api.dtos.enums.container_status import DockerContainerStatus
+from thestage.services.clients.thestage_api.dtos.frontend_status import FrontendStatusDto
+from thestage.services.clients.thestage_api.dtos.installed_service import DockerContainerInstalledServicesDto
+from thestage.services.clients.thestage_api.dtos.instance_rented_response import InstanceRentedDto
+from thestage.services.clients.thestage_api.dtos.project_response import ProjectDto
+from thestage.services.clients.thestage_api.dtos.selfhosted_instance_response import SelfHostedInstanceDto
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse, TheStageBasePaginatedResponse
+from thestage.services.clients.thestage_api.dtos.pagination_data import PaginationData
+
+
+class DockerContainerDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    id: Optional[int] = Field(None, alias='id')
+    instance_rented_id: Optional[int] = Field(None, alias='instanceRentedId')
+    selfhosted_instance_id: Optional[int] = Field(None, alias='selfhostedInstanceId')
+    instance_rented: Optional[InstanceRentedDto] = Field(None, alias='instanceRented')
+    selfhosted_instance: Optional[SelfHostedInstanceDto] = Field(None, alias='selfhostedInstance')
+
+    project_id: Optional[int] = Field(None, alias='projectId')
+    project: Optional[ProjectDto] = Field(None, alias='project')
+
+    is_default: Optional[bool] = Field(None, alias='isDefault')
+    system_name: Optional[str] = Field(None, alias='systemName')
+    title: Optional[str] = Field(None, alias='title')
+    slug: Optional[str] = Field(None, alias='slug')
+    docker_image: Optional[str] = Field(None, alias='dockerImage')
+    assigned_devices: Optional[DockerContainerAssignedDeviceDto] = Field(None, alias='assignedDevices')
+    mappings: Optional[DockerContainerMappingDto] = Field(None, alias='mappings')
+
+    frontend_status: Optional[FrontendStatusDto] = Field(None, alias='frontendStatus')
+
+    pending_action: Optional[DockerContainerAction] = Field(None, alias='pendingAction')
+    installed_services: Optional[DockerContainerInstalledServicesDto] = Field(None, alias='installedServices')
+    created_at: Optional[str] = Field(None, alias='createdAt')
+    updated_at: Optional[str] = Field(None, alias='updatedAt')
+
+
+class DockerContainerPaginated(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    entities: List[DockerContainerDto] = Field(default_factory=list, alias='entities')
+    current_page: Optional[int] = Field(None, alias='currentPage')
+    last_page: Optional[bool] = Field(None, alias='lastPage')
+    total_pages: Optional[int] = Field(None, alias='totalPages')
+    pagination_data: Optional[PaginationData] = Field(None, alias='paginationData')
+
+#
+# class DockerContainerListResponse(TheStageBasePaginatedResponse):
+#     paginated_list: Optional[DockerContainerPaginated] = Field(None, alias='paginatedList')
+
+
+class DockerContainerViewResponse(TheStageBaseResponse):
+    docker_container: Optional[DockerContainerDto] = Field(None, alias='dockerContainer')
+
+
+class ContainerBusinessStatusMapperResponse(TheStageBasePaginatedResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    docker_container_status_map: Dict[str, str] = Field(default={}, alias='dockerContainerStatusMap')

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

@@ -1,10 +1,10 @@
-from typing import Optional, List
-
-from pydantic import Field, BaseModel, ConfigDict
-
-
-class DockerContainerAssignedDeviceDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    assigned_cpus: List[int] = Field(default_factory=list, alias='assignedCpusIds')
-    assigned_gpu_ids: List[int] = Field(default_factory=list, alias='assignedGpuIds')
+from typing import Optional, List
+
+from pydantic import Field, BaseModel, ConfigDict
+
+
+class DockerContainerAssignedDeviceDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    assigned_cpus: List[int] = Field(default_factory=list, alias='assignedCpusIds')
+    assigned_gpu_ids: List[int] = Field(default_factory=list, alias='assignedGpuIds')

thestage/services/clients/thestage_api/dtos/docker_container_controller/docker_container_list_request.py

@@ -1,13 +1,13 @@
-from typing import Optional, List
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
-
-
-class DockerContainerListRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
-    statuses: Optional[List[str]] = Field(None, alias='statuses')
-    projectId: Optional[int] = Field(None, alias='projectId')
+from typing import Optional, List
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
+
+
+class DockerContainerListRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    entityFilterRequest: EntityFilterRequest = Field(None, alias='entityFilterRequest')
+    statuses: Optional[List[str]] = Field(None, alias='statuses')
+    projectId: Optional[int] = Field(None, alias='projectId')

thestage/services/clients/thestage_api/dtos/docker_container_controller/docker_container_list_response.py

@@ -1,13 +1,13 @@
-from typing import List, Optional
-
-from pydantic import Field, ConfigDict, BaseModel
-
-from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
-from thestage.services.clients.thestage_api.dtos.container_response import DockerContainerDto
-from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
-from thestage.services.clients.thestage_api.dtos.pagination_data import PaginationData
-
-class DockerContainerListResponse(TheStageBaseResponse):
-    model_config = ConfigDict(use_enum_values=True)
-
-    paginatedList: PaginatedEntityList[DockerContainerDto] = Field(None, alias='paginatedList')
+from typing import List, Optional
+
+from pydantic import Field, ConfigDict, BaseModel
+
+from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
+from thestage.services.clients.thestage_api.dtos.container_response import DockerContainerDto
+from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
+from thestage.services.clients.thestage_api.dtos.pagination_data import PaginationData
+
+class DockerContainerListResponse(TheStageBaseResponse):
+    model_config = ConfigDict(use_enum_values=True)
+
+    paginatedList: PaginatedEntityList[DockerContainerDto] = Field(None, alias='paginatedList')

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

@@ -1,10 +1,10 @@
-from typing import Optional, List, Dict
-
-from pydantic import Field, BaseModel, ConfigDict
-
-
-class DockerContainerMappingDto(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    directory_mappings: Dict[str, str] = Field(default_factory=dict, alias='directoryMappings')
-    port_mappings: Dict[str, str] = Field(default_factory=dict, alias='portMappings')
+from typing import Optional, List, Dict
+
+from pydantic import Field, BaseModel, ConfigDict
+
+
+class DockerContainerMappingDto(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    directory_mappings: Dict[str, str] = Field(default_factory=dict, alias='directoryMappings')
+    port_mappings: Dict[str, str] = Field(default_factory=dict, alias='portMappings')

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

@@ -1,14 +1,14 @@
-from typing import Optional
-
-from pydantic import ConfigDict, BaseModel, Field
-
-from thestage.entities.enums.order_direction_type import OrderDirectionType
-
-
-class EntityFilterRequest(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    orderByField: Optional[str] = Field(None, alias='orderByField')
-    orderByDirection: Optional[OrderDirectionType] = Field(None, alias='orderByDirection')
-    page: Optional[int] = Field(None, alias='page')
-    limit: Optional[int] = Field(None, alias='limit')
+from typing import Optional
+
+from pydantic import ConfigDict, BaseModel, Field
+
+from thestage.entities.enums.order_direction_type import OrderDirectionType
+
+
+class EntityFilterRequest(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    orderByField: Optional[str] = Field(None, alias='orderByField')
+    orderByDirection: Optional[OrderDirectionType] = Field(None, alias='orderByDirection')
+    page: Optional[int] = Field(None, alias='page')
+    limit: Optional[int] = Field(None, alias='limit')

thestage/services/clients/thestage_api/dtos/enums/container_pending_action.py

@@ -1,10 +1,10 @@
-from enum import Enum
-
-
-class DockerContainerAction(str, Enum):
-    CREATE: str = 'CREATE'
-    UPDATE: str = 'UPDATE'
-    START: str = 'START'
-    RESTART: str = 'RESTART'
-    STOP: str = 'STOP'
-    DELETE: str = 'DELETE'
+from enum import Enum
+
+
+class DockerContainerAction(str, Enum):
+    CREATE: str = 'CREATE'
+    UPDATE: str = 'UPDATE'
+    START: str = 'START'
+    RESTART: str = 'RESTART'
+    STOP: str = 'STOP'
+    DELETE: str = 'DELETE'