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

thestage/controllers/config_controller.py

@@ -1,137 +1,137 @@
-from pathlib import Path
-
-import click
-
-from thestage.cli_command import CliCommand
-from thestage.cli_command_helper import get_command_metadata, check_command_permission
-from thestage.services.core_files.config_entity import ConfigEntity
-
-from thestage.entities.enums.yes_no_response import YesOrNoResponse
-from thestage.i18n.translation import __
-from thestage.helpers.logger.app_logger import app_logger, get_log_path_from_os
-from thestage.services.connect.connect_service import ConnectService
-from thestage.services.service_factory import ServiceFactory
-from thestage.controllers.utils_controller import get_current_directory
-
-import typer
-
-app = typer.Typer(no_args_is_help=True, help=__("Manage configuration settings"))
-
-
-@app.command(name='get', no_args_is_help=False, help=__("Display all configuration settings"), **get_command_metadata(CliCommand.CONFIG_GET))
-def config_get():
-    command_name = CliCommand.CONFIG_GET
-    app_logger.info(f'Running {command_name} from {get_current_directory()}')
-    check_command_permission(command_name)
-
-    config_provider = ServiceFactory().get_config_provider()
-    config: ConfigEntity = config_provider.get_config()
-
-    if not config:
-        typer.echo(__('No configuration found'))
-        raise typer.Exit(1)
-
-    config_provider.save_config()
-
-    typer.echo(__('THESTAGE TOKEN: %token%', {'token': config.main.thestage_auth_token or ''}))
-    typer.echo(__('THESTAGE API LINK: %link%', {'link': config.main.thestage_api_url or ''}))
-
-    if config.runtime.config_global_path:
-        typer.echo(__('CONFIG PATH: %path%', {'path': str(config.runtime.config_global_path or '') + f'/config.json'}))
-
-    typer.echo(__('APPLICATION LOGS PATH: %path%', {'path': str(get_log_path_from_os())}))
-
-    raise typer.Exit(0)
-
-
-@app.command(name='set', no_args_is_help=True, help=__("Update configuration settings"), **get_command_metadata(CliCommand.CONFIG_SET))
-def config_set(
-    token: str = typer.Option(
-            None,
-            "--api-token",
-            "-t",
-            help=__("Set or update API token"),
-            is_eager=False,
-        ),
-):
-    command_name = CliCommand.CONFIG_SET
-    app_logger.info(f'Running {command_name} from {get_current_directory()}')
-    check_command_permission(command_name)
-
-    service_factory = ServiceFactory()
-    app_service = service_factory.get_app_config_service()
-
-    if token:
-        app_service.app_change_token(token=token)
-
-    typer.echo('Configuration updated successfully')
-    raise typer.Exit(0)
-
-
-@app.command(name='clear', no_args_is_help=False, help=__("Clear configuration"), **get_command_metadata(CliCommand.CONFIG_CLEAR))
-def config_clear():
-    command_name = CliCommand.CONFIG_CLEAR
-    app_logger.info(f'Running {command_name} from {get_current_directory()}')
-    check_command_permission(command_name)
-
-    config_provider = ServiceFactory().get_config_provider()
-    config_dir = config_provider.get_config().runtime.config_global_path
-    config_provider.clear_config()
-    typer.echo(f'Removed {config_dir}')
-
-    raise typer.Exit(0)
-
-
-@app.command(name='upload-ssh-key', no_args_is_help=True, help=__("Send your public SSH key to the platform and / or rented server instance"), **get_command_metadata(CliCommand.CONFIG_UPLOAD_SSH_KEY))
-def upload_ssh_key(
-        ssh_public_key: str = typer.Argument(
-            help=__("Path to your public SSH key file or your public SSH key contents"),
-        ),
-        instance_rented_slug: str = typer.Option(
-            None,
-            "--instance-uid",
-            "-uid",
-            help=__("Unique ID of your rented instance to add the key to (optional)"),
-            is_eager=False,
-        )
-):
-    command_name = CliCommand.CONFIG_UPLOAD_SSH_KEY
-    app_logger.info(f'Running {command_name} from {get_current_directory()}')
-    check_command_permission(command_name)
-
-    service_factory = ServiceFactory()
-    connect_service: ConnectService = service_factory.get_connect_service()
-
-    is_path_provided_confirmed = False
-    ssh_key_contents = ssh_public_key
-
-    if ssh_public_key.startswith("/") or ssh_public_key.startswith("~") or ssh_public_key.endswith(".pub") or len(ssh_public_key) < 30:
-        is_path_provided_confirmed = True
-
-    ssh_key_path = Path(ssh_public_key).absolute()
-    if is_path_provided_confirmed and not ssh_key_path.exists():
-        typer.echo(f"No key was found at {ssh_key_path}")
-        raise typer.Exit(1)
-
-    if is_path_provided_confirmed or ssh_key_path.exists():
-        if '.' not in ssh_key_path.name:
-            proceed_with_no_extension: YesOrNoResponse = typer.prompt(
-                text=f"File '{ssh_key_path.name}' probably contains a private key. Proceed?",
-                show_choices=True,
-                default=YesOrNoResponse.YES.value,
-                type=click.Choice([r.value for r in YesOrNoResponse]),
-                show_default=True,
-            )
-            if proceed_with_no_extension == YesOrNoResponse.NO:
-                raise typer.Exit(0)
-        ssh_key_contents = ssh_key_path.open("r").read()
-        if 'private key-----' in ssh_key_contents.lower():
-            typer.echo(f"{ssh_key_path} is identified as a private key. Please provide a public SSH key.")
-            raise typer.Exit(1)
-
-    connect_service.upload_ssh_key(
-        public_key_contents=ssh_key_contents,
-        instance_slug=instance_rented_slug,
-    )
-
-    raise typer.Exit(0)
+from pathlib import Path
+
+import click
+
+from thestage.cli_command import CliCommand
+from thestage.cli_command_helper import get_command_metadata, check_command_permission
+from thestage.services.core_files.config_entity import ConfigEntity
+
+from thestage.entities.enums.yes_no_response import YesOrNoResponse
+from thestage.i18n.translation import __
+from thestage.helpers.logger.app_logger import app_logger, get_log_path_from_os
+from thestage.services.connect.connect_service import ConnectService
+from thestage.services.service_factory import ServiceFactory
+from thestage.controllers.utils_controller import get_current_directory
+
+import typer
+
+app = typer.Typer(no_args_is_help=True, help=__("Manage configuration settings"))
+
+
+@app.command(name='get', no_args_is_help=False, help=__("Display all configuration settings"), **get_command_metadata(CliCommand.CONFIG_GET))
+def config_get():
+    command_name = CliCommand.CONFIG_GET
+    app_logger.info(f'Running {command_name} from {get_current_directory()}')
+    check_command_permission(command_name)
+
+    config_provider = ServiceFactory().get_config_provider()
+    config: ConfigEntity = config_provider.get_config()
+
+    if not config:
+        typer.echo(__('No configuration found'))
+        raise typer.Exit(1)
+
+    config_provider.save_config()
+
+    typer.echo(__('THESTAGE TOKEN: %token%', {'token': config.main.thestage_auth_token or ''}))
+    typer.echo(__('THESTAGE API LINK: %link%', {'link': config.main.thestage_api_url or ''}))
+
+    if config.runtime.config_global_path:
+        typer.echo(__('CONFIG PATH: %path%', {'path': str(config.runtime.config_global_path or '') + f'/config.json'}))
+
+    typer.echo(__('APPLICATION LOGS PATH: %path%', {'path': str(get_log_path_from_os())}))
+
+    raise typer.Exit(0)
+
+
+@app.command(name='set', no_args_is_help=True, help=__("Update configuration settings"), **get_command_metadata(CliCommand.CONFIG_SET))
+def config_set(
+    token: str = typer.Option(
+            None,
+            "--api-token",
+            "-t",
+            help=__("Set or update API token"),
+            is_eager=False,
+        ),
+):
+    command_name = CliCommand.CONFIG_SET
+    app_logger.info(f'Running {command_name} from {get_current_directory()}')
+    check_command_permission(command_name)
+
+    service_factory = ServiceFactory()
+    app_service = service_factory.get_app_config_service()
+
+    if token:
+        app_service.app_change_token(token=token)
+
+    typer.echo('Configuration updated successfully')
+    raise typer.Exit(0)
+
+
+@app.command(name='clear', no_args_is_help=False, help=__("Clear configuration"), **get_command_metadata(CliCommand.CONFIG_CLEAR))
+def config_clear():
+    command_name = CliCommand.CONFIG_CLEAR
+    app_logger.info(f'Running {command_name} from {get_current_directory()}')
+    check_command_permission(command_name)
+
+    config_provider = ServiceFactory().get_config_provider()
+    config_dir = config_provider.get_config().runtime.config_global_path
+    config_provider.clear_config()
+    typer.echo(f'Removed {config_dir}')
+
+    raise typer.Exit(0)
+
+
+@app.command(name='upload-ssh-key', no_args_is_help=True, help=__("Send your public SSH key to the platform and / or rented server instance"), **get_command_metadata(CliCommand.CONFIG_UPLOAD_SSH_KEY))
+def upload_ssh_key(
+        ssh_public_key: str = typer.Argument(
+            help=__("Path to your public SSH key file or your public SSH key contents"),
+        ),
+        instance_rented_slug: str = typer.Option(
+            None,
+            "--instance-uid",
+            "-uid",
+            help=__("Unique ID of your rented instance to add the key to (optional)"),
+            is_eager=False,
+        )
+):
+    command_name = CliCommand.CONFIG_UPLOAD_SSH_KEY
+    app_logger.info(f'Running {command_name} from {get_current_directory()}')
+    check_command_permission(command_name)
+
+    service_factory = ServiceFactory()
+    connect_service: ConnectService = service_factory.get_connect_service()
+
+    is_path_provided_confirmed = False
+    ssh_key_contents = ssh_public_key
+
+    if ssh_public_key.startswith("/") or ssh_public_key.startswith("~") or ssh_public_key.endswith(".pub") or len(ssh_public_key) < 30:
+        is_path_provided_confirmed = True
+
+    ssh_key_path = Path(ssh_public_key).absolute()
+    if is_path_provided_confirmed and not ssh_key_path.exists():
+        typer.echo(f"No key was found at {ssh_key_path}")
+        raise typer.Exit(1)
+
+    if is_path_provided_confirmed or ssh_key_path.exists():
+        if '.' not in ssh_key_path.name:
+            proceed_with_no_extension: YesOrNoResponse = typer.prompt(
+                text=f"File '{ssh_key_path.name}' probably contains a private key. Proceed?",
+                show_choices=True,
+                default=YesOrNoResponse.YES.value,
+                type=click.Choice([r.value for r in YesOrNoResponse]),
+                show_default=True,
+            )
+            if proceed_with_no_extension == YesOrNoResponse.NO:
+                raise typer.Exit(0)
+        ssh_key_contents = ssh_key_path.open("r").read()
+        if 'private key-----' in ssh_key_contents.lower():
+            typer.echo(f"{ssh_key_path} is identified as a private key. Please provide a public SSH key.")
+            raise typer.Exit(1)
+
+    connect_service.upload_ssh_key(
+        public_key_contents=ssh_key_contents,
+        instance_slug=instance_rented_slug,
+    )
+
+    raise typer.Exit(0)