thestage 0.6.1__py3-none-any.whl → 0.6.3__py3-none-any.whl

thestage/services/container/mapper/container_mapper.py

@@ -1,30 +1,30 @@
-from typing import Optional, Tuple
-
-from thestage.services.clients.thestage_api.dtos.container_response import DockerContainerDto
-from thestage.entities.container import DockerContainerEntity
-from thestage.services.abstract_mapper import AbstractMapper
-
-
-class ContainerMapper(AbstractMapper):
-
-    def build_entity(self, item: DockerContainerDto) -> Optional[DockerContainerEntity]:
-        if not item:
-            return None
-
-        instance_type = ''
-        instance_slug = ''
-        if item.instance_rented:
-            instance_type = 'RENTED'
-            instance_slug = item.instance_rented.slug
-        if item.selfhosted_instance:
-            instance_type = 'SELF-HOSTED'
-            instance_slug = item.selfhosted_instance.slug
-
-        return DockerContainerEntity(
-            status=item.frontend_status.status_translation if item.frontend_status else '',
-            slug=item.slug or '',
-            title=item.title or '',
-            instance_type=instance_type,
-            instance_slug=instance_slug,
-            docker_image=item.docker_image or '',
-        )
+from typing import Optional, Tuple
+
+from thestage.services.clients.thestage_api.dtos.container_response import DockerContainerDto
+from thestage.entities.container import DockerContainerEntity
+from thestage.services.abstract_mapper import AbstractMapper
+
+
+class ContainerMapper(AbstractMapper):
+
+    def build_entity(self, item: DockerContainerDto) -> Optional[DockerContainerEntity]:
+        if not item:
+            return None
+
+        instance_type = ''
+        instance_slug = ''
+        if item.instance_rented:
+            instance_type = 'RENTED'
+            instance_slug = item.instance_rented.slug
+        if item.selfhosted_instance:
+            instance_type = 'SELF-HOSTED'
+            instance_slug = item.selfhosted_instance.slug
+
+        return DockerContainerEntity(
+            status=item.frontend_status.status_translation if item.frontend_status else '',
+            slug=item.slug or '',
+            title=item.title or '',
+            instance_type=instance_type,
+            instance_slug=instance_slug,
+            docker_image=item.docker_image or '',
+        )

thestage/services/core_files/config_entity.py

@@ -1,26 +1,26 @@
-from typing import Optional, Dict
-
-from pydantic import BaseModel, Field
-
-from thestage.cli_command import CliCommand, CliCommandAvailability
-
-
-# saved to file
-class MainConfigEntity(BaseModel):
-    thestage_auth_token: Optional[str] = Field(None, alias='thestage_auth_token')
-    thestage_api_url: Optional[str] = Field(None, alias='thestage_api_url')
-
-
-# not saved to file
-class RuntimeConfigEntity(BaseModel):
-    working_directory: Optional[str] = Field(None, alias='working_directory')
-    config_global_path: Optional[str] = Field(None, alias='config_global_path')
-    allowed_commands: Dict[CliCommand, CliCommandAvailability] = {}
-    is_token_valid: bool = Field(None, alias='is_token_valid')
-
-
-class ConfigEntity(BaseModel):
-    global_config_path: str = Field(None, alias='global_config_path')
-    can_use_inference: bool = Field(None, alias='can_use_inference')
-    main: MainConfigEntity = Field(default_factory=MainConfigEntity, alias='main')
-    runtime: RuntimeConfigEntity = Field(default_factory=RuntimeConfigEntity, alias="runtime") # TODO merge with main
+from typing import Optional, Dict
+
+from pydantic import BaseModel, Field
+
+from thestage.cli_command import CliCommand, CliCommandAvailability
+
+
+# saved to file
+class MainConfigEntity(BaseModel):
+    thestage_auth_token: Optional[str] = Field(None, alias='thestage_auth_token')
+    thestage_api_url: Optional[str] = Field(None, alias='thestage_api_url')
+
+
+# not saved to file
+class RuntimeConfigEntity(BaseModel):
+    working_directory: Optional[str] = Field(None, alias='working_directory')
+    config_global_path: Optional[str] = Field(None, alias='config_global_path')
+    allowed_commands: Dict[CliCommand, CliCommandAvailability] = {}
+    is_token_valid: bool = Field(None, alias='is_token_valid')
+
+
+class ConfigEntity(BaseModel):
+    global_config_path: str = Field(None, alias='global_config_path')
+    can_use_inference: bool = Field(None, alias='can_use_inference')
+    main: MainConfigEntity = Field(default_factory=MainConfigEntity, alias='main')
+    runtime: RuntimeConfigEntity = Field(default_factory=RuntimeConfigEntity, alias="runtime") # TODO merge with main

thestage/services/filesystem_service.py

@@ -1,133 +1,133 @@
-import json
-import os
-import shutil
-from json import JSONDecodeError
-from pathlib import Path
-from typing import Optional, List, Dict, Any
-
-from thestage.entities.file_item import FileItemEntity
-from thestage.exceptions.file_system_exception import FileSystemException
-
-
-class FileSystemService:
-
-    def get_ssh_path(self) -> Optional[Path]:
-        home_path = self.get_home_path()
-        ssh_path = home_path.joinpath('.ssh')
-        if not ssh_path.exists():
-            raise FileSystemException(f"Path does not exist: {ssh_path}")
-        return ssh_path
-
-    def get_home_path(self) -> Optional[Path]:
-        try:
-            return Path.home()
-        except RuntimeError | OSError as ex1:
-            raise FileSystemException("Error getting user home path") from ex1
-
-    def create_if_not_exists_dir(self, path: Path) -> Path:
-        if not path.exists():
-            try:
-                path.mkdir(exist_ok=True, parents=True)
-            except FileNotFoundError as ex1:
-                raise FileSystemException(message=f"FileNotFoundError (dir): {path}") from ex1
-            except OSError as ex2:
-                raise FileSystemException(message=f"Could not create directory: {path}") from ex2
-        return path
-
-    def create_if_not_exists_file(self, path: Path) -> Path:
-        if not path.exists():
-            try:
-                path.touch(exist_ok=True)
-            except FileNotFoundError as ex1:
-                raise FileSystemException(message=f"FileNotFoundError (file): {path}") from ex1
-            except OSError as ex2:
-                raise FileSystemException(message=f"Could not create file: {path}") from ex2
-        return path
-
-    def get_path(self, directory: str, auto_create: bool = True) -> Path:
-        path = Path(directory)
-        if auto_create:
-            self.create_if_not_exists_dir(path)
-        return path
-
-    def is_folder_empty(self, folder: str, auto_create: bool) -> bool:
-        path = self.get_path(folder, auto_create)
-        if not path.exists():
-            return True
-        if not path.is_dir():
-            raise FileSystemException(message=f"Expected directory but found a file: {path}")
-        objects = os.listdir(path)
-        if len(objects) == 0:
-            return True
-        else:
-            return False
-
-    def is_folder_exists(self, folder: str, auto_create: bool = True) -> bool:
-        path = self.get_path(folder, auto_create=auto_create)
-        if path.exists():
-            return True
-        else:
-            return False
-
-    def find_line_in_text_file(self, file: str, find: str) -> bool:
-        path = self.get_path(file, auto_create=False)
-        if path and path.exists():
-            with open(path, 'r') as file:
-                for line in file.readlines():
-                    if (find + "\n") == line:
-                        return True
-        return False
-
-    def add_line_to_text_file(self, file: str, new_line: str):
-        path = self.get_path(file, auto_create=False)
-        if path and path.exists():
-            with open(path, 'a') as file:
-                file.write(new_line)
-                file.write('\n')
-
-    # TODO remove this fucking useless shit
-    def check_if_path_exist(self, file: str) -> bool:
-        path = self.get_path(file, auto_create=False)
-        if path.exists():
-            return True
-        else:
-            return False
-
-    def get_path_items(self, folder: str) -> List[FileItemEntity]:
-        path = self.get_path(folder, auto_create=False)
-        path_items = []
-        if not path.exists():
-            return path_items
-
-        parent = FileItemEntity.build_from_path(path=path)
-        path_items.append(parent)
-        if path.is_dir():
-            objects = os.listdir(path)
-            if objects:
-                for item in objects:
-                    elem = path.joinpath(item)
-                    if elem.is_dir():
-                        parent.children.extend(self.get_path_items(folder=str(elem)))
-                    else:
-                        parent.children.append(FileItemEntity.build_from_path(path=elem))
-        return path_items
-
-    def remove_folder(self, path: str):
-        real_path = self.get_path(directory=path, auto_create=False)
-        if real_path and real_path.exists():
-            shutil.rmtree(real_path)
-
-
-    def read_config_file(self, path: Path) -> Dict[str, Any]:
-        result = {}
-        try:
-            if path and path.exists():
-                with path.open("r") as file:
-                    try:
-                        if os.stat(path).st_size != 0:
-                            result = json.load(file)
-                    except JSONDecodeError as e:
-                        raise Exception(f"Config file is malformed: {path}") from e
-        except OSError:
-            raise FileSystemException(f"Could not open config file: {path}")
-        return result
+import json
+import os
+import shutil
+from json import JSONDecodeError
+from pathlib import Path
+from typing import Optional, List, Dict, Any
+
+from thestage.entities.file_item import FileItemEntity
+from thestage.exceptions.file_system_exception import FileSystemException
+
+
+class FileSystemService:
+
+    def get_ssh_path(self) -> Optional[Path]:
+        home_path = self.get_home_path()
+        ssh_path = home_path.joinpath('.ssh')
+        if not ssh_path.exists():
+            raise FileSystemException(f"Path does not exist: {ssh_path}")
+        return ssh_path
+
+    def get_home_path(self) -> Optional[Path]:
+        try:
+            return Path.home()
+        except RuntimeError | OSError as ex1:
+            raise FileSystemException("Error getting user home path") from ex1
+
+    def create_if_not_exists_dir(self, path: Path) -> Path:
+        if not path.exists():
+            try:
+                path.mkdir(exist_ok=True, parents=True)
+            except FileNotFoundError as ex1:
+                raise FileSystemException(message=f"FileNotFoundError (dir): {path}") from ex1
+            except OSError as ex2:
+                raise FileSystemException(message=f"Could not create directory: {path}") from ex2
+        return path
+
+    def create_if_not_exists_file(self, path: Path) -> Path:
+        if not path.exists():
+            try:
+                path.touch(exist_ok=True)
+            except FileNotFoundError as ex1:
+                raise FileSystemException(message=f"FileNotFoundError (file): {path}") from ex1
+            except OSError as ex2:
+                raise FileSystemException(message=f"Could not create file: {path}") from ex2
+        return path
+
+    def get_path(self, directory: str, auto_create: bool = True) -> Path:
+        path = Path(directory)
+        if auto_create:
+            self.create_if_not_exists_dir(path)
+        return path
+
+    def is_folder_empty(self, folder: str, auto_create: bool) -> bool:
+        path = self.get_path(folder, auto_create)
+        if not path.exists():
+            return True
+        if not path.is_dir():
+            raise FileSystemException(message=f"Expected directory but found a file: {path}")
+        objects = os.listdir(path)
+        if len(objects) == 0:
+            return True
+        else:
+            return False
+
+    def is_folder_exists(self, folder: str, auto_create: bool = True) -> bool:
+        path = self.get_path(folder, auto_create=auto_create)
+        if path.exists():
+            return True
+        else:
+            return False
+
+    def find_line_in_text_file(self, file: str, find: str) -> bool:
+        path = self.get_path(file, auto_create=False)
+        if path and path.exists():
+            with open(path, 'r') as file:
+                for line in file.readlines():
+                    if (find + "\n") == line:
+                        return True
+        return False
+
+    def add_line_to_text_file(self, file: str, new_line: str):
+        path = self.get_path(file, auto_create=False)
+        if path and path.exists():
+            with open(path, 'a') as file:
+                file.write(new_line)
+                file.write('\n')
+
+    # TODO remove this fucking useless shit
+    def check_if_path_exist(self, file: str) -> bool:
+        path = self.get_path(file, auto_create=False)
+        if path.exists():
+            return True
+        else:
+            return False
+
+    def get_path_items(self, folder: str) -> List[FileItemEntity]:
+        path = self.get_path(folder, auto_create=False)
+        path_items = []
+        if not path.exists():
+            return path_items
+
+        parent = FileItemEntity.build_from_path(path=path)
+        path_items.append(parent)
+        if path.is_dir():
+            objects = os.listdir(path)
+            if objects:
+                for item in objects:
+                    elem = path.joinpath(item)
+                    if elem.is_dir():
+                        parent.children.extend(self.get_path_items(folder=str(elem)))
+                    else:
+                        parent.children.append(FileItemEntity.build_from_path(path=elem))
+        return path_items
+
+    def remove_folder(self, path: str):
+        real_path = self.get_path(directory=path, auto_create=False)
+        if real_path and real_path.exists():
+            shutil.rmtree(real_path)
+
+
+    def read_config_file(self, path: Path) -> Dict[str, Any]:
+        result = {}
+        try:
+            if path and path.exists():
+                with path.open("r") as file:
+                    try:
+                        if os.stat(path).st_size != 0:
+                            result = json.load(file)
+                    except JSONDecodeError as e:
+                        raise Exception(f"Config file is malformed: {path}") from e
+        except OSError:
+            raise FileSystemException(f"Could not open config file: {path}")
+        return result