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

thestage/.env

@@ -1,5 +1,5 @@
-; THESTAGE_CONFIG_DIR=.thestage
-; THESTAGE_CONFIG_FILE=config.json
-; THESTAGE_API_URL=https://backend-staging3.thestage.ai
-; THESTAGE_API_URL=http://localhost:8100
-; LOG_FILE=thestage.log
+THESTAGE_CONFIG_DIR=.thestage
+THESTAGE_CONFIG_FILE=config.json
+THESTAGE_API_URL=https://backend-staging.thestage.ai
+THESTAGE_API_URL=https://backend.thestage.ai
+LOG_FILE=thestage.log

thestage/__init__.py

@@ -1,4 +1,3 @@
-from . import *
-__app_name__ = "thestage"
-
-__version__ = "0.5.471"
+from . import *
+__app_name__ = "thestage"
+__version__ = "0.6.1"

thestage/__main__.py

@@ -1,9 +1,9 @@
-import thestage.main as main
-
-
-def start():
-    main.main()
-
-
-if __name__ == "__main__":
-    start()
+import thestage.main as main
+
+
+def start():
+    main.main()
+
+
+if __name__ == "__main__":
+    start()

thestage/cli_command.py

@@ -1,56 +1,56 @@
-from enum import Enum
-
-class CliCommand(str, Enum):
-    VERSION = "VERSION"
-    CONNECT = "CONNECT"
-    PROJECT_CLONE = "PROJECT_CLONE"
-    PROJECT_INIT = "PROJECT_INIT"
-    PROJECT_RUN = "PROJECT_RUN"
-    PROJECT_CHECKOUT = "PROJECT_CHECKOUT"
-    PROJECT_PULL = "PROJECT_PULL"
-    PROJECT_RESET = "PROJECT_RESET"
-    PROJECT_INFERENCE_SIMULATOR_RUN = "PROJECT_INFERENCE_SIMULATOR_RUN"
-    PROJECT_INFERENCE_SIMULATOR_SAVE_METADATA = "PROJECT_INFERENCE_SIMULATOR_SAVE_METADATA"
-    PROJECT_INFERENCE_SIMULATOR_PUSH = "PROJECT_INFERENCE_SIMULATOR_PUSH"
-    PROJECT_INFERENCE_SIMULATOR_LS = "PROJECT_INFERENCE_SIMULATOR_LS"
-    PROJECT_INFERENCE_SIMULATOR_LOGS = "PROJECT_INFERENCE_SIMULATOR_LOGS"
-    PROJECT_MODEL_LS = "PROJECT_MODEL_LS"
-    PROJECT_MODEL_DEPLOY_INSTANCE = "PROJECT_MODEL_DEPLOY_INSTANCE"
-    PROJECT_MODEL_DEPLOY_SAGEMAKER = "PROJECT_MODEL_DEPLOY_SAGEMAKER"
-    PROJECT_TASK_CANCEL = "PROJECT_TASK_CANCEL"
-    PROJECT_TASK_LS = "PROJECT_TASK_LS"
-    PROJECT_TASK_LOGS = "PROJECT_TASK_LOGS"
-    PROJECT_CONFIG_SET_DEFAULT_CONTAINER = "PROJECT_CONFIG_SET_DEFAULT_CONTAINER"
-    PROJECT_CONFIG_GET = "PROJECT_CONFIG_GET"
-    CONTAINER_LS = "CONTAINER_LS"
-    CONTAINER_INFO = "CONTAINER_INFO"
-    CONTAINER_CONNECT = "CONTAINER_CONNECT"
-    CONTAINER_UPLOAD = "CONTAINER_UPLOAD"
-    CONTAINER_DOWNLOAD = "CONTAINER_DOWNLOAD"
-    CONTAINER_START = "CONTAINER_START"
-    CONTAINER_STOP = "CONTAINER_STOP"
-    CONTAINER_RESTART = "CONTAINER_RESTART"
-    CONTAINER_LOGS = "CONTAINER_LOGS"
-    INSTANCE_RENTED_LS = "INSTANCE_RENTED_LS"
-    INSTANCE_RENTED_CONNECT = "INSTANCE_RENTED_CONNECT"
-    INSTANCE_SELF_HOSTED_LS = "INSTANCE_SELF_HOSTED_LS"
-    INSTANCE_SELF_HOSTED_CONNECT = "INSTANCE_SELF_HOSTED_CONNECT"
-    CONFIG_GET = "CONFIG_GET"
-    CONFIG_SET = "CONFIG_SET"
-    CONFIG_CLEAR = "CONFIG_CLEAR"
-    CONFIG_UPLOAD_SSH_KEY = "CONFIG_UPLOAD_SSH_KEY"
-
-
-class CliCommandAvailability(str, Enum):
-    ALLOWED = "ALLOWED"
-    RESTRICTED = "RESTRICTED"
-    DEPRECATED = "DEPRECATED"
-
-
-ALWAYS_AVAILABLE_COMMANDS = [
-    CliCommand.VERSION,
-    CliCommand.PROJECT_CONFIG_GET,
-    CliCommand.CONFIG_GET,
-    CliCommand.CONFIG_SET,
-    CliCommand.CONFIG_CLEAR,
-]
+from enum import Enum
+
+class CliCommand(str, Enum):
+    VERSION = "VERSION"
+    CONNECT = "CONNECT"
+    PROJECT_CLONE = "PROJECT_CLONE"
+    PROJECT_INIT = "PROJECT_INIT"
+    PROJECT_RUN = "PROJECT_RUN"
+    PROJECT_CHECKOUT = "PROJECT_CHECKOUT"
+    PROJECT_PULL = "PROJECT_PULL"
+    PROJECT_RESET = "PROJECT_RESET"
+    PROJECT_INFERENCE_SIMULATOR_RUN = "PROJECT_INFERENCE_SIMULATOR_RUN"
+    PROJECT_INFERENCE_SIMULATOR_SAVE_METADATA = "PROJECT_INFERENCE_SIMULATOR_SAVE_METADATA"
+    PROJECT_INFERENCE_SIMULATOR_PUSH = "PROJECT_INFERENCE_SIMULATOR_PUSH"
+    PROJECT_INFERENCE_SIMULATOR_LS = "PROJECT_INFERENCE_SIMULATOR_LS"
+    PROJECT_INFERENCE_SIMULATOR_LOGS = "PROJECT_INFERENCE_SIMULATOR_LOGS"
+    PROJECT_MODEL_LS = "PROJECT_MODEL_LS"
+    PROJECT_MODEL_DEPLOY_INSTANCE = "PROJECT_MODEL_DEPLOY_INSTANCE"
+    PROJECT_MODEL_DEPLOY_SAGEMAKER = "PROJECT_MODEL_DEPLOY_SAGEMAKER"
+    PROJECT_TASK_CANCEL = "PROJECT_TASK_CANCEL"
+    PROJECT_TASK_LS = "PROJECT_TASK_LS"
+    PROJECT_TASK_LOGS = "PROJECT_TASK_LOGS"
+    PROJECT_CONFIG_SET_DEFAULT_CONTAINER = "PROJECT_CONFIG_SET_DEFAULT_CONTAINER"
+    PROJECT_CONFIG_GET = "PROJECT_CONFIG_GET"
+    CONTAINER_LS = "CONTAINER_LS"
+    CONTAINER_INFO = "CONTAINER_INFO"
+    CONTAINER_CONNECT = "CONTAINER_CONNECT"
+    CONTAINER_UPLOAD = "CONTAINER_UPLOAD"
+    CONTAINER_DOWNLOAD = "CONTAINER_DOWNLOAD"
+    CONTAINER_START = "CONTAINER_START"
+    CONTAINER_STOP = "CONTAINER_STOP"
+    CONTAINER_RESTART = "CONTAINER_RESTART"
+    CONTAINER_LOGS = "CONTAINER_LOGS"
+    INSTANCE_RENTED_LS = "INSTANCE_RENTED_LS"
+    INSTANCE_RENTED_CONNECT = "INSTANCE_RENTED_CONNECT"
+    INSTANCE_SELF_HOSTED_LS = "INSTANCE_SELF_HOSTED_LS"
+    INSTANCE_SELF_HOSTED_CONNECT = "INSTANCE_SELF_HOSTED_CONNECT"
+    CONFIG_GET = "CONFIG_GET"
+    CONFIG_SET = "CONFIG_SET"
+    CONFIG_CLEAR = "CONFIG_CLEAR"
+    CONFIG_UPLOAD_SSH_KEY = "CONFIG_UPLOAD_SSH_KEY"
+
+
+class CliCommandAvailability(str, Enum):
+    ALLOWED = "ALLOWED"
+    RESTRICTED = "RESTRICTED"
+    DEPRECATED = "DEPRECATED"
+
+
+ALWAYS_AVAILABLE_COMMANDS = [
+    CliCommand.VERSION,
+    CliCommand.PROJECT_CONFIG_GET,
+    CliCommand.CONFIG_GET,
+    CliCommand.CONFIG_SET,
+    CliCommand.CONFIG_CLEAR,
+]

thestage/cli_command_helper.py

@@ -1,51 +1,51 @@
-from typing import Callable
-
-import typer
-
-from thestage.cli_command import CliCommand, CliCommandAvailability
-from thestage.color_scheme.color_scheme import ColorScheme
-from thestage.config import config_storage
-from rich import print
-
-
-def cli_command(command_id: CliCommand):
-    def decorator(func: Callable):
-        setattr(func, "__cli_command__", command_id)
-        return func
-    return decorator
-
-
-def get_command_metadata(command_id: CliCommand) -> dict:
-    return {
-        "rich_help_panel": get_command_help_panel(command_id),
-        "deprecated": is_command_deprecated(command_id),
-    }
-
-
-def get_command_group_help_panel() -> str:
-    return "Command Groups"
-
-
-def get_command_help_panel(command: CliCommand) -> str:
-    if config_storage.APP_CONFIG.runtime.allowed_commands.get(command) == CliCommandAvailability.ALLOWED:
-        return "Allowed Commands"
-    if config_storage.APP_CONFIG.runtime.allowed_commands.get(command) == CliCommandAvailability.RESTRICTED:
-        return "Restricted Commands"
-    if config_storage.APP_CONFIG.runtime.allowed_commands.get(command) == CliCommandAvailability.DEPRECATED:
-        return "Deprecated Commands"
-
-
-def is_command_deprecated(command: CliCommand) -> bool:
-    if config_storage.APP_CONFIG.runtime.allowed_commands.get(command) == CliCommandAvailability.DEPRECATED:
-        return True
-    return False
-
-
-def check_command_permission(executed_command: CliCommand):
-    if config_storage.APP_CONFIG.runtime.is_token_valid == False and executed_command != CliCommand.CONFIG_SET:
-        print(f"[{ColorScheme.WARNING.value}]Your API Token is not valid. You can update the token using 'thestage config set' command[{ColorScheme.WARNING.value}]")
-
-    is_allowed = config_storage.APP_CONFIG.runtime.allowed_commands.get(executed_command) == CliCommandAvailability.ALLOWED
-    if not is_allowed:
-        typer.echo("Action is not allowed")
-        raise typer.Exit(code=1)
+from typing import Callable
+
+import typer
+
+from thestage.cli_command import CliCommand, CliCommandAvailability
+from thestage.color_scheme.color_scheme import ColorScheme
+from thestage.config import config_storage
+from rich import print
+
+
+def cli_command(command_id: CliCommand):
+    def decorator(func: Callable):
+        setattr(func, "__cli_command__", command_id)
+        return func
+    return decorator
+
+
+def get_command_metadata(command_id: CliCommand) -> dict:
+    return {
+        "rich_help_panel": get_command_help_panel(command_id),
+        "deprecated": is_command_deprecated(command_id),
+    }
+
+
+def get_command_group_help_panel() -> str:
+    return "Command Groups"
+
+
+def get_command_help_panel(command: CliCommand) -> str:
+    if config_storage.APP_CONFIG.runtime.allowed_commands.get(command) == CliCommandAvailability.ALLOWED:
+        return "Allowed Commands"
+    if config_storage.APP_CONFIG.runtime.allowed_commands.get(command) == CliCommandAvailability.RESTRICTED:
+        return "Restricted Commands"
+    if config_storage.APP_CONFIG.runtime.allowed_commands.get(command) == CliCommandAvailability.DEPRECATED:
+        return "Deprecated Commands"
+
+
+def is_command_deprecated(command: CliCommand) -> bool:
+    if config_storage.APP_CONFIG.runtime.allowed_commands.get(command) == CliCommandAvailability.DEPRECATED:
+        return True
+    return False
+
+
+def check_command_permission(executed_command: CliCommand):
+    if config_storage.APP_CONFIG.runtime.is_token_valid == False and executed_command != CliCommand.CONFIG_SET:
+        print(f"[{ColorScheme.WARNING.value}]Your API Token is not valid. You can update the token using 'thestage config set' command[{ColorScheme.WARNING.value}]")
+
+    is_allowed = config_storage.APP_CONFIG.runtime.allowed_commands.get(executed_command) == CliCommandAvailability.ALLOWED
+    if not is_allowed:
+        typer.echo("Action is not allowed")
+        raise typer.Exit(code=1)

thestage/color_scheme/color_scheme.py

@@ -1,7 +1,7 @@
-from enum import Enum
-
-# https://rich.readthedocs.io/en/stable/appendix/colors.html
-class ColorScheme(str, Enum):
-    GIT_HEADLESS = "orange_red1"
-    WARNING = "orange_red1"
-    USEFUL_INFO = "deep_sky_blue1"
+from enum import Enum
+
+# https://rich.readthedocs.io/en/stable/appendix/colors.html
+class ColorScheme(str, Enum):
+    GIT_HEADLESS = "orange_red1"
+    WARNING = "orange_red1"
+    USEFUL_INFO = "deep_sky_blue1"

thestage/config/__init__.py

@@ -1,18 +1,18 @@
-# Init project settings by ENV environment variable
-import locale
-from pathlib import Path
-
-
-from python_gettext_translations.translations import init_translations
-
-from thestage.config.env_base import *
-
-
-THESTAGE_LOCAL_LANGUAGE = 'en_GB'
-if locale.getlocale():
-    THESTAGE_LOCAL_LANGUAGE = locale.getlocale()[0]
-
-translation = Path(f'i18n/')
-
-if translation.exists() and translation.is_dir():
-    init_translations(f'i18n/')
+# Init project settings by ENV environment variable
+import locale
+from pathlib import Path
+
+
+from python_gettext_translations.translations import init_translations
+
+from thestage.config.env_base import *
+
+
+THESTAGE_LOCAL_LANGUAGE = 'en_GB'
+if locale.getlocale():
+    THESTAGE_LOCAL_LANGUAGE = locale.getlocale()[0]
+
+translation = Path(f'i18n/')
+
+if translation.exists() and translation.is_dir():
+    init_translations(f'i18n/')

thestage/config/config_storage.py

@@ -1,5 +1,5 @@
-from typing import Optional
-
-from thestage.services.core_files.config_entity import ConfigEntity
-
-APP_CONFIG: Optional[ConfigEntity] = None
+from typing import Optional
+
+from thestage.services.core_files.config_entity import ConfigEntity
+
+APP_CONFIG: Optional[ConfigEntity] = None

thestage/config/env_base.py

@@ -1,7 +1,7 @@
-import os
-
-THESTAGE_CONFIG_DIR = os.getenv('THESTAGE_CONFIG_DIR', '.thestage')
-THESTAGE_CONFIG_FILE = os.getenv('THESTAGE_CONFIG_FILE', 'config.json')
-THESTAGE_AUTH_TOKEN = os.getenv('THESTAGE_AUTH_TOKEN', None)
-THESTAGE_LOGGING_FILE = os.getenv('THESTAGE_LOGGING_FILE', 'thestage.log')
-THESTAGE_API_URL = os.getenv('THESTAGE_API_URL', 'https://backend.thestage.ai')
+import os
+
+THESTAGE_CONFIG_DIR = os.getenv('THESTAGE_CONFIG_DIR', '.thestage')
+THESTAGE_CONFIG_FILE = os.getenv('THESTAGE_CONFIG_FILE', 'config.json')
+THESTAGE_AUTH_TOKEN = os.getenv('THESTAGE_AUTH_TOKEN', None)
+THESTAGE_LOGGING_FILE = os.getenv('THESTAGE_LOGGING_FILE', 'thestage.log')
+THESTAGE_API_URL = os.getenv('THESTAGE_API_URL', 'https://backend.thestage.ai')

thestage/controllers/base_controller.py

@@ -1,67 +1,67 @@
-from pathlib import Path
-from typing import Optional
-
-from thestage.cli_command import CliCommand
-from thestage.cli_command_helper import get_command_metadata, check_command_permission
-from thestage.i18n.translation import __
-from thestage.helpers.logger.app_logger import app_logger
-from thestage.controllers.utils_controller import get_current_directory, validate_config_and_get_service_factory
-from thestage import __app_name__, __version__
-
-import typer
-
-from thestage.services.connect.connect_service import ConnectService
-
-app = typer.Typer(no_args_is_help=True,)
-
-
-@app.command(name='version', help="Get application's name and version", no_args_is_help=False, **get_command_metadata(CliCommand.VERSION))
-def version():
-    command_name = CliCommand.VERSION
-    app_logger.info(f'Running {command_name} from {get_current_directory()}')
-    check_command_permission(command_name)
-
-    typer.echo(f"{__app_name__} v{__version__}")
-    raise typer.Exit(0)
-
-
-@app.command(name="connect", no_args_is_help=True, help=__("Connect to server instance or container or task"), **get_command_metadata(CliCommand.CONNECT))
-def connect(
-        uid: Optional[str] = typer.Argument(
-            help=__("Unique ID of server instance or container or task ID"), ),
-        username: Optional[str] = typer.Option(
-            None,
-            '--username',
-            '-u',
-            help=__("Username for the server instance (required when connecting to self-hosted instance)"),
-            is_eager=False,
-        ),
-        private_ssh_key_path: str = typer.Option(
-            None,
-            "--private-key-path",
-            "-pk",
-            help=__("Path to private key that will be accepted by remote server (optional)"),
-            is_eager=False,
-        ),
-):
-    command_name = CliCommand.CONNECT
-    app_logger.info(f'Running {command_name} from {get_current_directory()}')
-    check_command_permission(command_name)
-
-    if private_ssh_key_path and not Path(private_ssh_key_path).is_file():
-        typer.echo(f'No file found at provided path {private_ssh_key_path}')
-        raise typer.Exit(1)
-
-    service_factory = validate_config_and_get_service_factory()
-
-    connect_service: ConnectService = service_factory.get_connect_service()
-
-    connect_service.connect_to_entity(
-        uid=uid,
-        username=username,
-        private_key_path=private_ssh_key_path
-    )
-
-
-    app_logger.info(f'Stop connect to entity')
-    raise typer.Exit(0)
+from pathlib import Path
+from typing import Optional
+
+from thestage.cli_command import CliCommand
+from thestage.cli_command_helper import get_command_metadata, check_command_permission
+from thestage.i18n.translation import __
+from thestage.helpers.logger.app_logger import app_logger
+from thestage.controllers.utils_controller import get_current_directory, validate_config_and_get_service_factory
+from thestage import __app_name__, __version__
+
+import typer
+
+from thestage.services.connect.connect_service import ConnectService
+
+app = typer.Typer(no_args_is_help=True,)
+
+
+@app.command(name='version', help="Get application's name and version", no_args_is_help=False, **get_command_metadata(CliCommand.VERSION))
+def version():
+    command_name = CliCommand.VERSION
+    app_logger.info(f'Running {command_name} from {get_current_directory()}')
+    check_command_permission(command_name)
+
+    typer.echo(f"{__app_name__} v{__version__}")
+    raise typer.Exit(0)
+
+
+@app.command(name="connect", no_args_is_help=True, help=__("Connect to server instance or container or task"), **get_command_metadata(CliCommand.CONNECT))
+def connect(
+        uid: Optional[str] = typer.Argument(
+            help=__("Unique ID of server instance or container or task ID"), ),
+        username: Optional[str] = typer.Option(
+            None,
+            '--username',
+            '-u',
+            help=__("Username for the server instance (required when connecting to self-hosted instance)"),
+            is_eager=False,
+        ),
+        private_ssh_key_path: str = typer.Option(
+            None,
+            "--private-key-path",
+            "-pk",
+            help=__("Path to private key that will be accepted by remote server (optional)"),
+            is_eager=False,
+        ),
+):
+    command_name = CliCommand.CONNECT
+    app_logger.info(f'Running {command_name} from {get_current_directory()}')
+    check_command_permission(command_name)
+
+    if private_ssh_key_path and not Path(private_ssh_key_path).is_file():
+        typer.echo(f'No file found at provided path {private_ssh_key_path}')
+        raise typer.Exit(1)
+
+    service_factory = validate_config_and_get_service_factory()
+
+    connect_service: ConnectService = service_factory.get_connect_service()
+
+    connect_service.connect_to_entity(
+        uid=uid,
+        username=username,
+        private_key_path=private_ssh_key_path
+    )
+
+
+    app_logger.info(f'Stop connect to entity')
+    raise typer.Exit(0)