From 03d90daedcd236404caa4d562476b9ee8beb5a70 Mon Sep 17 00:00:00 2001
From: A_D <aunderscored@gmail.com>
Date: Thu, 12 Aug 2021 16:46:04 +0200
Subject: [PATCH] added support for --trace-all argument to ignore trace-if
 directives

---
 EDMarketConnector.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/EDMarketConnector.py b/EDMarketConnector.py
index 9ead323c..259fcce3 100755
--- a/EDMarketConnector.py
+++ b/EDMarketConnector.py
@@ -76,6 +76,12 @@ if __name__ == '__main__':  # noqa: C901
         action='store_true',
     )
 
+    parser.add_argument(
+        "--trace-all",
+        help="Disable trace-on functionality",
+        action='store_true'
+    )
+
     parser.add_argument(
         '--reset-ui',
         help='reset UI theme and transparency to defaults',
@@ -123,9 +129,16 @@ if __name__ == '__main__':  # noqa: C901
 
     args = parser.parse_args()
 
+    level_to_set: Optional[int] = None
     if args.trace:
-        logger.setLevel(logging.TRACE)
-        edmclogger.set_channels_loglevel(logging.TRACE)
+        level_to_set = logging.TRACE  # type: ignore # it exists
+
+    elif args.trace_all:
+        level_to_set = logging.TRACE_ALL  # type: ignore # it exists
+
+    if level_to_set is not None:
+        logger.setLevel(level_to_set)
+        edmclogger.set_channels_loglevel(level_to_set)
 
     if args.force_localserver_for_auth:
         config.set_auth_force_localserver()