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

import shutil
from dataclasses import dataclass
from math import floor
from pathlib import Path


def get_term_filler(name: str) -> int:
    width, height = shutil.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}"


@dataclass
class CmdOut:
    stdout: str
    stderr: str
    cwd: Path
    command: str
    returncode: int
    msg: str | None

    def __str__(self) -> str:
        error_str = 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}
        """
        return error_str


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

    description: str | None
    location: str

    def __init__(
        self,
        msg: str | None = None,
        *,
        description: str | None = None,
        location: str | None = None,
    ) -> None:
        self.description = description
        self.location = location or "Unknown location"
        self.msg = msg or ""
        if self.description:
            exception_msg = f"{self.location}: \n{self.msg} - {self.description}"
        else:
            exception_msg = f"{self.location}: \n{self.msg}"
        super().__init__(exception_msg)


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})"