From 41d69ac5e79b1ed2e7d6cd915710480e183eb71a Mon Sep 17 00:00:00 2001 From: Den Rozhnovskiy Date: Mon, 30 Dec 2024 08:57:57 +0500 Subject: [PATCH] fix: hide sensitive info in logs --- pytmbot/utils/utilities.py | 26 +++++++++++++++++--------- 1 file changed, 17 insertions(+), 9 deletions(-) diff --git a/pytmbot/utils/utilities.py b/pytmbot/utils/utilities.py index 078f016..fa7441e 100644 --- a/pytmbot/utils/utilities.py +++ b/pytmbot/utils/utilities.py @@ -17,6 +17,7 @@ # Utility functions +@lru_cache(maxsize=None) def parse_cli_args() -> argparse.Namespace: """ Parses command line arguments using `argparse`. @@ -32,33 +33,40 @@ def parse_cli_args() -> argparse.Namespace: default="prod", help="PyTMBot mode (dev or prod)", ) + parser.add_argument( "--log-level", choices=["DEBUG", "INFO", "ERROR"], default="INFO", help="Log level", ) + parser.add_argument( - "--colorize-logs", - action=argparse.BooleanOptionalAction, - default=True, - help="Enable or disable log colorization", + "--colorize_logs", + choices=["True", "False"], + default="True", + help="Colorize logs", ) + parser.add_argument( "--webhook", - action=argparse.BooleanOptionalAction, - default=False, + choices=["True", "False"], + default="False", help="Start in webhook mode", ) + parser.add_argument( - "--socket-host", + "--socket_host", default="127.0.0.1", - help="Socket host for webhook mode", + help="Socket host for listening in webhook mode", ) + parser.add_argument( "--plugins", nargs="+", default=[], help="List of plugins to load" ) - return parser.parse_args() + + args = parser.parse_args() + return args def round_up_tuple(numbers: Tuple[float, ...]) -> Dict[int, float]: