utilities.py

from typing import Any, Callable, Iterable, Optional, TypeVar

A = TypeVar('A')
B = TypeVar('B')
C = TypeVar('C')


def idem(x: A) -> A:
    return x


def equals(x: A) -> Callable[[A], bool]:
    return lambda y: x == y


def first(tuple: tuple[A, B]) -> A:
    (a, b) = tuple
    return a


def second(tuple: tuple[A, B]) -> B:
    (a, b) = tuple
    return b


def strs(xs: Any) -> list[str]:
    return list(map(str, xs))


def compose2(f: Callable[[B], C], g: Callable[[A], B]) -> Callable[[A], C]:
    return lambda x: f(g(x))


def fromOptional(value: Optional[A], fallback: A) -> A:
    return value if value is not None else fallback


def itemList(prefix: str, strs: Iterable[str]) -> str:
    return "\n".join(map(lambda s: prefix + s, strs))


def stripIndentation(string: str) -> str:
    return "\n".join(map(lambda line: line.lstrip(), string.split("\n")))


def flag(name: str) -> str:
    return "--" + name


def shortFlag(name: str) -> str:
    return "-" + name