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

thestage/controllers/utils_controller.py

@@ -1,28 +1,32 @@
-import os
-import pathlib
-from typing import Optional, Dict, Tuple
-
-from thestage.services.core_files.config_entity import ConfigEntity
-from thestage.helpers.error_handler import error_handler
-from thestage.services.service_factory import ServiceFactory
-from thestage.services.config_provider.config_provider import ConfigProvider
-
-
-def get_current_directory() -> pathlib.Path:
-    return pathlib.Path.cwd()
-
-
-@error_handler()
-def validate_config_and_get_service_factory(
-        working_directory: Optional[str] = None,
-) -> ServiceFactory:
-    local_path = get_current_directory() if not working_directory else os.path.abspath(working_directory)
-    config_provider = ConfigProvider(local_path=local_path)
-    service_factory = ServiceFactory(config_provider=config_provider)
-    config: ConfigEntity = config_provider.get_full_config()
-
-    validation_service = service_factory.get_validation_service()
-    validation_service.check_token(config=config)
-    config_provider.save_config(config=config)
-
-    return service_factory
+import os
+import pathlib
+from typing import Optional
+
+from thestage.helpers.error_handler import error_handler
+from thestage.services.service_factory import ServiceFactory
+
+
+def get_current_directory() -> pathlib.Path:
+    return pathlib.Path.cwd()
+
+
+@error_handler()
+def validate_config_and_get_service_factory(
+        working_directory: Optional[str] = None,
+) -> ServiceFactory:
+    local_path = get_current_directory() if not working_directory else os.path.abspath(working_directory)
+
+    service_factory = ServiceFactory()
+    config_provider = service_factory.get_config_provider()
+
+    config = config_provider.get_config()
+    config.runtime.working_directory = str(local_path)
+
+    config_provider.update_config(updated_config=config)
+
+    validation_service = service_factory.get_validation_service()
+
+    validation_service.check_token()
+    config_provider.save_config()
+
+    return service_factory

thestage/debug_main.dist.py

@@ -1,28 +1,28 @@
-from thestage import __app_name__
-
-from thestage.controllers import base_controller, container_controller, instance_controller, \
-    config_controller, project_controller
-
-base_controller.app.add_typer(container_controller.app, name="container")
-base_controller.app.add_typer(instance_controller.app, name="instance")
-base_controller.app.add_typer(config_controller.app, name="config")
-base_controller.app.add_typer(project_controller.app, name="project")
-
-
-def main():
-    # example of a command to debug
-    project_controller.app([
-        "run",
-        "-wd",
-        "/Users/alexey/Documents/clonetest",
-        "-dcuid",
-        '62-1',
-        '-t',
-        'task',
-        '-com',
-        "'echo 123'"
-    ], prog_name=__app_name__)
-
-
-if __name__ == "__main__":
-    main()
+from thestage import __app_name__
+
+from thestage.controllers import base_controller, container_controller, instance_controller, \
+    config_controller, project_controller
+
+base_controller.app.add_typer(container_controller.app, name="container")
+base_controller.app.add_typer(instance_controller.app, name="instance")
+base_controller.app.add_typer(config_controller.app, name="config")
+base_controller.app.add_typer(project_controller.app, name="project")
+
+
+def main():
+    # example of a command to debug
+    project_controller.app([
+        "run",
+        "-wd",
+        "/Users/alexey/Documents/clonetest",
+        "-dcuid",
+        '62-1',
+        '-t',
+        'task',
+        '-com',
+        "'echo 123'"
+    ], prog_name=__app_name__)
+
+
+if __name__ == "__main__":
+    main()

thestage/entities/container.py

@@ -1,22 +1,17 @@
-from typing import Optional
-
-from pydantic import BaseModel, ConfigDict, Field
-
-
-class DockerContainerEntity(BaseModel):
-
-    model_config = ConfigDict(
-        populate_by_name=True,
-        use_enum_values=True,
-    )
-
-    # id: Optional[int] = Field(None, alias='id')
-    slug: Optional[str] = Field(None, alias='UNIQUE ID')
-    status: Optional[str] = Field(None, alias='STATUS')
-    title: Optional[str] = Field(None, alias='TITLE')
-    instance_type: Optional[str] = Field(None, alias='INSTANCE TYPE')
-    instance_slug: Optional[str] = Field(None, alias='INSTANCE UNIQUE ID')
-    # instance_rented_slug: Optional[str] = Field(None, alias='INSTANCE UNIQUE ID')
-    # selfhosted_rented_slug: Optional[str] = Field(None, alias='SELF-HOSTED UNIQUE ID')
-    # system_name: Optional[str] = Field(None, alias='SYSTEM NAME')
-    # docker_image: Optional[str] = Field(None, alias='DOCKER IMAGE')
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class DockerContainerEntity(BaseModel):
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        use_enum_values=True,
+    )
+
+    slug: Optional[str] = Field(None, alias='UNIQUE ID')
+    status: Optional[str] = Field(None, alias='STATUS')
+    title: Optional[str] = Field(None, alias='TITLE')
+    instance_type: Optional[str] = Field(None, alias='INSTANCE TYPE')
+    instance_slug: Optional[str] = Field(None, alias='INSTANCE UNIQUE ID')

thestage/entities/enums/order_direction_type.py

@@ -1,6 +1,6 @@
-from enum import Enum
-
-
-class OrderDirectionType(str, Enum):
-    ASC = "ASC"
-    DESC = "DESC"
+from enum import Enum
+
+
+class OrderDirectionType(str, Enum):
+    ASC = "ASC"
+    DESC = "DESC"

thestage/entities/enums/shell_type.py

@@ -1,7 +1,7 @@
-from enum import Enum
-
-
-class ShellType(str, Enum):
-    BASH = "/bin/bash"
-    SH = "/bin/sh"
-    UNKNOWN = "UNKNOWN"
+from enum import Enum
+
+
+class ShellType(str, Enum):
+    BASH = "/bin/bash"
+    SH = "/bin/sh"
+    UNKNOWN = "UNKNOWN"

thestage/entities/enums/tail_output_type.py

@@ -1,6 +1,6 @@
-from enum import Enum
-
-
-class TailOutputType(str, Enum):
-    YES = '1'
-    NO = '0'
+from enum import Enum
+
+
+class TailOutputType(str, Enum):
+    YES = '1'
+    NO = '0'

thestage/entities/enums/yes_no_response.py

@@ -1,7 +1,7 @@
-from enum import Enum
-
-
-class YesOrNoResponse(str, Enum):
-    YES = "y"
-    NO = "n"
-
+from enum import Enum
+
+
+class YesOrNoResponse(str, Enum):
+    YES = "y"
+    NO = "n"
+

thestage/entities/file_item.py

@@ -1,27 +1,27 @@
-import os
-from pathlib import Path
-from typing import Optional, List
-
-from pydantic import BaseModel, ConfigDict, Field
-
-
-class FileItemEntity(BaseModel):
-    model_config = ConfigDict(use_enum_values=True)
-
-    name: Optional[str] = Field(None)
-    path: Optional[str] = Field(None)
-    is_file: Optional[bool] = Field(False)
-    is_folder: Optional[bool] = Field(False)
-    file_size: Optional[int] = Field(None)
-    children: List['FileItemEntity'] = Field(default=[])
-
-    @staticmethod
-    def build_from_path(path: Path) -> 'FileItemEntity':
-        file_stat = os.stat(path)
-        return FileItemEntity(
-                name=path.name,
-                path=str(path.absolute()),
-                is_file=path.is_file(),
-                is_folder=path.is_dir(),
-                file_size=file_stat.st_size,
-            )
+import os
+from pathlib import Path
+from typing import Optional, List
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class FileItemEntity(BaseModel):
+    model_config = ConfigDict(use_enum_values=True)
+
+    name: Optional[str] = Field(None)
+    path: Optional[str] = Field(None)
+    is_file: Optional[bool] = Field(False)
+    is_folder: Optional[bool] = Field(False)
+    file_size: Optional[int] = Field(None)
+    children: List['FileItemEntity'] = Field(default=[])
+
+    @staticmethod
+    def build_from_path(path: Path) -> 'FileItemEntity':
+        file_stat = os.stat(path)
+        return FileItemEntity(
+                name=path.name,
+                path=str(path.absolute()),
+                is_file=path.is_file(),
+                is_folder=path.is_dir(),
+                file_size=file_stat.st_size,
+            )

thestage/entities/project_inference_simulator.py

@@ -1,18 +1,18 @@
-from typing import Optional
-
-from pydantic import BaseModel, ConfigDict, Field
-
-
-class ProjectInferenceSimulatorEntity(BaseModel):
-
-    model_config = ConfigDict(
-        populate_by_name=True,
-        use_enum_values=True,
-    )
-
-    slug: Optional[str] = Field(None, alias='UNIQUE ID')
-    status: Optional[str] = Field(None, alias='STATUS')
-    http_endpoint: Optional[str] = Field(None, alias="HTTP")
-    grpc_endpoint: Optional[str] = Field(None, alias="GRPC")
-    metrics_endpoint: Optional[str] = Field(None, alias="METRICS")
-    started_at: Optional[str] = Field(None, alias='STARTED AT')
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class ProjectInferenceSimulatorEntity(BaseModel):
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        use_enum_values=True,
+    )
+
+    slug: Optional[str] = Field(None, alias='UNIQUE ID')
+    status: Optional[str] = Field(None, alias='STATUS')
+    http_endpoint: Optional[str] = Field(None, alias="HTTP")
+    grpc_endpoint: Optional[str] = Field(None, alias="GRPC")
+    metrics_endpoint: Optional[str] = Field(None, alias="METRICS")
+    started_at: Optional[str] = Field(None, alias='STARTED AT')

thestage/entities/project_inference_simulator_model.py

@@ -1,17 +1,17 @@
-from typing import Optional
-
-from pydantic import BaseModel, ConfigDict, Field
-
-
-class ProjectInferenceSimulatorModelEntity(BaseModel):
-
-    model_config = ConfigDict(
-        populate_by_name=True,
-        use_enum_values=True,
-    )
-
-    id: Optional[int] = Field(None, alias='ID')
-    slug: Optional[str] = Field(None, alias='UNIQUE_ID')
-    status: Optional[str] = Field(None, alias='STATUS')
-    commit_hash: Optional[str] = Field(None, alias='COMMIT_HASH')
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class ProjectInferenceSimulatorModelEntity(BaseModel):
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        use_enum_values=True,
+    )
+
+    id: Optional[int] = Field(None, alias='ID')
+    slug: Optional[str] = Field(None, alias='UNIQUE_ID')
+    status: Optional[str] = Field(None, alias='STATUS')
+    commit_hash: Optional[str] = Field(None, alias='COMMIT_HASH')
     environment_metadata: Optional[str] = Field(None, alias='ENVIRONMENT_METADATA')

thestage/entities/project_task.py

@@ -1,19 +1,19 @@
-from typing import Optional
-
-from pydantic import BaseModel, ConfigDict, Field
-
-
-class ProjectTaskEntity(BaseModel):
-
-    model_config = ConfigDict(
-        populate_by_name=True,
-        use_enum_values=True,
-    )
-
-    id: Optional[int] = Field(None, alias='ID')
-    title: Optional[str] = Field(None, alias='TITLE')
-    status: Optional[str] = Field(None, alias='STATUS')
-    docker_container_id: Optional[int] = Field(None, alias='DOCKER_CONTAINER_ID')
-    docker_container_slug: Optional[str] = Field(None, alias='DOCKER_CONTAINER_UID')
-    started_at: Optional[str] = Field(None, alias='STARTED AT')
-    finished_at: Optional[str] = Field(None, alias='FINISHED_AT')
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class ProjectTaskEntity(BaseModel):
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        use_enum_values=True,
+    )
+
+    id: Optional[int] = Field(None, alias='ID')
+    title: Optional[str] = Field(None, alias='TITLE')
+    status: Optional[str] = Field(None, alias='STATUS')
+    docker_container_id: Optional[int] = Field(None, alias='DOCKER_CONTAINER_ID')
+    docker_container_slug: Optional[str] = Field(None, alias='DOCKER_CONTAINER_UID')
+    started_at: Optional[str] = Field(None, alias='STARTED AT')
+    finished_at: Optional[str] = Field(None, alias='FINISHED_AT')

thestage/entities/rented_instance.py

@@ -1,24 +1,19 @@
-from typing import Optional
-
-from pydantic import BaseModel, ConfigDict, Field
-
-
-class RentedInstanceEntity(BaseModel):
-
-    model_config = ConfigDict(
-        populate_by_name=True,
-        use_enum_values=True
-    )
-
-    # id: Optional[int] = Field(None, alias='ID')
-    status: Optional[str] = Field(None, alias='STATUS')
-    title: Optional[str] = Field(None, alias='TITLE')
-    slug: Optional[str] = Field(None, alias='UNIQUE ID')
-    cpu_type: Optional[str] = Field(None, alias='CPU TYPE')
-    cpu_cores: Optional[str] = Field(None, alias='CPU CORES')
-    gpu_type: Optional[str] = Field(None, alias='GPU TYPE')
-    # ram_size_gb: Optional[int] = Field(None, alias='RAM SIZE GB')
-    ip_address: Optional[str] = Field(None, alias='IP ADDRESS')
-    # username: Optional[str] = Field(None, alias='USERNAME')
-    # created_at: Optional[str] = Field(None, alias='CREATED AT')
-    # updated_at: Optional[str] = Field(None, alias='UPDATED AT')
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class RentedInstanceEntity(BaseModel):
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        use_enum_values=True
+    )
+
+    status: Optional[str] = Field(None, alias='STATUS')
+    title: Optional[str] = Field(None, alias='TITLE')
+    slug: Optional[str] = Field(None, alias='UNIQUE ID')
+    cpu_type: Optional[str] = Field(None, alias='CPU TYPE')
+    cpu_cores: Optional[str] = Field(None, alias='CPU CORES')
+    gpu_type: Optional[str] = Field(None, alias='GPU TYPE')
+    ip_address: Optional[str] = Field(None, alias='IP ADDRESS')

thestage/entities/self_hosted_instance.py

@@ -1,20 +1,18 @@
-from typing import Optional
-
-from pydantic import BaseModel, ConfigDict, Field
-
-
-class SelfHostedInstanceEntity(BaseModel):
-    model_config = ConfigDict(
-        use_enum_values=True,
-        populate_by_name=True,
-    )
-
-    status: Optional[str] = Field(None, alias='STATUS')
-    title: Optional[str] = Field(None, alias='TITLE')
-    slug: Optional[str] = Field(None, alias='SLUG')
-    cpu_type: Optional[str] = Field(str, alias='CPU_TYPE')
-    cpu_cores: Optional[int] = Field(None, alias='CPU_CORES')
-    gpu_type: Optional[str] = Field(str, alias='GPU_TYPE')
-    ip_address: Optional[str] = Field(None, alias='IP_ADDRESS')
-    # created_at: Optional[str] = Field(None, alias='CREATED_AT')
-    # updated_at: Optional[str] = Field(None, alias='UPDATED_AT')
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class SelfHostedInstanceEntity(BaseModel):
+    model_config = ConfigDict(
+        use_enum_values=True,
+        populate_by_name=True,
+    )
+
+    status: Optional[str] = Field(None, alias='STATUS')
+    title: Optional[str] = Field(None, alias='TITLE')
+    slug: Optional[str] = Field(None, alias='SLUG')
+    cpu_type: Optional[str] = Field(str, alias='CPU_TYPE')
+    cpu_cores: Optional[int] = Field(None, alias='CPU_CORES')
+    gpu_type: Optional[str] = Field(str, alias='GPU_TYPE')
+    ip_address: Optional[str] = Field(None, alias='IP_ADDRESS')

thestage/exceptions/auth_exception.py

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

thestage/exceptions/base_exception.py

@@ -1,13 +1,13 @@
-
-class BaseAbstractException(Exception):
-    def __init__(self, message: str):
-        self._message = message
-
-    def __str__(self):
-        return self._message
-
-    def __repr__(self):
-        return self._message
-
-    def get_message(self) -> str:
-        return self._message
+
+class BaseAbstractException(Exception):
+    def __init__(self, message: str):
+        self._message = message
+
+    def __str__(self):
+        return self._message
+
+    def __repr__(self):
+        return self._message
+
+    def get_message(self) -> str:
+        return self._message

thestage/exceptions/business_logic_exception.py

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

thestage/exceptions/config_exception.py

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

thestage/exceptions/file_system_exception.py

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

thestage/exceptions/git_access_exception.py

@@ -1,17 +1,17 @@
-from thestage.exceptions.base_exception import BaseAbstractException
-
-
-class GitAccessException(BaseAbstractException):
-
-    __url: str
-
-    def __init__(self, message: str, url: str, dop_message: str):
-        super(GitAccessException, self).__init__(message=message)
-        self.__url = url
-        self.__dop_message = dop_message
-
-    def get_url(self) -> str:
-        return self.__url
-
-    def get_dop_message(self) -> str:
-        return self.__dop_message
+from thestage.exceptions.base_exception import BaseAbstractException
+
+
+class GitAccessException(BaseAbstractException):
+
+    __url: str
+
+    def __init__(self, message: str, url: str, dop_message: str):
+        super(GitAccessException, self).__init__(message=message)
+        self.__url = url
+        self.__dop_message = dop_message
+
+    def get_url(self) -> str:
+        return self.__url
+
+    def get_dop_message(self) -> str:
+        return self.__dop_message

thestage/exceptions/remote_server_exception.py

@@ -1,24 +1,24 @@
-from typing import Optional
-
-from thestage.exceptions.base_exception import BaseAbstractException
-
-
-class RemoteServerException(BaseAbstractException):
-
-    def __init__(
-            self,
-            message: str,
-            ip_address: Optional[str] = None,
-            username: Optional[str] = None,
-    ):
-        super(RemoteServerException, self).__init__(message=message)
-        self._ip_address = ip_address
-        self._username = username
-
-    @property
-    def ip_address(self):
-        return self._ip_address
-
-    @property
-    def username(self):
-        return self._username
+from typing import Optional
+
+from thestage.exceptions.base_exception import BaseAbstractException
+
+
+class RemoteServerException(BaseAbstractException):
+
+    def __init__(
+            self,
+            message: str,
+            ip_address: Optional[str] = None,
+            username: Optional[str] = None,
+    ):
+        super(RemoteServerException, self).__init__(message=message)
+        self._ip_address = ip_address
+        self._username = username
+
+    @property
+    def ip_address(self):
+        return self._ip_address
+
+    @property
+    def username(self):
+        return self._username