run.py

# Copyright (C) 2023-present kastaid
# https://github.com/kastaid/ds
# MIT License

import argparse
import shlex
import subprocess
import sys
from pathlib import Path

from version import __version__

RST = "\x1b[0m"
BOLD = "\x1b[1m"
GREEN = "\x1b[32m"
YELLOW = "\x1b[33m"
BLUE = "\x1b[34m"
CYAN = "\x1b[36m"

python = "python3"
nocache = f"{python} -B"
app = f"{python} -m ds"
app_watch = f"{python} -m scripts.autoreload {app}"
ruff_check = "ruff check ."
ruff_format = "ruff format ."
ruff_fix = "ruff check --fix --unsafe-fixes ."
prettyjson = f"{nocache} -m scripts.prettyjson"


def run_cmd(cmd: str) -> None:
    try:
        proc = subprocess.run(shlex.split(cmd), shell=False, check=False)
        if proc.returncode != 0:
            print(f"Exit code {proc.returncode}")
            sys.exit(1)
    except BaseException:
        sys.exit(1)


def clean() -> None:
    for i in Path().rglob("*.py[co]"):
        i.unlink(missing_ok=True)
    for i in Path().rglob("__pycache__"):
        i.rmdir()


def lint() -> None:
    print(f"{CYAN}>> {prettyjson}{RST}")
    run_cmd(prettyjson)
    print(f"{CYAN}>> {ruff_check}{RST}")
    run_cmd(ruff_check)
    print(f"{CYAN}>> {ruff_format}{RST}")
    run_cmd(ruff_format)


def fix() -> None:
    print(f"{CYAN}>> {ruff_fix}{RST}")
    run_cmd(ruff_fix)


class CapitalisedHelpFormatter(argparse.HelpFormatter):
    def add_usage(self, usage, actions, groups, prefix=None):
        if not prefix:
            prefix = "Usage: "
        return super().add_usage(usage, actions, groups, prefix)


parser = argparse.ArgumentParser(
    formatter_class=CapitalisedHelpFormatter,
    prog=f"{GREEN}{python} -m run{RST}",
    usage="%(prog)s [options]",
    epilog="Source code https://github.com/kastaid/ds",
    add_help=False,
)
parser._optionals.title = "Options"
parser.add_argument(
    "-p",
    "--prod",
    help="run in production mode",
    action="store_true",
)
parser.add_argument(
    "-d",
    "--dev",
    help="run in development mode",
    action="store_true",
)
parser.add_argument(
    "-w",
    "--watch",
    help="run and watch in development mode",
    action="store_true",
)
parser.add_argument(
    "-l",
    "--lint",
    help="run linting and format code",
    action="store_true",
)
parser.add_argument(
    "-f",
    "--fix",
    help="run fixed code",
    action="store_true",
)
parser.add_argument(
    "-c",
    "--clean",
    help="remove python caches",
    action="store_true",
)
parser.add_argument(
    "-v",
    "--version",
    help="show this program version",
    action="version",
    version=__version__,
)
parser.add_argument(
    "-h",
    "--help",
    help="show this help information",
    default=argparse.SUPPRESS,
    action="help",
)


def main() -> None:
    args = parser.parse_args()
    if args.prod:
        print(f"{BOLD}{GREEN}PRODUCTION MODE...{RST}")
        clean()
        print(f"{BOLD}{BLUE}>> {app}{RST}")
        run_cmd(app)
    elif args.dev:
        print(f"{BOLD}{GREEN}DEVELOPMENT MODE...{RST}")
        clean()
        lint()
        print(f"{BOLD}{BLUE}>> {app}{RST}")
        run_cmd(app)
    elif args.watch:
        print(f"{BOLD}{GREEN}WATCHED DEVELOPMENT MODE...{RST}")
        clean()
        print(f"{BOLD}{BLUE}>> {app_watch}{RST}")
        run_cmd(app_watch)
    elif args.lint:
        print(f"{BOLD}{YELLOW}Run linting and format code...{RST}")
        clean()
        lint()
    elif args.fix:
        print(f"{BOLD}{YELLOW}Run fixed code...{RST}")
        clean()
        fix()
    elif args.clean:
        clean()
    else:
        print(f"{python} -m run --help")


if __name__ == "__main__":
    raise SystemExit(main())