Commit @8d73e2b84f493d24261dca62f26195c83377a742 - yjyoon/whisper_server

Fedir Zadniprovskyi 01-07

use dict to configure the logger (also fixes #114)

@8d73e2b84f493d24261dca62f26195c83377a742

0d207cc

8d73e2b

src/faster_whisper_server/logger.py

--- src/faster_whisper_server/logger.py

+++ src/faster_whisper_server/logger.py


 import logging
+import logging.config
 
 
 def setup_logger(log_level: str) -> None:
-    logging.getLogger().setLevel(logging.INFO)
-    logger = logging.getLogger(__name__)
-    logger.setLevel(log_level.upper())
-    logging.basicConfig(format="%(asctime)s:%(levelname)s:%(name)s:%(funcName)s:%(lineno)d:%(message)s")
+    assert log_level.upper() in ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"], log_level
+    # https://www.youtube.com/watch?v=9L77QExPmI0
+    # https://docs.python.org/3/library/logging.config.html
+    logging_config = {
+        "version": 1,  # required
+        "disable_existing_loggers": False,
+        "formatters": {
+            "simple": {"format": "%(asctime)s:%(levelname)s:%(name)s:%(funcName)s:%(lineno)d:%(message)s"},
+        },
+        "handlers": {
+            "stdout": {
+                "class": "logging.StreamHandler",
+                "formatter": "simple",
+                "stream": "ext://sys.stdout",
+            },
+        },
+        "loggers": {
+            "root": {
+                "level": log_level.upper(),
+                "handlers": ["stdout"],
+            },
+        },
+    }
+
+    logging.config.dictConfig(logging_config)

Add a comment

Open 0
Closed 0

List

...	...	@@ -1,8 +1,30 @@
1	1	import logging
	2	+import logging.config
2	3
3	4
4	5	def setup_logger(log_level: str) -> None:
5		- logging.getLogger().setLevel(logging.INFO)
6		- logger = logging.getLogger(__name__)
7		- logger.setLevel(log_level.upper())
8		- logging.basicConfig(format="%(asctime)s:%(levelname)s:%(name)s:%(funcName)s:%(lineno)d:%(message)s")
	6	+ assert log_level.upper() in ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"], log_level
	7	+ # https://www.youtube.com/watch?v=9L77QExPmI0
	8	+ # https://docs.python.org/3/library/logging.config.html
	9	+ logging_config = {
	10	+ "version": 1, # required
	11	+ "disable_existing_loggers": False,
	12	+ "formatters": {
	13	+ "simple": {"format": "%(asctime)s:%(levelname)s:%(name)s:%(funcName)s:%(lineno)d:%(message)s"},
	14	+ },
	15	+ "handlers": {
	16	+ "stdout": {
	17	+ "class": "logging.StreamHandler",
	18	+ "formatter": "simple",
	19	+ "stream": "ext://sys.stdout",
	20	+ },
	21	+ },
	22	+ "loggers": {
	23	+ "root": {
	24	+ "level": log_level.upper(),
	25	+ "handlers": ["stdout"],
	26	+ },
	27	+ },
	28	+ }
	29	+
	30	+ logging.config.dictConfig(logging_config)

Delete comment