thestage 0.6.7__py3-none-any.whl → 0.7.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (202) hide show
  1. thestage/__init__.py +1 -1
  2. thestage/cli_command.py +3 -0
  3. thestage/cli_command_helper.py +2 -2
  4. thestage/config/__init__.py +1 -1
  5. thestage/{services → config/business}/app_config_service.py +3 -4
  6. thestage/{services/config_provider → config/business}/config_provider.py +6 -5
  7. thestage/config/{config_storage.py → business/config_storage.py} +1 -1
  8. thestage/{services → config/business}/validation_service.py +9 -10
  9. thestage/{controllers/config_controller.py → config/communication/config_command.py} +7 -7
  10. thestage/{services/connect → connect/business}/connect_service.py +22 -19
  11. thestage/{services → connect/business}/remote_server_service.py +4 -5
  12. thestage/connect/communication/connect_api_client.py +84 -0
  13. thestage/{services/clients/thestage_api/dtos/ssh_key_controller → connect/dto}/add_ssh_key_to_user_response.py +0 -1
  14. thestage/{services/clients/thestage_api/dtos/ssh_key_controller → connect/dto}/add_ssh_public_key_to_instance_response.py +1 -4
  15. thestage/{services/clients/thestage_api/dtos/base_controller → connect/dto}/connect_resolve_response.py +0 -1
  16. thestage/controllers/base_controller.py +2 -2
  17. thestage/debug_main.dist.py +16 -14
  18. thestage/{services/container → docker_container/business}/container_service.py +147 -22
  19. thestage/{services/container → docker_container/business}/mapper/container_mapper.py +3 -3
  20. thestage/docker_container/business/mapper/image_mapper.py +16 -0
  21. thestage/docker_container/communication/__init__.py +0 -0
  22. thestage/{controllers/container_controller.py → docker_container/communication/docker_command.py} +192 -29
  23. thestage/docker_container/communication/docker_container_api_client.py +139 -0
  24. thestage/docker_container/communication/image_command.py +40 -0
  25. thestage/docker_container/dto/__init__.py +0 -0
  26. thestage/docker_container/dto/container_action_request.py +11 -0
  27. thestage/{services/clients/thestage_api/dtos → docker_container/dto}/container_response.py +11 -4
  28. thestage/docker_container/dto/docker_container_create_request.py +28 -0
  29. thestage/docker_container/dto/docker_container_create_response.py +11 -0
  30. thestage/{services/clients/thestage_api/dtos/docker_container_controller → docker_container/dto}/docker_container_list_response.py +3 -5
  31. thestage/docker_container/dto/docker_image_list_request.py +11 -0
  32. thestage/docker_container/dto/docker_image_list_response.py +13 -0
  33. thestage/docker_container/dto/enum/__init__.py +0 -0
  34. thestage/docker_container/dto/image_entity.py +14 -0
  35. thestage/git/__init__.py +0 -0
  36. thestage/git/business/__init__.py +0 -0
  37. thestage/git/communication/__init__.py +0 -0
  38. thestage/{services/clients/git → git/communication}/git_client.py +4 -4
  39. thestage/global_dto/__init__.py +0 -0
  40. thestage/global_dto/enums/__init__.py +0 -0
  41. thestage/helpers/error_handler.py +4 -4
  42. thestage/helpers/logger/app_logger.py +1 -3
  43. thestage/i18n/en_GB/messages.po +14 -14
  44. thestage/inference_model/__init__.py +0 -0
  45. thestage/inference_model/business/__init__.py +0 -0
  46. thestage/inference_model/business/inference_model_service.py +281 -0
  47. thestage/inference_model/business/mapper/__init__.py +0 -0
  48. thestage/{services/project/mapper/project_inference_simulator_model_mapper.py → inference_model/business/mapper/inference_model_mapper.py} +5 -5
  49. thestage/inference_model/communication/__init__.py +0 -0
  50. thestage/inference_model/communication/inference_model_api_client.py +139 -0
  51. thestage/inference_model/communication/inference_model_command.py +246 -0
  52. thestage/inference_model/dto/__init__.py +0 -0
  53. thestage/{services/clients/thestage_api/dtos/inference_controller → inference_model/dto}/deploy_inference_model_to_instance_response.py +0 -1
  54. thestage/inference_model/dto/enum/__init__.py +0 -0
  55. thestage/{services/project/dto/inference_simulator_model_dto.py → inference_model/dto/inference_model.py} +1 -1
  56. thestage/{entities/project_inference_simulator_model.py → inference_model/dto/inference_model_entity.py} +1 -1
  57. thestage/{services/clients/thestage_api/dtos/inference_controller → inference_model/dto}/inference_simulator_model_list_for_project_response.py +2 -3
  58. thestage/{services/clients/thestage_api/dtos/project_controller/project_push_inference_simulator_model_request.py → inference_model/dto/push_inference_simulator_model_request.py} +1 -1
  59. thestage/{services/clients/thestage_api/dtos/project_controller/project_push_inference_simulator_model_response.py → inference_model/dto/push_inference_simulator_model_response.py} +1 -1
  60. thestage/inference_simulator/__init__.py +0 -0
  61. thestage/inference_simulator/business/__init__.py +0 -0
  62. thestage/inference_simulator/business/inference_simulator_service.py +338 -0
  63. thestage/inference_simulator/business/mapper/__init__.py +0 -0
  64. thestage/{services/project/mapper/project_inference_simulator_mapper.py → inference_simulator/business/mapper/inference_simulator_mapper.py} +5 -5
  65. thestage/inference_simulator/communication/__init__.py +0 -0
  66. thestage/inference_simulator/communication/inference_simulator_api_client.py +114 -0
  67. thestage/inference_simulator/communication/inference_simulator_command.py +347 -0
  68. thestage/inference_simulator/dto/__init__.py +0 -0
  69. thestage/inference_simulator/dto/enum/__init__.py +0 -0
  70. thestage/inference_simulator/dto/get_inference_simulator_response.py +12 -0
  71. thestage/{services/project/dto/inference_simulator_dto.py → inference_simulator/dto/inference_simulator.py} +1 -1
  72. thestage/{entities/project_inference_simulator.py → inference_simulator/dto/inference_simulator_entity.py} +1 -1
  73. thestage/{services/clients/thestage_api/dtos/inference_controller → inference_simulator/dto}/inference_simulator_list_response.py +2 -2
  74. thestage/{services/clients/thestage_api/dtos/project_controller/project_start_inference_simulator_request.py → inference_simulator/dto/start_inference_simulator_request.py} +1 -1
  75. thestage/inference_simulator/dto/start_inference_simulator_response.py +10 -0
  76. thestage/instance/__init__.py +0 -0
  77. thestage/instance/business/__init__.py +0 -0
  78. thestage/{services/instance → instance/business}/instance_service.py +26 -27
  79. thestage/instance/business/mapper/__init__.py +0 -0
  80. thestage/{services/instance/mapper/instance_mapper.py → instance/business/mapper/rented_instance_mapper.py} +3 -3
  81. thestage/{services/instance/mapper/selfhosted_mapper.py → instance/business/mapper/selfhosted_instance_mapper.py} +5 -7
  82. thestage/instance/communication/__init__.py +0 -0
  83. thestage/instance/communication/instance_api_client.py +150 -0
  84. thestage/{controllers/instance_controller.py → instance/communication/instance_command.py} +5 -5
  85. thestage/instance/dto/__init__.py +0 -0
  86. thestage/instance/dto/enum/__init__.py +0 -0
  87. thestage/{services/clients/thestage_api/dtos → instance/dto}/instance_detected_gpus.py +1 -2
  88. thestage/{services/clients/thestage_api/dtos → instance/dto}/instance_rented_response.py +2 -2
  89. thestage/{services/clients/thestage_api/dtos → instance/dto}/selfhosted_instance_response.py +2 -3
  90. thestage/logging/__init__.py +0 -0
  91. thestage/logging/business/__init__.py +0 -0
  92. thestage/{services/logging → logging/business}/logging_service.py +40 -28
  93. thestage/logging/communication/__init__.py +0 -0
  94. thestage/logging/communication/logging_api_client.py +63 -0
  95. thestage/logging/dto/__init__.py +0 -0
  96. thestage/{services/clients/thestage_api/dtos/logging_controller → logging/dto}/log_polling_response.py +2 -2
  97. thestage/{services/clients/thestage_api/dtos/logging_controller → logging/dto}/user_logs_query_response.py +2 -2
  98. thestage/main.py +54 -8
  99. thestage/project/__init__.py +0 -0
  100. thestage/project/business/__init__.py +0 -0
  101. thestage/project/business/project_service.py +480 -0
  102. thestage/project/communication/__init__.py +0 -0
  103. thestage/project/communication/project_api_client.py +46 -0
  104. thestage/project/communication/project_command.py +284 -0
  105. thestage/project/dto/__init__.py +0 -0
  106. thestage/services/clients/thestage_api/core/api_client_core.py +1 -1
  107. thestage/services/clients/thestage_api/dtos/entity_filter_request.py +1 -1
  108. thestage/services/clients/thestage_api/dtos/sftp_path_helper.py +1 -1
  109. thestage/services/filesystem_service.py +2 -2
  110. thestage/services/service_factory.py +130 -43
  111. thestage/task/__init__.py +0 -0
  112. thestage/task/business/__init__.py +0 -0
  113. thestage/task/business/mapper/__init__.py +0 -0
  114. thestage/{services/project/mapper/project_task_mapper.py → task/business/mapper/task_mapper.py} +5 -5
  115. thestage/task/business/task_service.py +307 -0
  116. thestage/task/communication/__init__.py +0 -0
  117. thestage/task/communication/task_api_client.py +122 -0
  118. thestage/task/communication/task_command.py +212 -0
  119. thestage/task/dto/__init__.py +0 -0
  120. thestage/task/dto/enum/__init__.py +0 -0
  121. thestage/{services/clients/thestage_api/dtos/task_controller/task_list_for_project_response.py → task/dto/list_for_project_response.py} +2 -2
  122. thestage/{services/clients/thestage_api/dtos/project_controller/project_run_task_request.py → task/dto/run_task_request.py} +1 -1
  123. thestage/task/dto/run_task_response.py +13 -0
  124. thestage/{services/task/dto/task_dto.py → task/dto/task.py} +1 -4
  125. thestage/{entities/project_task.py → task/dto/task_entity.py} +1 -1
  126. thestage/{services/clients/thestage_api/dtos/task_controller/task_view_response.py → task/dto/view_response.py} +2 -2
  127. {thestage-0.6.7.dist-info → thestage-0.7.0.dist-info}/METADATA +1 -1
  128. thestage-0.7.0.dist-info/RECORD +226 -0
  129. {thestage-0.6.7.dist-info → thestage-0.7.0.dist-info}/WHEEL +1 -1
  130. thestage/controllers/project_controller.py +0 -1056
  131. thestage/services/clients/thestage_api/api_client.py +0 -751
  132. thestage/services/clients/thestage_api/dtos/container_param_request.py +0 -11
  133. thestage/services/clients/thestage_api/dtos/inference_controller/get_inference_simulator_response.py +0 -13
  134. thestage/services/clients/thestage_api/dtos/project_controller/project_run_task_response.py +0 -13
  135. thestage/services/clients/thestage_api/dtos/project_controller/project_start_inference_simulator_response.py +0 -10
  136. thestage/services/clients/thestage_api/dtos/user_controller/user_profile.py +0 -12
  137. thestage/services/project/project_service.py +0 -1287
  138. thestage-0.6.7.dist-info/RECORD +0 -167
  139. /thestage/{entities → color_scheme}/__init__.py +0 -0
  140. /thestage/{entities/enums → config/business}/__init__.py +0 -0
  141. /thestage/{services/clients/git → config/communication}/__init__.py +0 -0
  142. /thestage/{services/clients/thestage_api/dtos/enums → config/dto}/__init__.py +0 -0
  143. /thestage/{services/core_files → config/dto}/config_entity.py +0 -0
  144. /thestage/{services/connect → config}/dto/remote_server_config.py +0 -0
  145. /thestage/{services/config_provider → connect}/__init__.py +0 -0
  146. /thestage/{services/container → connect/business}/__init__.py +0 -0
  147. /thestage/{services/container/mapper → connect/communication}/__init__.py +0 -0
  148. /thestage/{services/instance → connect/dto}/__init__.py +0 -0
  149. /thestage/{services/clients/thestage_api/dtos/ssh_key_controller → connect/dto}/add_ssh_key_to_user_request.py +0 -0
  150. /thestage/{services/clients/thestage_api/dtos/ssh_key_controller → connect/dto}/add_ssh_public_key_to_instance_request.py +0 -0
  151. /thestage/{services/clients/thestage_api/dtos/ssh_key_controller → connect/dto}/is_user_has_public_ssh_key_request.py +0 -0
  152. /thestage/{services/clients/thestage_api/dtos/ssh_key_controller → connect/dto}/is_user_has_public_ssh_key_response.py +0 -0
  153. /thestage/{services/instance/mapper → docker_container}/__init__.py +0 -0
  154. /thestage/{services/project → docker_container/business}/__init__.py +0 -0
  155. /thestage/{services/project → docker_container/business}/mapper/__init__.py +0 -0
  156. /thestage/{entities/container.py → docker_container/dto/container_entity.py} +0 -0
  157. /thestage/{services/clients/thestage_api/dtos/docker_container_controller → docker_container/dto}/docker_container_list_request.py +0 -0
  158. /thestage/{services/clients/thestage_api/dtos → docker_container/dto}/docker_container_mapping.py +0 -0
  159. /thestage/{services/clients/thestage_api/dtos/enums → docker_container/dto/enum}/container_pending_action.py +0 -0
  160. /thestage/{services/clients/thestage_api/dtos/enums → docker_container/dto/enum}/container_status.py +0 -0
  161. /thestage/{services/logging/exception → exceptions}/log_polling_exception.py +0 -0
  162. /thestage/git/{ProgressPrinter.py → business/ProgressPrinter.py} +0 -0
  163. /thestage/{entities → global_dto}/enums/order_direction_type.py +0 -0
  164. /thestage/{entities → global_dto}/enums/shell_type.py +0 -0
  165. /thestage/{entities → global_dto}/enums/tail_output_type.py +0 -0
  166. /thestage/{entities → global_dto}/enums/yes_no_response.py +0 -0
  167. /thestage/{entities → global_dto}/file_item.py +0 -0
  168. /thestage/{services/clients/thestage_api/dtos/inference_controller → inference_model/dto}/deploy_inference_model_to_instance_request.py +0 -0
  169. /thestage/{services/clients/thestage_api/dtos/inference_controller → inference_model/dto}/deploy_inference_model_to_sagemaker_request.py +0 -0
  170. /thestage/{services/clients/thestage_api/dtos/inference_controller → inference_model/dto}/deploy_inference_model_to_sagemaker_response.py +0 -0
  171. /thestage/{services/clients/thestage_api/dtos/enums → inference_model/dto/enum}/inference_model_status.py +0 -0
  172. /thestage/{services/clients/thestage_api/dtos/inference_controller → inference_model/dto}/inference_simulator_model_list_for_project_request.py +0 -0
  173. /thestage/{services/clients/thestage_api/dtos → inference_model/dto}/inference_simulator_model_response.py +0 -0
  174. /thestage/{services/clients/thestage_api/dtos/enums → inference_simulator/dto/enum}/inference_simulator_status.py +0 -0
  175. /thestage/{services/clients/thestage_api/dtos/inference_controller → inference_simulator/dto}/get_inference_simulator_request.py +0 -0
  176. /thestage/{services/clients/thestage_api/dtos/inference_controller → inference_simulator/dto}/inference_simulator_list_request.py +0 -0
  177. /thestage/{services/clients/thestage_api/dtos → inference_simulator/dto}/inference_simulator_response.py +0 -0
  178. /thestage/{services/clients/thestage_api/dtos/enums → instance/dto/enum}/cpu_type.py +0 -0
  179. /thestage/{services/clients/thestage_api/dtos/enums → instance/dto/enum}/gpu_name.py +0 -0
  180. /thestage/{services/clients/thestage_api/dtos/enums → instance/dto/enum}/instance_rented_status.py +0 -0
  181. /thestage/{services/clients/thestage_api/dtos/enums → instance/dto/enum}/provider_name.py +0 -0
  182. /thestage/{services/clients/thestage_api/dtos/enums → instance/dto/enum}/selfhosted_status.py +0 -0
  183. /thestage/{entities → instance/dto}/rented_instance.py +0 -0
  184. /thestage/{entities → instance/dto}/self_hosted_instance.py +0 -0
  185. /thestage/{services/logging → logging}/byte_print_style.py +0 -0
  186. /thestage/{services/clients/thestage_api/dtos/logging_controller → logging/dto}/docker_container_log_stream_request.py +0 -0
  187. /thestage/{services/logging → logging}/dto/log_message.py +0 -0
  188. /thestage/{services/clients/thestage_api/dtos/logging_controller → logging/dto}/log_polling_request.py +0 -0
  189. /thestage/{services/logging → logging}/dto/log_type.py +0 -0
  190. /thestage/{services/clients/thestage_api/dtos/logging_controller → logging/dto}/task_log_stream_request.py +0 -0
  191. /thestage/{services/clients/thestage_api/dtos/logging_controller → logging/dto}/user_logs_query_request.py +0 -0
  192. /thestage/{services/logging → logging}/logging_constants.py +0 -0
  193. /thestage/{services/clients/thestage_api/dtos/project_controller/project_get_deploy_ssh_key_request.py → project/dto/get_deploy_ssh_key_request.py} +0 -0
  194. /thestage/{services/clients/thestage_api/dtos/project_controller/project_get_deploy_ssh_key_response.py → project/dto/get_deploy_ssh_key_response.py} +0 -0
  195. /thestage/{services/project → project}/dto/project_config.py +0 -0
  196. /thestage/{services/clients/thestage_api/dtos → project/dto}/project_response.py +0 -0
  197. /thestage/{services/clients/thestage_api/dtos/enums → task/dto/enum}/task_execution_status.py +0 -0
  198. /thestage/{services/clients/thestage_api/dtos/enums → task/dto/enum}/task_status.py +0 -0
  199. /thestage/{services/clients/thestage_api/dtos/task_controller/task_list_for_project_request.py → task/dto/list_for_project_request.py} +0 -0
  200. /thestage/{services/clients/thestage_api/dtos/task_controller/task_status_localized_map_response.py → task/dto/status_localized_map_response.py} +0 -0
  201. {thestage-0.6.7.dist-info → thestage-0.7.0.dist-info}/entry_points.txt +0 -0
  202. {thestage-0.6.7.dist-info → thestage-0.7.0.dist-info}/licenses/LICENSE.txt +0 -0
thestage/services/clients/thestage_api/dtos/container_param_request.py DELETED
@@ -1,11 +0,0 @@
1
- from typing import Optional, List
2
-
3
- from pydantic import Field, BaseModel, ConfigDict
4
-
5
- from thestage.services.clients.thestage_api.dtos.enums.container_pending_action import DockerContainerAction
6
-
7
- class DockerContainerActionRequestDto(BaseModel):
8
- model_config = ConfigDict(use_enum_values=True)
9
-
10
- container_public_id: Optional[str] = Field(None, alias='dockerContainerPublicId')
11
- action: DockerContainerAction = Field(None, alias='action')
thestage/services/clients/thestage_api/dtos/inference_controller/get_inference_simulator_response.py DELETED
@@ -1,13 +0,0 @@
1
- from typing import Optional, List
2
-
3
- from pydantic import Field, ConfigDict, BaseModel
4
-
5
- from thestage.services.clients.thestage_api.dtos.entity_filter_request import EntityFilterRequest
6
- from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
7
-
8
-
9
- class GetInferenceSimulatorResponse(BaseModel):
10
- model_config = ConfigDict(use_enum_values=True)
11
- inferenceSimulator: Optional[InferenceSimulatorDto] = Field(None, alias='inferenceSimulator')
12
-
13
-
thestage/services/clients/thestage_api/dtos/project_controller/project_run_task_response.py DELETED
@@ -1,13 +0,0 @@
1
- from typing import Optional, List
2
-
3
- from pydantic import Field, ConfigDict
4
-
5
- from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
6
- from thestage.services.task.dto.task_dto import TaskDto
7
-
8
-
9
- class ProjectRunTaskResponse(TheStageBaseResponse):
10
- model_config = ConfigDict(use_enum_values=True)
11
-
12
- task: TaskDto = Field(None, alias='task')
13
- tasksInQueue: Optional[List[TaskDto]] = Field(None, alias='tasksInQueue')
thestage/services/clients/thestage_api/dtos/project_controller/project_start_inference_simulator_response.py DELETED
@@ -1,10 +0,0 @@
1
- from pydantic import Field, ConfigDict
2
-
3
- from thestage.services.clients.thestage_api.dtos.base_response import TheStageBaseResponse
4
- from thestage.services.project.dto.inference_simulator_dto import InferenceSimulatorDto
5
-
6
-
7
- class ProjectStartInferenceSimulatorResponse(TheStageBaseResponse):
8
- model_config = ConfigDict(use_enum_values=True)
9
-
10
- inferenceSimulator: InferenceSimulatorDto = Field(None, alias='inferenceSimulator')
thestage/services/clients/thestage_api/dtos/user_controller/user_profile.py DELETED
@@ -1,12 +0,0 @@
1
- from typing import Optional
2
-
3
- from pydantic import BaseModel, Field
4
-
5
-
6
- class UserProfile(BaseModel):
7
- email: Optional[str] = Field(None, alias='email')
8
-
9
-
10
-
11
- class UserProfileResponse(BaseModel):
12
- userProfile: Optional[UserProfile] = Field(None, alias='userProfile')