Hugging Face's logo

Spaces:
Misaka20001
/
HibiAPI
Runtime error

App Files Community
HibiAPI
File size: 3,244 Bytes 
0a1b571
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
 
import os
from pathlib import Path

import typer
import uvicorn

from hibiapi import __file__ as root_file
from hibiapi import __version__
from hibiapi.utils.config import CONFIG_DIR, DEFAULT_DIR, Config
from hibiapi.utils.log import LOG_LEVEL, logger

COPYRIGHT = r"""
<b><g>
  _    _ _ _     _          _____ _____  
 | |  | (_) |   (_)   /\   |  __ \_   _| 
 | |__| |_| |__  _   /  \  | |__) || |   
 |  __  | | '_ \| | / /\ \ |  ___/ | |   
 | |  | | | |_) | |/ ____ \| |    _| |_  
 |_|  |_|_|_.__/|_/_/    \_\_|   |_____| 
</g><e>
A program that implements easy-to-use APIs for a variety of commonly used sites
Repository: https://github.com/mixmoe/HibiAPI
</e></b>""".strip()  # noqa:W291


LOG_CONFIG = {
    "version": 1,
    "disable_existing_loggers": False,
    "handlers": {
        "default": {
            "class": "hibiapi.utils.log.LoguruHandler",
        },
    },
    "loggers": {
        "uvicorn.error": {
            "handlers": ["default"],
            "level": LOG_LEVEL,
        },
        "uvicorn.access": {
            "handlers": ["default"],
            "level": LOG_LEVEL,
        },
    },
}

RELOAD_CONFIG = {
    "reload": True,
    "reload_dirs": [
        *map(str, [Path(root_file).parent.absolute(), CONFIG_DIR.absolute()])
    ],
    "reload_includes": ["*.py", "*.yml"],
}


cli = typer.Typer()


@cli.callback(invoke_without_command=True)
@cli.command()
def run(
    ctx: typer.Context,
    host: str = Config["server"]["host"].as_str(),
    port: int = Config["server"]["port"].as_number(),
    workers: int = 1,
    reload: bool = False,
):
    if ctx.invoked_subcommand is not None:
        return

    if ctx.info_name != (func_name := run.__name__):
        logger.warning(
            f"Directly usage of command <r>{ctx.info_name}</r> is <b>deprecated</b>, "
            f"please use <g>{ctx.info_name} {func_name}</g> instead."
        )

    try:
        terminal_width, _ = os.get_terminal_size()
    except OSError:
        terminal_width = 0
    logger.warning(
        "\n".join(i.center(terminal_width) for i in COPYRIGHT.splitlines()),
    )
    logger.info(f"HibiAPI version: <g><b>{__version__}</b></g>")

    uvicorn.run(
        "hibiapi.app:app",
        host=host,
        port=port,
        access_log=False,
        log_config=LOG_CONFIG,
        workers=workers,
        forwarded_allow_ips=Config["server"]["allowed-forward"].get_optional(str),
        **(RELOAD_CONFIG if reload else {}),
    )


@cli.command()
def config(force: bool = False):
    total_written = 0
    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
    for file in os.listdir(DEFAULT_DIR):
        default_path = DEFAULT_DIR / file
        config_path = CONFIG_DIR / file
        if not (existed := config_path.is_file()) or force:
            total_written += config_path.write_text(
                default_path.read_text(encoding="utf-8"),
                encoding="utf-8",
            )
            typer.echo(
                typer.style(("Overwritten" if existed else "Created") + ": ", fg="blue")
                + typer.style(str(config_path), fg="yellow")
            )
    if total_written > 0:
        typer.echo(f"Config folder generated, {total_written=}")


if __name__ == "__main__":
    cli()