|
1 | 1 | import logging |
2 | | -import os |
3 | 2 | from datetime import datetime, timezone |
4 | 3 | from pathlib import Path |
5 | 4 | from typing import Any, Dict, Union |
|
9 | 8 | from rich.table import Table |
10 | 9 | from rich.theme import Theme |
11 | 10 |
|
| 11 | +from dstack._internal import settings |
12 | 12 | from dstack._internal.cli.utils.rich import DstackRichHandler |
13 | 13 | from dstack._internal.core.errors import CLIError, DstackError |
14 | 14 | from dstack._internal.utils.common import get_dstack_dir |
@@ -68,29 +68,22 @@ def configure_logging(): |
68 | 68 |
|
69 | 69 | log_file = _get_cli_log_file() |
70 | 70 |
|
71 | | - level_names = logging.getLevelNamesMapping() |
72 | | - stdout_level_name = os.getenv("DSTACK_CLI_LOG_LEVEL", "INFO").upper() |
73 | | - stdout_level = level_names[stdout_level_name] |
74 | | - dstack_logger.setLevel(stdout_level) |
75 | | - |
76 | 71 | stdout_handler = DstackRichHandler(console=console) |
77 | 72 | stdout_handler.setFormatter(logging.Formatter(fmt="%(message)s", datefmt="[%X]")) |
78 | | - stdout_handler.setLevel(stdout_level) |
| 73 | + stdout_handler.setLevel(settings.CLI_LOG_LEVEL) |
79 | 74 | dstack_logger.addHandler(stdout_handler) |
80 | 75 |
|
81 | | - file_level_name = os.getenv("DSTACK_CLI_FILE_LOG_LEVEL", "DEBUG").upper() |
82 | | - file_level = level_names[file_level_name] |
83 | | - |
84 | 76 | file_handler = logging.FileHandler(log_file) |
85 | 77 | file_handler.setFormatter( |
86 | 78 | logging.Formatter( |
87 | 79 | fmt="%(asctime)s - %(name)s - %(levelname)s - %(message)s", datefmt="%Y-%m-%d %H:%M:%S" |
88 | 80 | ) |
89 | 81 | ) |
90 | | - file_handler.setLevel(file_level) |
| 82 | + file_handler.setLevel(settings.CLI_FILE_LOG_LEVEL) |
91 | 83 | dstack_logger.addHandler(file_handler) |
92 | 84 |
|
93 | | - dstack_logger.setLevel(min(stdout_level, file_level)) |
| 85 | + # the logger allows all messages, filtering is done by the handlers |
| 86 | + dstack_logger.setLevel(logging.DEBUG) |
94 | 87 |
|
95 | 88 |
|
96 | 89 | def confirm_ask(prompt, **kwargs) -> bool: |
|
0 commit comments