Fedir Zadniprovskyi commited on
Commit
836a395
·
1 Parent(s): 1cffcb2

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

Browse files
Files changed (1) hide show
  1. src/faster_whisper_server/logger.py +26 -4
src/faster_whisper_server/logger.py CHANGED
@@ -1,8 +1,30 @@
1
  import logging
 
2
 
3
 
4
  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")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import logging
2
+ import logging.config
3
 
4
 
5
  def setup_logger(log_level: str) -> None:
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)