src.parser

 1import argparse
 2import logging
 3
 4# Настройка логгирования
 5logger = logging.getLogger()
 6sql_logger = logging.getLogger("sqlalchemy.engine.Engine")
 7sched_logger = logging.getLogger("apscheduler")
 8
 9
10def parse_args():
11    """Парсинг аргументов командной строки.
12
13    Создает парсер аргументов и обрабатывает входные параметры для
14    настройки логирования. Поддерживаются следующие параметры:
15
16    - `--nolog`: Опции для отключения логирования:
17        - `all`: Отключает все логирование.
18        - `file`: Отключает логирование в файл.
19        - `console`: Отключает логирование в консоль.
20
21    Returns:
22        argparse.Namespace: Объект с аргументами, переданными через командную строку.
23    """
24    parser = argparse.ArgumentParser(
25        prog="DanVPN Bot",
26        description="Wireguard config manager",
27        epilog="DanVPN bot start script",
28    )
29
30    parser.add_argument(
31        "--nolog",
32        nargs="?",
33        const="all",
34        choices=["file", "console", "all"],
35        help="Unable log",
36    )
37
38    args = parser.parse_args()
39
40    match args.nolog:
41        case "all":
42            logging.disable()
43        case "file":
44            logger.removeHandler(logger.handlers[1])
45            sql_logger.removeHandler(sql_logger.handlers[1])
46            sched_logger.removeHandler(sched_logger.handlers[1])
47        case "console":
48            logger.removeHandler(logger.handlers[0])
49            sql_logger.removeHandler(sql_logger.handlers[0])
50        case _:
51            pass
52
53    return args