clan-core/pkgs/clan-cli/clan_cli/errors.py

import os
from math import floor
from pathlib import Path
from typing import NamedTuple


def get_term_filler(name: str) -> int:
    width, height = os.get_terminal_size()

    filler = floor((width - len(name)) / 2)
    return filler - 1


def text_heading(heading: str) -> str:
    filler = get_term_filler(heading)
    return f"{'=' * filler} {heading} {'=' * filler}"


class CmdOut(NamedTuple):
    stdout: str
    stderr: str
    cwd: Path
    command: str
    returncode: int
    msg: str | None = None

    def __str__(self) -> str:
        return f"""
{text_heading(heading="Command")}
{self.command}
{text_heading(heading="Stderr")}
{self.stderr}
{text_heading(heading="Stdout")}
{self.stdout}
{text_heading(heading="Metadata")}
Message: {self.msg}
Working Directory: '{self.cwd}'
Return Code: {self.returncode}
"""


class ClanError(Exception):
    """Base class for exceptions in this module."""

    pass


class ClanHttpError(ClanError):
    status_code: int
    msg: str

    def __init__(self, status_code: int, msg: str) -> None:
        self.status_code = status_code
        self.msg = msg
        super().__init__(msg)


class ClanCmdError(ClanError):
    cmd: CmdOut

    def __init__(self, cmd: CmdOut) -> None:
        self.cmd = cmd
        super().__init__()

    def __str__(self) -> str:
        return str(self.cmd)

    def __repr__(self) -> str:
        return f"ClanCmdError({self.cmd})"