fix: hide sensitive info in logs

orenlab · Dec 30, 2024 · 41d69ac · 41d69ac
1 parent 260c798
commit 41d69ac
Showing 1 changed file with 17 additions and 9 deletions.
diff --git 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]: