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

thestage/git/ProgressPrinter.py

@@ -1,23 +1,23 @@
-from git import RemoteProgress
-
-
-class ProgressPrinter(RemoteProgress):
-    def __init__( self ):
-        super().__init__()
-
-        self.__all_dropped_lines = []
-
-    def update( self, op_code, cur_count, max_count=None, message='' ):
-        pass
-
-    def line_dropped( self, line ):
-        if line.startswith( 'POST git-upload-pack' ):
-            return
-
-        self.__all_dropped_lines.append( line )
-
-    def allErrorLines( self ):
-        return self.error_lines + self.__all_dropped_lines
-
-    def allDroppedLines( self ):
+from git import RemoteProgress
+
+
+class ProgressPrinter(RemoteProgress):
+    def __init__( self ):
+        super().__init__()
+
+        self.__all_dropped_lines = []
+
+    def update( self, op_code, cur_count, max_count=None, message='' ):
+        pass
+
+    def line_dropped( self, line ):
+        if line.startswith( 'POST git-upload-pack' ):
+            return
+
+        self.__all_dropped_lines.append( line )
+
+    def allErrorLines( self ):
+        return self.error_lines + self.__all_dropped_lines
+
+    def allDroppedLines( self ):
         return self.__all_dropped_lines

thestage/helpers/error_handler.py

@@ -1,115 +1,115 @@
-import traceback
-from typing import Any, Callable
-
-import requests
-import typer
-from click.exceptions import Exit, Abort
-from git import GitCommandError
-from paramiko.ssh_exception import PasswordRequiredException
-
-from thestage.config import THESTAGE_API_URL
-from thestage.exceptions.file_system_exception import FileSystemException
-from thestage.exceptions.remote_server_exception import RemoteServerException
-from thestage.i18n.translation import __
-from thestage.exceptions.git_access_exception import GitAccessException
-from thestage.exceptions.auth_exception import AuthException
-from thestage.exceptions.business_logic_exception import BusinessLogicException
-from thestage.exceptions.config_exception import ConfigException
-from thestage.helpers.logger.app_logger import app_logger
-from thestage.services.clients.thestage_api.core.http_client_exception import HttpClientException
-
-
-def error_handler() -> Callable:
-    def wrap(f):
-        def wrapper(*args: Any, **kwargs: Any) -> Any:
-            try:
-                result = f(*args, **kwargs)
-                return result
-            except AuthException as e1:
-                typer.echo(__('Authentication failed: update API token'))
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except BusinessLogicException as e2:
-                typer.echo(__('Business logic error encountered: contact TheStage AI team'))
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except ConfigException as e3:
-                typer.echo(__(
-                    'Configuration error encountered: %error_message%',
-                    {
-                        'error_message': e3.get_message()
-                    }
-                ))
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except FileSystemException as e4:
-                typer.echo(__(
-                    "File system error encountered: %error_message%",
-                    {
-                        'error_message': e4.get_message()
-                    }
-                ))
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except HttpClientException as e5:
-                typer.echo(__(
-                    f"TheStage server error: %error_message%",
-
-                    {
-                        'stage_url': THESTAGE_API_URL,
-                        'error_message': e5.get_message()
-                    }
-                ))
-                app_logger.error(f"Connection error to {THESTAGE_API_URL} - {e5}")
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except GitAccessException as e6:
-                typer.echo(e6.get_message())
-                typer.echo(e6.get_dop_message())
-                typer.echo(__(
-                    "Visit %git_url% to accept the invitation or check your access to the repository",
-                    {
-                        'git_url': e6.get_url(),
-                    }
-                ))
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except GitCommandError as e7:
-                typer.echo(f'Git command error encountered: {e7.stderr} (status: {e7.status})')
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except RemoteServerException as e8:
-                typer.echo(__(
-                    'Error connecting to server or Docker container at %ip_address% as %username%',
-                    {
-                        'ip_address': e8.ip_address,
-                        'username': e8.username,
-                    }))
-                app_logger.error(f'Error connecting to server or Docker container at {e8.ip_address} as {e8.username} - {e8}')
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except Abort as e9:
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except requests.exceptions.ConnectionError as e10:
-                # TODO we don't know for sure if it is Thestage connection error - throw appropriate exception (on token validation?)
-                typer.echo("Connection error")
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except PasswordRequiredException as e11:
-                # technically we can use encrypted keys but dealing with passwords is big bs
-                typer.echo("Provided key requires password. Please use non-encrypted key.")
-                app_logger.error(f'{traceback.format_exc()}')
-                raise typer.Exit(1)
-            except Exception as e100:
-                if isinstance(e100, Exit):
-                    raise e100
-                else:
-                    typer.echo(__('Undefined error occurred'))
-                    # typer.echo(e100.__class__.__name__)
-                    # print(traceback.format_exc())
-                    # TODO send all exceptions to backend?
-                    app_logger.error(f'{traceback.format_exc()}')
-                    raise typer.Exit(1)
-        return wrapper
-    return wrap
+import traceback
+from typing import Any, Callable
+
+import requests
+import typer
+from click.exceptions import Exit, Abort
+from git import GitCommandError
+from paramiko.ssh_exception import PasswordRequiredException
+
+from thestage.config import THESTAGE_API_URL
+from thestage.exceptions.file_system_exception import FileSystemException
+from thestage.exceptions.remote_server_exception import RemoteServerException
+from thestage.i18n.translation import __
+from thestage.exceptions.git_access_exception import GitAccessException
+from thestage.exceptions.auth_exception import AuthException
+from thestage.exceptions.business_logic_exception import BusinessLogicException
+from thestage.exceptions.config_exception import ConfigException
+from thestage.helpers.logger.app_logger import app_logger
+from thestage.services.clients.thestage_api.core.http_client_exception import HttpClientException
+
+
+def error_handler() -> Callable:
+    def wrap(f):
+        def wrapper(*args: Any, **kwargs: Any) -> Any:
+            try:
+                result = f(*args, **kwargs)
+                return result
+            except AuthException as e1:
+                typer.echo(__('Authentication failed: update API token'))
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except BusinessLogicException as e2:
+                typer.echo(__('Business logic error encountered: contact TheStage AI team'))
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except ConfigException as e3:
+                typer.echo(__(
+                    'Configuration error encountered: %error_message%',
+                    {
+                        'error_message': e3.get_message()
+                    }
+                ))
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except FileSystemException as e4:
+                typer.echo(__(
+                    "File system error encountered: %error_message%",
+                    {
+                        'error_message': e4.get_message()
+                    }
+                ))
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except HttpClientException as e5:
+                typer.echo(__(
+                    f"TheStage server error: %error_message%",
+
+                    {
+                        'stage_url': THESTAGE_API_URL,
+                        'error_message': e5.get_message()
+                    }
+                ))
+                app_logger.error(f"Connection error to {THESTAGE_API_URL} - {e5}")
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except GitAccessException as e6:
+                typer.echo(e6.get_message())
+                typer.echo(e6.get_dop_message())
+                typer.echo(__(
+                    "Visit %git_url% to accept the invitation or check your access to the repository",
+                    {
+                        'git_url': e6.get_url(),
+                    }
+                ))
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except GitCommandError as e7:
+                typer.echo(f'Git command error encountered: {e7.stderr} (status: {e7.status})')
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except RemoteServerException as e8:
+                typer.echo(__(
+                    'Error connecting to server or Docker container at %ip_address% as %username%',
+                    {
+                        'ip_address': e8.ip_address,
+                        'username': e8.username,
+                    }))
+                app_logger.error(f'Error connecting to server or Docker container at {e8.ip_address} as {e8.username} - {e8}')
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except Abort as e9:
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except requests.exceptions.ConnectionError as e10:
+                # TODO we don't know for sure if it is Thestage connection error - throw appropriate exception (on token validation?)
+                typer.echo("Connection error")
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except PasswordRequiredException as e11:
+                # technically we can use encrypted keys but dealing with passwords is big bs
+                typer.echo("Provided key requires password. Please use non-encrypted key.")
+                app_logger.error(f'{traceback.format_exc()}')
+                raise typer.Exit(1)
+            except Exception as e100:
+                if isinstance(e100, Exit):
+                    raise e100
+                else:
+                    typer.echo(__('Undefined error occurred'))
+                    # typer.echo(e100.__class__.__name__)
+                    # print(traceback.format_exc())
+                    # TODO send all exceptions to backend?
+                    app_logger.error(f'{traceback.format_exc()}')
+                    raise typer.Exit(1)
+        return wrapper
+    return wrap

thestage/helpers/exception_hook.py

@@ -1,14 +1,14 @@
-import sys
-import traceback
-
-from thestage.helpers.logger.app_logger import app_logger
-
-
-def exception_handler(exception_type, exception, tb):
-    print(exception)
-    tb_list = traceback.format_exception(exception_type, exception, tb)
-    tb_text = ''.join(tb_list)
-    app_logger.error(tb_text)
-
-
-sys.excepthook = exception_handler
+import sys
+import traceback
+
+from thestage.helpers.logger.app_logger import app_logger
+
+
+def exception_handler(exception_type, exception, tb):
+    print(exception)
+    tb_list = traceback.format_exception(exception_type, exception, tb)
+    tb_text = ''.join(tb_list)
+    app_logger.error(tb_text)
+
+
+sys.excepthook = exception_handler

thestage/helpers/logger/app_logger.py

@@ -1,50 +1,50 @@
-import logging
-import platform
-from logging.handlers import RotatingFileHandler
-from pathlib import Path
-from thestage import __app_name__, __version__
-
-from thestage.config import THESTAGE_CONFIG_DIR
-
-from thestage.config import THESTAGE_LOGGING_FILE
-from thestage.exceptions.file_system_exception import FileSystemException
-
-
-def get_log_path_from_os() -> Path:
-    system = platform.system()
-    if system == 'Linux':
-        path = Path.home().joinpath(THESTAGE_CONFIG_DIR).joinpath('logs')
-    elif system == 'Windows':
-        path = Path.home().joinpath(THESTAGE_CONFIG_DIR).joinpath('logs')
-    elif system == 'Darwin':
-        path = Path.home().joinpath('library').joinpath('logs').joinpath('thestage')
-    else:
-        path = Path.home().joinpath(THESTAGE_CONFIG_DIR).joinpath('logs')
-
-    if not path.exists():
-        try:
-            path.mkdir(exist_ok=True, parents=True)
-        except OSError:
-            raise FileSystemException("Error create log dir")
-
-    return path
-
-
-def build_logger(level) -> logging.Logger:
-    logger = logging.getLogger(f"{__app_name__} v{__version__}")
-    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-    logger.setLevel(level)
-
-    log_path = get_log_path_from_os()
-    tsr_log_file = log_path.joinpath(THESTAGE_LOGGING_FILE)
-
-    if tsr_log_file:
-        file_h = RotatingFileHandler(filename=tsr_log_file, maxBytes=1024 * 10, backupCount=5)
-        file_h.setLevel(level)
-        file_h.setFormatter(formatter)
-        logger.addHandler(file_h)
-
-    return logger
-
-
-app_logger = build_logger(level=logging.DEBUG)
+import logging
+import platform
+from logging.handlers import RotatingFileHandler
+from pathlib import Path
+from thestage import __app_name__, __version__
+
+from thestage.config import THESTAGE_CONFIG_DIR
+
+from thestage.config import THESTAGE_LOGGING_FILE
+from thestage.exceptions.file_system_exception import FileSystemException
+
+
+def get_log_path_from_os() -> Path:
+    system = platform.system()
+    if system == 'Linux':
+        path = Path.home().joinpath(THESTAGE_CONFIG_DIR).joinpath('logs')
+    elif system == 'Windows':
+        path = Path.home().joinpath(THESTAGE_CONFIG_DIR).joinpath('logs')
+    elif system == 'Darwin':
+        path = Path.home().joinpath('library').joinpath('logs').joinpath('thestage')
+    else:
+        path = Path.home().joinpath(THESTAGE_CONFIG_DIR).joinpath('logs')
+
+    if not path.exists():
+        try:
+            path.mkdir(exist_ok=True, parents=True)
+        except OSError:
+            raise FileSystemException("Error create log dir")
+
+    return path
+
+
+def build_logger(level) -> logging.Logger:
+    logger = logging.getLogger(f"{__app_name__} v{__version__}")
+    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+    logger.setLevel(level)
+
+    log_path = get_log_path_from_os()
+    tsr_log_file = log_path.joinpath(THESTAGE_LOGGING_FILE)
+
+    if tsr_log_file:
+        file_h = RotatingFileHandler(filename=tsr_log_file, maxBytes=1024 * 10, backupCount=5)
+        file_h.setLevel(level)
+        file_h.setFormatter(formatter)
+        logger.addHandler(file_h)
+
+    return logger
+
+
+app_logger = build_logger(level=logging.DEBUG)

thestage/helpers/ssh_util.py

@@ -1,38 +1,38 @@
-from typing import Optional, List
-
-import typer
-from paramiko.dsskey import DSSKey
-from paramiko.ecdsakey import ECDSAKey
-from paramiko.ed25519key import Ed25519Key
-from paramiko.pkey import PKey
-from paramiko.rsakey import RSAKey
-from paramiko.ssh_exception import SSHException, PasswordRequiredException
-
-
-def parse_private_key(filename: str) -> Optional[PKey]:
-    error_output: List[str] = []
-    pkey: Optional[PKey] = None
-    pkey_class: Optional[str] = None
-
-    for pkey_class_item in (RSAKey, ECDSAKey, DSSKey, Ed25519Key):
-        try:
-            key = pkey_class_item.from_private_key_file(filename=filename)
-            pkey = key
-            pkey_class = pkey_class_item.__name__
-            break
-        except PasswordRequiredException as e1:
-            raise e1
-        except SSHException as e2:
-            error_output.append(f'[{pkey_class_item.__name__}]: {str(e2)}')
-        except Exception as e3:
-            error_output.append(f'[{pkey_class_item.__name__}]: Unexpected error: {e3.__class__.__name__}: {str(e3)}')
-
-    if pkey is None:
-        for line in error_output:
-            typer.echo(line)
-
-    if pkey_class and pkey_class == 'DSSKey':
-        typer.echo('DSA keys are not supported')
-        return None
-
-    return pkey
+from typing import Optional, List
+
+import typer
+from paramiko.dsskey import DSSKey
+from paramiko.ecdsakey import ECDSAKey
+from paramiko.ed25519key import Ed25519Key
+from paramiko.pkey import PKey
+from paramiko.rsakey import RSAKey
+from paramiko.ssh_exception import SSHException, PasswordRequiredException
+
+
+def parse_private_key(filename: str) -> Optional[PKey]:
+    error_output: List[str] = []
+    pkey: Optional[PKey] = None
+    pkey_class: Optional[str] = None
+
+    for pkey_class_item in (RSAKey, ECDSAKey, DSSKey, Ed25519Key):
+        try:
+            key = pkey_class_item.from_private_key_file(filename=filename)
+            pkey = key
+            pkey_class = pkey_class_item.__name__
+            break
+        except PasswordRequiredException as e1:
+            raise e1
+        except SSHException as e2:
+            error_output.append(f'[{pkey_class_item.__name__}]: {str(e2)}')
+        except Exception as e3:
+            error_output.append(f'[{pkey_class_item.__name__}]: Unexpected error: {e3.__class__.__name__}: {str(e3)}')
+
+    if pkey is None:
+        for line in error_output:
+            typer.echo(line)
+
+    if pkey_class and pkey_class == 'DSSKey':
+        typer.echo('DSA keys are not supported')
+        return None
+
+    return pkey