Added optional logging configuration. (#159)

Author: s3rius

poetry.lock

@@ -27,11 +27,9 @@ class WorkerArgs:
     modules: List[str]
     tasks_pattern: str = "tasks.py"
     fs_discover: bool = False
+    configure_logging: bool = True
     log_level: LogLevel = LogLevel.INFO
     workers: int = 2
-    log_collector_format: str = (
-        "[%(asctime)s][%(levelname)-7s][%(module)s:%(funcName)s:%(lineno)d] %(message)s"
-    )
     max_threadpool_threads: int = 10
     no_parse: bool = False
     shutdown_timeout: float = 5
@@ -118,18 +116,6 @@ def from_cli(  # noqa: WPS213
             default=2,
             help="Number of worker child processes",
         )
-        parser.add_argument(
-            "--log-collector-format",
-            "-lcf",
-            type=str,
-            default=(
-                "[%(asctime)s]"
-                "[%(levelname)-7s]"
-                "[%(module)s:%(funcName)s:%(lineno)d] "
-                "%(message)s"
-            ),
-            help="Format which is used when collecting logs from function execution",
-        )
         parser.add_argument(
             "--no-parse",
             action="store_true",
@@ -187,6 +173,12 @@ def from_cli(  # noqa: WPS213
             default=0,
             help="Maximum prefetched tasks per worker process. ",
         )
+        parser.add_argument(
+            "--no-configure-logging",
+            action="store_false",
+            dest="configure_logging",
+            help="Use this parameter if your application configures custom logging.",
+        )
 
         namespace = parser.parse_args(args)
         return WorkerArgs(**namespace.__dict__)

taskiq/cli/worker/args.py

@@ -158,10 +158,13 @@ def run_worker(args: WorkerArgs) -> None:  # noqa: WPS213
 
     :raises ValueError: if reload flag is used, but dependencies are not installed.
     """
-    logging.basicConfig(
-        level=logging.getLevelName(args.log_level),
-        format="[%(asctime)s][%(name)s][%(levelname)-7s][%(processName)s] %(message)s",
-    )
+    if args.configure_logging:
+        logging.basicConfig(
+            level=logging.getLevelName(args.log_level),
+            format="[%(asctime)s][%(name)s][%(levelname)-7s]"
+            + "[%(processName)s] %(message)s",
+        )
+    logging.getLogger("taskiq").setLevel(level=logging.getLevelName(args.log_level))
     logging.getLogger("watchdog.observers.inotify_buffer").setLevel(level=logging.INFO)
     logger.info("Starting %s worker processes.", args.workers)