thestage 0.5.471__py3-none-any.whl → 0.6.2__py3-none-any.whl

thestage/services/instance/mapper/selfhosted_mapper.py

@@ -1,33 +1,33 @@
-from abc import ABC
-from typing import Optional, Any, Tuple
-
-from thestage.entities.rented_instance import RentedInstanceEntity
-from thestage.entities.self_hosted_instance import SelfHostedInstanceEntity
-from thestage.services.clients.thestage_api.dtos.selfhosted_instance_response import SelfHostedInstanceDto
-from thestage.services.abstract_mapper import AbstractMapper
-
-
-class SelfHostedMapper(AbstractMapper):
-
-    def build_entity(self, item: SelfHostedInstanceDto) -> Optional[SelfHostedInstanceEntity]:
-        if not item:
-            return None
-
-        gpus = []
-        if item.detected_gpus:
-            gpus = [item.type.value for item in item.detected_gpus.gpus]
-
-        if len(gpus) == 0:
-            gpus = ['NO_GPU']
-
-        return SelfHostedInstanceEntity(
-            slug=item.slug,
-            title=item.title,
-            cpu_type=item.cpu_type,
-            cpu_cores=item.cpu_cores,
-            gpu_type=', '.join(gpus),
-            ip_address=item.ip_address,
-            status=item.frontend_status.status_translation if item.frontend_status else None,
-            created_at=item.created_at.strftime("%Y-%m-%d %H:%M:%S") if item.created_at else '',
-            updated_at=item.updated_at.strftime("%Y-%m-%d %H:%M:%S") if item.updated_at else '',
-        )
+from abc import ABC
+from typing import Optional, Any, Tuple
+
+from thestage.entities.rented_instance import RentedInstanceEntity
+from thestage.entities.self_hosted_instance import SelfHostedInstanceEntity
+from thestage.services.clients.thestage_api.dtos.selfhosted_instance_response import SelfHostedInstanceDto
+from thestage.services.abstract_mapper import AbstractMapper
+
+
+class SelfHostedMapper(AbstractMapper):
+
+    def build_entity(self, item: SelfHostedInstanceDto) -> Optional[SelfHostedInstanceEntity]:
+        if not item:
+            return None
+
+        gpus = []
+        if item.detected_gpus:
+            gpus = [item.type.value for item in item.detected_gpus.gpus]
+
+        if len(gpus) == 0:
+            gpus = ['NO_GPU']
+
+        return SelfHostedInstanceEntity(
+            slug=item.slug,
+            title=item.title,
+            cpu_type=item.cpu_type,
+            cpu_cores=item.cpu_cores,
+            gpu_type=', '.join(gpus),
+            ip_address=item.ip_address,
+            status=item.frontend_status.status_translation if item.frontend_status else None,
+            created_at=item.created_at.strftime("%Y-%m-%d %H:%M:%S") if item.created_at else '',
+            updated_at=item.updated_at.strftime("%Y-%m-%d %H:%M:%S") if item.updated_at else '',
+        )

thestage/services/logging/byte_print_style.py

@@ -1,6 +1,6 @@
-class BytePrintStyle:
-    RED = "\033[31m"
-    GREEN = "\033[32m"
-    BLUE = "\033[34m"
-    ORANGE = "\033[38;5;214m"
+class BytePrintStyle:
+    RED = "\033[31m"
+    GREEN = "\033[32m"
+    BLUE = "\033[34m"
+    ORANGE = "\033[38;5;214m"
     RESET = "\033[0m"

thestage/services/logging/dto/log_message.py

@@ -1,15 +1,15 @@
-from typing import Optional
-
-from pydantic import BaseModel, ConfigDict, Field
-
-class LogMessage(BaseModel):
-
-    model_config = ConfigDict(
-        populate_by_name=True,
-        use_enum_values=True,
-    )
-
-    logType: Optional[str] = Field(None, alias='logType')
-    message: Optional[str] = Field(None, alias='message')
-    timestamp: Optional[str] = Field(None, alias='timestamp')
-    messageCode: Optional[int] = Field(None, alias='messageCode')
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+class LogMessage(BaseModel):
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        use_enum_values=True,
+    )
+
+    logType: Optional[str] = Field(None, alias='logType')
+    message: Optional[str] = Field(None, alias='message')
+    timestamp: Optional[str] = Field(None, alias='timestamp')
+    messageCode: Optional[int] = Field(None, alias='messageCode')

thestage/services/logging/dto/log_type.py

@@ -1,6 +1,6 @@
-from enum import Enum
-
-
-class LogType(str, Enum):
-    STDOUT = "STDOUT",
-    STDERR = "STDERR"
+from enum import Enum
+
+
+class LogType(str, Enum):
+    STDOUT = "STDOUT",
+    STDERR = "STDERR"

thestage/services/logging/exception/log_polling_exception.py

@@ -1,6 +1,6 @@
-from thestage.exceptions.base_exception import BaseAbstractException
-
-
-class LogPollingException(BaseAbstractException):
-    def __init__(self, message: str):
-        super(LogPollingException, self).__init__(message=message)
+from thestage.exceptions.base_exception import BaseAbstractException
+
+
+class LogPollingException(BaseAbstractException):
+    def __init__(self, message: str):
+        super(LogPollingException, self).__init__(message=message)

thestage/services/logging/logging_constants.py

@@ -1,3 +1,3 @@
-LOG_MESSAGE_CODE_REGULAR_MESSAGE: int = 0
-LOG_MESSAGE_CODE_TASK_FINISHED: int = 100
-LOG_MESSAGE_CODE_INFERENCE_SIMULATOR_FAILED: int = 1
+LOG_MESSAGE_CODE_REGULAR_MESSAGE: int = 0
+LOG_MESSAGE_CODE_TASK_FINISHED: int = 100
+LOG_MESSAGE_CODE_INFERENCE_SIMULATOR_FAILED: int = 1