Chimie Théorique » scripts_chimie4psmn » DockOnSurf

def check_bak(file_name):

    """Checks if a file already exists and backs it up if so.

    @param file_name: file to be checked if exists

    """

    import os

    new_name = file_name

    bak_num = 0

    while os.path.isdir(new_name) or os.path.isfile(new_name):

        bak_num += 1

        new_name = new_name.split(".bak")[0] + f".bak{bak_num}"

    if bak_num > 0:

        os.rename(file_name, new_name)

        logger.warning(f"'{file_name}' already present. Backed it up to "

                       f"{new_name}")

def try_command(command, expct_error_types: list, *args, **kwargs):

    """Try to run a command and record exceptions (expected and not) on a log.

    @param command: method or function, the command to be executed.

    @param expct_error_types: tuple of tuples, every inner tuple is supposed to

    contain an exception type (eg. ValueError, TypeError, etc.) to be caught and

    a message to print in the log and on the screen explaining the exception.

    Error types that are not allow to be called with a custom message as only

    error argument are not supported.

    The outer tuple encloses all couples of error types and their relative

    messages.

    *args and **kwargs: arguments and keyword-arguments of the command to be

    executed.

    When trying to run 'command' with its args and kwargs, if an exception

    present on the 'error_types' occurs, its relative error message is recorded

    on the log and a same type exception is raised with the custom message.

    """

    err = False

    try:

        return_val = command(*args, **kwargs)

    except Exception as e:

        for expct_err in expct_error_types:

            if isinstance(e, expct_err[0]):

                logger.error(expct_err[1])

                err = expct_err[0](expct_err[1])

                break

        else:

            logger.exception(unexp_error)

            err = e

    else:

        err = False

        return return_val

    finally:

        if isinstance(err, BaseException):

            raise err