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

thestage/services/instance/instance_service.py

@@ -1,304 +1,304 @@
-from pathlib import Path
-from typing import List, Optional, Dict
-
-import typer
-
-from thestage.entities.rented_instance import RentedInstanceEntity
-from thestage.entities.self_hosted_instance import SelfHostedInstanceEntity
-from thestage.i18n.translation import __
-from thestage.services.clients.thestage_api.dtos.enums.selfhosted_status import SelfhostedBusinessStatus
-from thestage.services.clients.thestage_api.dtos.enums.instance_rented_status import InstanceRentedBusinessStatus
-from thestage.services.abstract_service import AbstractService
-from thestage.helpers.error_handler import error_handler
-from thestage.services.clients.thestage_api.api_client import TheStageApiClient
-from thestage.services.clients.thestage_api.dtos.instance_rented_response import InstanceRentedDto
-from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
-from thestage.services.clients.thestage_api.dtos.selfhosted_instance_response import SelfHostedInstanceDto
-from thestage.services.config_provider.config_provider import ConfigProvider
-from thestage.services.instance.mapper.instance_mapper import InstanceMapper
-from thestage.services.instance.mapper.selfhosted_mapper import SelfHostedMapper
-from thestage.services.remote_server_service import RemoteServerService
-
-
-class InstanceService(AbstractService):
-
-    __thestage_api_client: TheStageApiClient = None
-    __config_provider: ConfigProvider = None
-
-    def __init__(
-            self,
-            thestage_api_client: TheStageApiClient,
-            config_provider: ConfigProvider,
-            remote_server_service: RemoteServerService,
-    ):
-        self.__thestage_api_client = thestage_api_client
-        self.__remote_server_service = remote_server_service
-        self.__config_provider = config_provider
-
-    def get_rented_instance(
-            self,
-            instance_slug: str,
-    ) -> Optional[InstanceRentedDto]:
-        return self.__thestage_api_client.get_rented_instance(
-            instance_slug=instance_slug,
-        )
-
-    def get_self_hosted_instance(
-            self,
-            instance_slug: str,
-    ) -> Optional[SelfHostedInstanceDto]:
-        return self.__thestage_api_client.get_selfhosted_instance(
-            instance_slug=instance_slug,
-        )
-
-    @error_handler()
-    def check_instance_status_to_connect(
-            self,
-            instance: InstanceRentedDto,
-    ) -> InstanceRentedDto:
-        if instance:
-            if instance.frontend_status.status_key in [
-                InstanceRentedBusinessStatus.IN_QUEUE.name,
-                InstanceRentedBusinessStatus.CREATING.name,
-                InstanceRentedBusinessStatus.REBOOTING.name,
-                InstanceRentedBusinessStatus.STARTING.name,
-            ]:
-                typer.echo(__('Cannot connect to rented server instance: it is either in the process of being rented or rebooted'))
-                raise typer.Exit(1)
-            elif instance.frontend_status.status_key in [
-                InstanceRentedBusinessStatus.TERMINATING.name,
-                InstanceRentedBusinessStatus.RENTAL_ERROR.name,
-            ]:
-                typer.echo(__('Cannot connect to rented server instance: renting process failed'))
-                raise typer.Exit(1)
-            elif instance.frontend_status.status_key in [
-                InstanceRentedBusinessStatus.STOPPED.name,
-                InstanceRentedBusinessStatus.STOPPING.name,
-                InstanceRentedBusinessStatus.DELETED.name,
-            ]:
-                typer.echo(__('Cannot connect to rented server instance: it is either stopped or has been deleted'))
-                raise typer.Exit(1)
-            elif instance.frontend_status.status_key in [
-                InstanceRentedBusinessStatus.UNKNOWN.name,
-                InstanceRentedBusinessStatus.ALL.name,
-            ]:
-                typer.echo(__('Cannot connect to rented server instance: instance status unknown'))
-                raise typer.Exit(1)
-
-        return instance
-
-    @error_handler()
-    def check_selfhosted_status_to_connect(
-            self,
-            instance: SelfHostedInstanceDto,
-    ) -> SelfHostedInstanceDto:
-        if instance:
-            if instance.frontend_status.status_key in [
-                SelfhostedBusinessStatus.AWAITING_CONFIGURATION.name,
-            ]:
-                typer.echo(__('Cannot connect to self-hosted instance: it is awaiting configuration'))
-                raise typer.Exit(1)
-            elif instance.frontend_status.status_key in [
-                SelfhostedBusinessStatus.UNREACHABLE_DAEMON.name,
-                SelfhostedBusinessStatus.DELETED.name,
-            ]:
-                typer.echo(__('Cannot connect to self-hosted instance: it may be turned off or unreachable'))
-                raise typer.Exit(1)
-            elif instance.frontend_status.status_key in [
-                SelfhostedBusinessStatus.UNKNOWN.name,
-                SelfhostedBusinessStatus.ALL.name,
-            ]:
-                typer.echo(__('Cannot connect to self-hosted instance: instance status unknown'))
-                raise typer.Exit(1)
-
-        return instance
-
-    @error_handler()
-    def connect_to_rented_instance(
-            self,
-            instance_rented_slug: str,
-            input_ssh_key_path: Optional[str]
-    ):
-        instance = self.get_rented_instance(instance_slug=instance_rented_slug)
-
-        if instance:
-            self.check_instance_status_to_connect(
-                instance=instance,
-            )
-
-            ssh_path_from_config: Optional[str] = None
-            if not input_ssh_key_path:
-                ssh_path_from_config = self.__config_provider.get_valid_private_key_path_by_ip_address(instance.ip_address)
-                if ssh_path_from_config:
-                    typer.echo(f"Using configured ssh key for this instance: {ssh_path_from_config}")
-
-            if not input_ssh_key_path and not ssh_path_from_config:
-                typer.echo('Using SSH agent to connect to server instance')
-
-            self.__remote_server_service.connect_to_instance(
-                ip_address=instance.ip_address,
-                username=instance.host_username,
-                private_key_path=ssh_path_from_config or input_ssh_key_path
-            )
-
-            # cannot really detect how ssh connection was ended. capturing stderr using subprocess feels bad/unreliable.
-            if input_ssh_key_path:
-                self.__config_provider.update_remote_server_config_entry(ip_address=instance.ip_address, ssh_key_path=Path(input_ssh_key_path))
-        else:
-            typer.echo(__("Server instance not found: %instance_item%", {'instance_item': instance_rented_slug}))
-
-
-    @error_handler()
-    def connect_to_selfhosted_instance(
-            self,
-            selfhosted_instance_slug: str,
-            username: str,
-            input_ssh_key_path: Optional[str],
-    ):
-        if not username:
-            username = 'root'
-            typer.echo(__("No remote server username provided, using 'root' as username"))
-
-        instance = self.get_self_hosted_instance(instance_slug=selfhosted_instance_slug)
-
-        if instance:
-            self.check_selfhosted_status_to_connect(
-                instance=instance,
-            )
-
-            ssh_path_from_config: Optional[str] = None
-            if not input_ssh_key_path:
-                ssh_path_from_config = self.__config_provider.get_valid_private_key_path_by_ip_address(instance.ip_address)
-                if ssh_path_from_config:
-                    typer.echo(f"Using configured ssh key for this instance: {ssh_path_from_config}")
-
-            if not input_ssh_key_path and not ssh_path_from_config:
-                typer.echo('Using SSH agent to connect to server instance')
-
-            self.__remote_server_service.connect_to_instance(
-                ip_address=instance.ip_address,
-                username=username,
-                private_key_path=ssh_path_from_config or input_ssh_key_path
-            )
-
-            if input_ssh_key_path:
-                self.__config_provider.update_remote_server_config_entry(ip_address=instance.ip_address, ssh_key_path=Path(input_ssh_key_path))
-        else:
-            typer.echo(__("Server instance not found: %instance_item%", {'instance_item': selfhosted_instance_slug}))
-
-
-    @error_handler()
-    def get_rented_list(
-            self,
-            statuses: List[str],
-            row: int = 5,
-            page: int = 1,
-    ) -> PaginatedEntityList[InstanceRentedDto]:
-        data = self.__thestage_api_client.get_rented_instance_list(
-            statuses=statuses,
-            page=page,
-            limit=row,
-        )
-
-        return data
-
-    @error_handler()
-    def get_self_hosted_list(
-            self,
-            statuses: List[str],
-            row: int = 5,
-            page: int = 1,
-    ) -> PaginatedEntityList[SelfHostedInstanceDto]:
-        data = self.__thestage_api_client.get_selfhosted_instance_list(
-            statuses=statuses,
-            page=page,
-            limit=row,
-        )
-        return data
-
-
-    @error_handler()
-    def print_self_hosted_instance_list(self, statuses, row, page):
-        selfhosted_instance_status_map = self.__thestage_api_client.get_selfhosted_business_status_map()
-
-        if not statuses:
-            statuses = ({key: selfhosted_instance_status_map[key] for key in [
-                SelfhostedBusinessStatus.AWAITING_CONFIGURATION,
-                SelfhostedBusinessStatus.RUNNING,
-                SelfhostedBusinessStatus.UNREACHABLE_DAEMON,
-            ]}).values()
-
-        if "all" in statuses:
-            statuses = selfhosted_instance_status_map.values()
-
-        for input_status_item in statuses:
-            if input_status_item not in selfhosted_instance_status_map.values():
-                typer.echo(__("'%invalid_status%' is not one of %valid_statuses%", {
-                    'invalid_status': input_status_item,
-                    'valid_statuses': str(list(selfhosted_instance_status_map.values()))
-                }))
-                raise typer.Exit(1)
-
-        typer.echo(__(
-            "Listing self-hosted instances with the following statuses: %statuses%, to view all self-hosted instances, use --status all",
-            placeholders={
-                'statuses': ', '.join([status_item for status_item in statuses])
-            }))
-
-        backend_statuses: List[str] = [key for key, value in selfhosted_instance_status_map.items() if value in statuses]
-
-        self.print(
-            func_get_data=self.get_self_hosted_list,
-            func_special_params={
-                'statuses': backend_statuses,
-            },
-            mapper=SelfHostedMapper(),
-            headers=list(map(lambda x: x.alias, SelfHostedInstanceEntity.model_fields.values())),
-            row=row,
-            page=page,
-            show_index="never",
-        )
-
-
-    @error_handler()
-    def print_rented_instance_list(self, statuses, row, page):
-        instance_rented_status_map = self.__thestage_api_client.get_rented_business_status_map()
-
-        if not statuses:
-            statuses = ({key: instance_rented_status_map[key] for key in [
-                InstanceRentedBusinessStatus.ONLINE,
-                InstanceRentedBusinessStatus.CREATING,
-                InstanceRentedBusinessStatus.TERMINATING,
-                InstanceRentedBusinessStatus.REBOOTING,
-            ]}).values()
-
-        if "all" in statuses:
-            statuses = instance_rented_status_map.values()
-
-        for input_status_item in statuses:
-            if input_status_item not in instance_rented_status_map.values():
-                typer.echo(__("'%invalid_status%' is not one of %valid_statuses%", {
-                    'invalid_status': input_status_item,
-                    'valid_statuses': str(list(instance_rented_status_map.values()))
-                }))
-                raise typer.Exit(1)
-
-        typer.echo(__(
-            "Listing rented server instances with the following statuses: %statuses%, to view all rented server instances, use --status all",
-            placeholders={
-                'statuses': ', '.join([status_item for status_item in statuses])
-            }))
-
-        backend_statuses: List[str] = [key for key, value in instance_rented_status_map.items() if value in statuses]
-
-        self.print(
-            func_get_data=self.get_rented_list,
-            func_special_params={
-                'statuses': backend_statuses,
-            },
-            mapper=InstanceMapper(),
-            headers=list(map(lambda x: x.alias, RentedInstanceEntity.model_fields.values())),
-            row=row,
-            page=page,
-            show_index="never",
+from pathlib import Path
+from typing import List, Optional, Dict
+
+import typer
+
+from thestage.entities.rented_instance import RentedInstanceEntity
+from thestage.entities.self_hosted_instance import SelfHostedInstanceEntity
+from thestage.i18n.translation import __
+from thestage.services.clients.thestage_api.dtos.enums.selfhosted_status import SelfhostedBusinessStatus
+from thestage.services.clients.thestage_api.dtos.enums.instance_rented_status import InstanceRentedBusinessStatus
+from thestage.services.abstract_service import AbstractService
+from thestage.helpers.error_handler import error_handler
+from thestage.services.clients.thestage_api.api_client import TheStageApiClient
+from thestage.services.clients.thestage_api.dtos.instance_rented_response import InstanceRentedDto
+from thestage.services.clients.thestage_api.dtos.paginated_entity_list import PaginatedEntityList
+from thestage.services.clients.thestage_api.dtos.selfhosted_instance_response import SelfHostedInstanceDto
+from thestage.services.config_provider.config_provider import ConfigProvider
+from thestage.services.instance.mapper.instance_mapper import InstanceMapper
+from thestage.services.instance.mapper.selfhosted_mapper import SelfHostedMapper
+from thestage.services.remote_server_service import RemoteServerService
+
+
+class InstanceService(AbstractService):
+
+    __thestage_api_client: TheStageApiClient = None
+    __config_provider: ConfigProvider = None
+
+    def __init__(
+            self,
+            thestage_api_client: TheStageApiClient,
+            config_provider: ConfigProvider,
+            remote_server_service: RemoteServerService,
+    ):
+        self.__thestage_api_client = thestage_api_client
+        self.__remote_server_service = remote_server_service
+        self.__config_provider = config_provider
+
+    def get_rented_instance(
+            self,
+            instance_slug: str,
+    ) -> Optional[InstanceRentedDto]:
+        return self.__thestage_api_client.get_rented_instance(
+            instance_slug=instance_slug,
+        )
+
+    def get_self_hosted_instance(
+            self,
+            instance_slug: str,
+    ) -> Optional[SelfHostedInstanceDto]:
+        return self.__thestage_api_client.get_selfhosted_instance(
+            instance_slug=instance_slug,
+        )
+
+    @error_handler()
+    def check_instance_status_to_connect(
+            self,
+            instance: InstanceRentedDto,
+    ) -> InstanceRentedDto:
+        if instance:
+            if instance.frontend_status.status_key in [
+                InstanceRentedBusinessStatus.IN_QUEUE.name,
+                InstanceRentedBusinessStatus.CREATING.name,
+                InstanceRentedBusinessStatus.REBOOTING.name,
+                InstanceRentedBusinessStatus.STARTING.name,
+            ]:
+                typer.echo(__('Cannot connect to rented server instance: it is either in the process of being rented or rebooted'))
+                raise typer.Exit(1)
+            elif instance.frontend_status.status_key in [
+                InstanceRentedBusinessStatus.TERMINATING.name,
+                InstanceRentedBusinessStatus.RENTAL_ERROR.name,
+            ]:
+                typer.echo(__('Cannot connect to rented server instance: renting process failed'))
+                raise typer.Exit(1)
+            elif instance.frontend_status.status_key in [
+                InstanceRentedBusinessStatus.STOPPED.name,
+                InstanceRentedBusinessStatus.STOPPING.name,
+                InstanceRentedBusinessStatus.DELETED.name,
+            ]:
+                typer.echo(__('Cannot connect to rented server instance: it is either stopped or has been deleted'))
+                raise typer.Exit(1)
+            elif instance.frontend_status.status_key in [
+                InstanceRentedBusinessStatus.UNKNOWN.name,
+                InstanceRentedBusinessStatus.ALL.name,
+            ]:
+                typer.echo(__('Cannot connect to rented server instance: instance status unknown'))
+                raise typer.Exit(1)
+
+        return instance
+
+    @error_handler()
+    def check_selfhosted_status_to_connect(
+            self,
+            instance: SelfHostedInstanceDto,
+    ) -> SelfHostedInstanceDto:
+        if instance:
+            if instance.frontend_status.status_key in [
+                SelfhostedBusinessStatus.AWAITING_CONFIGURATION.name,
+            ]:
+                typer.echo(__('Cannot connect to self-hosted instance: it is awaiting configuration'))
+                raise typer.Exit(1)
+            elif instance.frontend_status.status_key in [
+                SelfhostedBusinessStatus.UNREACHABLE_DAEMON.name,
+                SelfhostedBusinessStatus.DELETED.name,
+            ]:
+                typer.echo(__('Cannot connect to self-hosted instance: it may be turned off or unreachable'))
+                raise typer.Exit(1)
+            elif instance.frontend_status.status_key in [
+                SelfhostedBusinessStatus.UNKNOWN.name,
+                SelfhostedBusinessStatus.ALL.name,
+            ]:
+                typer.echo(__('Cannot connect to self-hosted instance: instance status unknown'))
+                raise typer.Exit(1)
+
+        return instance
+
+    @error_handler()
+    def connect_to_rented_instance(
+            self,
+            instance_rented_slug: str,
+            input_ssh_key_path: Optional[str]
+    ):
+        instance = self.get_rented_instance(instance_slug=instance_rented_slug)
+
+        if instance:
+            self.check_instance_status_to_connect(
+                instance=instance,
+            )
+
+            ssh_path_from_config: Optional[str] = None
+            if not input_ssh_key_path:
+                ssh_path_from_config = self.__config_provider.get_valid_private_key_path_by_ip_address(instance.ip_address)
+                if ssh_path_from_config:
+                    typer.echo(f"Using configured ssh key for this instance: {ssh_path_from_config}")
+
+            if not input_ssh_key_path and not ssh_path_from_config:
+                typer.echo('Using SSH agent to connect to server instance')
+
+            self.__remote_server_service.connect_to_instance(
+                ip_address=instance.ip_address,
+                username=instance.host_username,
+                private_key_path=ssh_path_from_config or input_ssh_key_path
+            )
+
+            # cannot really detect how ssh connection was ended. capturing stderr using subprocess feels bad/unreliable.
+            if input_ssh_key_path:
+                self.__config_provider.update_remote_server_config_entry(ip_address=instance.ip_address, ssh_key_path=Path(input_ssh_key_path))
+        else:
+            typer.echo(__("Server instance not found: %instance_item%", {'instance_item': instance_rented_slug}))
+
+
+    @error_handler()
+    def connect_to_selfhosted_instance(
+            self,
+            selfhosted_instance_slug: str,
+            username: str,
+            input_ssh_key_path: Optional[str],
+    ):
+        if not username:
+            username = 'root'
+            typer.echo(__("No remote server username provided, using 'root' as username"))
+
+        instance = self.get_self_hosted_instance(instance_slug=selfhosted_instance_slug)
+
+        if instance:
+            self.check_selfhosted_status_to_connect(
+                instance=instance,
+            )
+
+            ssh_path_from_config: Optional[str] = None
+            if not input_ssh_key_path:
+                ssh_path_from_config = self.__config_provider.get_valid_private_key_path_by_ip_address(instance.ip_address)
+                if ssh_path_from_config:
+                    typer.echo(f"Using configured ssh key for this instance: {ssh_path_from_config}")
+
+            if not input_ssh_key_path and not ssh_path_from_config:
+                typer.echo('Using SSH agent to connect to server instance')
+
+            self.__remote_server_service.connect_to_instance(
+                ip_address=instance.ip_address,
+                username=username,
+                private_key_path=ssh_path_from_config or input_ssh_key_path
+            )
+
+            if input_ssh_key_path:
+                self.__config_provider.update_remote_server_config_entry(ip_address=instance.ip_address, ssh_key_path=Path(input_ssh_key_path))
+        else:
+            typer.echo(__("Server instance not found: %instance_item%", {'instance_item': selfhosted_instance_slug}))
+
+
+    @error_handler()
+    def get_rented_list(
+            self,
+            statuses: List[str],
+            row: int = 5,
+            page: int = 1,
+    ) -> PaginatedEntityList[InstanceRentedDto]:
+        data = self.__thestage_api_client.get_rented_instance_list(
+            statuses=statuses,
+            page=page,
+            limit=row,
+        )
+
+        return data
+
+    @error_handler()
+    def get_self_hosted_list(
+            self,
+            statuses: List[str],
+            row: int = 5,
+            page: int = 1,
+    ) -> PaginatedEntityList[SelfHostedInstanceDto]:
+        data = self.__thestage_api_client.get_selfhosted_instance_list(
+            statuses=statuses,
+            page=page,
+            limit=row,
+        )
+        return data
+
+
+    @error_handler()
+    def print_self_hosted_instance_list(self, statuses, row, page):
+        selfhosted_instance_status_map = self.__thestage_api_client.get_selfhosted_business_status_map()
+
+        if not statuses:
+            statuses = ({key: selfhosted_instance_status_map[key] for key in [
+                SelfhostedBusinessStatus.AWAITING_CONFIGURATION,
+                SelfhostedBusinessStatus.RUNNING,
+                SelfhostedBusinessStatus.UNREACHABLE_DAEMON,
+            ]}).values()
+
+        if "all" in statuses:
+            statuses = selfhosted_instance_status_map.values()
+
+        for input_status_item in statuses:
+            if input_status_item not in selfhosted_instance_status_map.values():
+                typer.echo(__("'%invalid_status%' is not one of %valid_statuses%", {
+                    'invalid_status': input_status_item,
+                    'valid_statuses': str(list(selfhosted_instance_status_map.values()))
+                }))
+                raise typer.Exit(1)
+
+        typer.echo(__(
+            "Listing self-hosted instances with the following statuses: %statuses%, to view all self-hosted instances, use --status all",
+            placeholders={
+                'statuses': ', '.join([status_item for status_item in statuses])
+            }))
+
+        backend_statuses: List[str] = [key for key, value in selfhosted_instance_status_map.items() if value in statuses]
+
+        self.print(
+            func_get_data=self.get_self_hosted_list,
+            func_special_params={
+                'statuses': backend_statuses,
+            },
+            mapper=SelfHostedMapper(),
+            headers=list(map(lambda x: x.alias, SelfHostedInstanceEntity.model_fields.values())),
+            row=row,
+            page=page,
+            show_index="never",
+        )
+
+
+    @error_handler()
+    def print_rented_instance_list(self, statuses, row, page):
+        instance_rented_status_map = self.__thestage_api_client.get_rented_business_status_map()
+
+        if not statuses:
+            statuses = ({key: instance_rented_status_map[key] for key in [
+                InstanceRentedBusinessStatus.ONLINE,
+                InstanceRentedBusinessStatus.CREATING,
+                InstanceRentedBusinessStatus.TERMINATING,
+                InstanceRentedBusinessStatus.REBOOTING,
+            ]}).values()
+
+        if "all" in statuses:
+            statuses = instance_rented_status_map.values()
+
+        for input_status_item in statuses:
+            if input_status_item not in instance_rented_status_map.values():
+                typer.echo(__("'%invalid_status%' is not one of %valid_statuses%", {
+                    'invalid_status': input_status_item,
+                    'valid_statuses': str(list(instance_rented_status_map.values()))
+                }))
+                raise typer.Exit(1)
+
+        typer.echo(__(
+            "Listing rented server instances with the following statuses: %statuses%, to view all rented server instances, use --status all",
+            placeholders={
+                'statuses': ', '.join([status_item for status_item in statuses])
+            }))
+
+        backend_statuses: List[str] = [key for key, value in instance_rented_status_map.items() if value in statuses]
+
+        self.print(
+            func_get_data=self.get_rented_list,
+            func_special_params={
+                'statuses': backend_statuses,
+            },
+            mapper=InstanceMapper(),
+            headers=list(map(lambda x: x.alias, RentedInstanceEntity.model_fields.values())),
+            row=row,
+            page=page,
+            show_index="never",
         )

thestage/services/instance/mapper/instance_mapper.py

@@ -1,24 +1,24 @@
-from typing import Optional
-
-from thestage.entities.rented_instance import RentedInstanceEntity
-from thestage.services.clients.thestage_api.dtos.instance_rented_response import InstanceRentedDto
-from thestage.services.abstract_mapper import AbstractMapper
-
-
-class InstanceMapper(AbstractMapper):
-
-    def build_entity(self, item: InstanceRentedDto) -> Optional[RentedInstanceEntity]:
-        if not item:
-            return None
-
-        return RentedInstanceEntity(
-            slug=item.slug if item.slug else '',
-            title=item.title if item.title else '',
-            cpu_type=item.cpu_type if item.cpu_type else '',
-            gpu_type=item.gpu_type if item.gpu_type else '',
-            cpu_cores=str(item.cpu_cores) if item.cpu_cores else '',
-            ip_address=item.ip_address if item.ip_address else '',
-            status=item.frontend_status.status_translation if item.frontend_status else '',
-            created_at=str(item.created_at.strftime("%Y-%m-%d %H:%M:%S")) if item.created_at else '',
-            updated_at=str(item.updated_at.strftime("%Y-%m-%d %H:%M:%S")) if item.updated_at else '',
-        )
+from typing import Optional
+
+from thestage.entities.rented_instance import RentedInstanceEntity
+from thestage.services.clients.thestage_api.dtos.instance_rented_response import InstanceRentedDto
+from thestage.services.abstract_mapper import AbstractMapper
+
+
+class InstanceMapper(AbstractMapper):
+
+    def build_entity(self, item: InstanceRentedDto) -> Optional[RentedInstanceEntity]:
+        if not item:
+            return None
+
+        return RentedInstanceEntity(
+            slug=item.slug if item.slug else '',
+            title=item.title if item.title else '',
+            cpu_type=item.cpu_type if item.cpu_type else '',
+            gpu_type=item.gpu_type if item.gpu_type else '',
+            cpu_cores=str(item.cpu_cores) if item.cpu_cores else '',
+            ip_address=item.ip_address if item.ip_address else '',
+            status=item.frontend_status.status_translation if item.frontend_status else '',
+            created_at=str(item.created_at.strftime("%Y-%m-%d %H:%M:%S")) if item.created_at else '',
+            updated_at=str(item.updated_at.strftime("%Y-%m-%d %H:%M:%S")) if item.updated_at else '',
+        )