start.py

# getter < https://t.me/kastaid >
# Copyright (C) 2022-present kastaid
#
# This file is a part of < https://github.com/kastaid/getter/ >
# Please read the GNU Affero General Public License in
# < https://github.com/kastaid/getter/blob/main/LICENSE/ >.

import argparse
import os
import shlex
import sys
from contextlib import (
    suppress,)
from pathlib import Path
from subprocess import (
    run,)
from version import (
    __version__,)

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

filechecker = "bash filecheck.sh --log-level info"
python = "python3"
nocache = f"{python} -B"
app = f"{python} -m pytel"
app_watch = f"{python} -m scripts.autoreload {app}"


# Pylint
autoflake = "autoflake --in-place --ignore-init-module-imports --remove-all-unused-imports -i -r ."
black = "black --line-length 25 --exclude version.py ./pytel"
isort = (
    "isort --settings-file=setup.cfg ."
)
flake8 = "flake8 --config=setup.cfg ."
prettyjson = (
    f"{nocache} -m scripts.prettyjson"
)


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


def clean() -> None:
    with suppress(BaseException):
        for _ in Path(".").rglob(
            "*.py[co]"
        ):
            _.unlink(missing_ok=True)
        for _ in Path(".").rglob(
            "__pycache__"
        ):
            _.rmdir()
        for _ in Path(".").rglob(
            "cache/pip"
        ):
            _.rmdir()


def lint() -> None:
    print(f"{CYAN}>> {prettyjson}{RST}")
    run_cmd(prettyjson)
    print(f"{CYAN}>> {autoflake}{RST}")
    run_cmd(autoflake)
    print(f"{CYAN}>> {black}{RST}")
    run_cmd(black)
    print(f"{CYAN}>> {isort}{RST}")
    run_cmd(isort)
    print(f"{CYAN}>> {flake8}{RST}")
    run_cmd(flake8)


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


parser = argparse.ArgumentParser(
    formatter_class=CapitalisedHelpFormatter,
    prog=f"{GREEN}{python} -m start{RST}",
    usage="%(prog)s [options]",
    epilog="Source code https://github.com/kastaid",
    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(
    "-c",
    "--clean",
    help="remove python caches",
    action="store_true",
)
parser.add_argument(
    "-ch",
    "--checker",
    help="checking attributes file",
    action="store_true",
)
parser.add_argument(
    "-v",
    "--version",
    help="show this program version",
    action="version",
    version=f"PYTEL-Premium v{__version__}",
)
parser.add_argument(
    "-h",
    "--help",
    help="show this help information",
    default=argparse.SUPPRESS,
    action="help",
)


def main():
    args = parser.parse_args()
    if args.prod:
        os.system("clear")
        print(
            f"{BOLD}{GREEN}PRODUCTION MODE...{RST}"
        )
        clean()
        print(
            f"{BOLD}{BLUE}>> {app}{RST}"
        )
        run_cmd(app)
    elif args.dev:
        os.system("clear")
        print(
            f"{BOLD}{GREEN}DEVELOPMENT MODE...{RST}"
        )
        clean()
        lint()
        print(
            f"{BOLD}{BLUE}>> {app}{RST}"
        )
        run_cmd(app)
    elif args.watch:
        os.system("clear")
        print(
            f"{BOLD}{GREEN}WATCHED DEVELOPMENT MODE...{RST}"
        )
        clean()
        print(
            f"{BOLD}{BLUE}>> {app_watch}{RST}"
        )
        run_cmd(app_watch)
    elif args.lint:
        os.system("clear")
        print(
            f"{BOLD}{YELLOW}Run linting and format code...{RST}"
        )
        clean()
        lint()
    elif args.clean:
        os.system("clear")
        clean()
        print(
            f"{BOLD}{BLUE}✓ Cached has been clear.{RST}"
        )
    elif args.checker:
        os.system("clear")
        print(
            f"{BOLD}{GREEN}Bash >>> Checking Attributes Files{RST}"
        )
        run_cmd(filechecker)
    else:
        print(
            f"{python} -m start --help"
        )


if __name__ == "__main__":
    main()