gifmaker/gifmaker/utils.py

# Standard
import re
import sys
import random
import string
import colorsys
from pathlib import Path
from datetime import datetime
from typing import Dict, Union, Tuple

# Libraries
import webcolors  # type: ignore


def random_string() -> str:
    vowels = "aeiou"
    consonants = "".join(set(string.ascii_lowercase) - set(vowels))

    def con() -> str:
        return random.choice(consonants)

    def vow() -> str:
        return random.choice(vowels)

    return con() + vow() + con() + vow() + con() + vow()


def get_extension(path: Path) -> str:
    return Path(path).suffix.lower().lstrip(".")


def exit(message: str) -> None:
    msg(f"\nExit: {message}\n")
    sys.exit(1)


def read_toml(path: Path) -> Union[Dict[str, str], None]:
    import tomllib

    if (not path.exists()) or (not path.is_file()):
        exit("TOML file does not exist")
        return None

    try:
        return tomllib.load(open(path, "rb"))
    except Exception as e:
        msg(f"Error: {e}")
        exit("Failed to read TOML file")
        return None


def random_color() -> Tuple[int, int, int]:
    from .config import config
    assert isinstance(config.Internal.random_colors, random.Random)

    def component():
        return config.Internal.random_colors.randint(0, 255)

    return component(), component(), component()


def random_light() -> Tuple[int, int, int]:
    color = random_color()
    return change_lightness(color, 255 - 20)


def random_dark() -> Tuple[int, int, int]:
    color = random_color()
    return change_lightness(color, 40)


def change_lightness(color: Tuple[int, int, int], lightness: int) -> Tuple[int, int, int]:
    hsv = list(colorsys.rgb_to_hsv(*color))
    hsv[2] = lightness
    rgb = colorsys.hsv_to_rgb(*hsv)
    return (int(rgb[0]), int(rgb[1]), int(rgb[2]))


def light_contrast(color: Tuple[int, int, int]) -> Tuple[int, int, int]:
    return change_lightness(color, 200)


def dark_contrast(color: Tuple[int, int, int]) -> Tuple[int, int, int]:
    return change_lightness(color, 55)


def random_digit(allow_zero: bool) -> int:
    if allow_zero:
        return random.randint(0, 9)
    else:
        return random.randint(1, 9)


def get_date(fmt: str) -> str:
    if fmt:
        return datetime.now().strftime(fmt)
    else:
        return str(int(datetime.now().timestamp()))


def add_alpha(rgb: Tuple[int, int, int], alpha: float) -> Tuple[int, int, int, int]:
    return int(rgb[0]), int(rgb[1]), int(rgb[2]), int(255 * alpha)


def color_name(name: str) -> Union[Tuple[int, int, int], None]:
    try:
        return tuple(webcolors.name_to_rgb(name))
    except BaseException:
        return None


def clean_lines(s: str) -> str:
    cleaned = s
    cleaned = re.sub(r" *(\n|\\n) *", "\n", cleaned)
    cleaned = re.sub(r" +", " ", cleaned)
    return cleaned.strip()


def divisible(number: int, by: int) -> int:
    while number % by != 0:
        number += 1

    return number


def msg(message: str) -> None:
    print(message, file=sys.stderr)


def respond(message: str) -> None:
    print(message, file=sys.stdout)


def colortext(color: str, text: str) -> str:
    codes = {
        "red": "\x1b[31m",
        "green": "\x1b[32m",
        "yellow": "\x1b[33m",
        "blue": "\x1b[34m",
        "magenta": "\x1b[35m",
        "cyan": "\x1b[36m",
    }

    if color in codes:
        code = codes[color]
        text = f"{code}{text}\x1b[0m"

    return text