mirror of
https://github.com/dbt-labs/dbt-core
synced 2025-12-18 22:11:27 +00:00
Compare commits
16 Commits
enable-pos
...
enhance/lo
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e2a316bf1a | ||
|
|
7ef8bac36e | ||
|
|
f99feda6e4 | ||
|
|
a3d49aacff | ||
|
|
c9f5869959 | ||
|
|
a956092ad9 | ||
|
|
2ec952b13d | ||
|
|
ab098cb09f | ||
|
|
c7a926af13 | ||
|
|
a795fc6787 | ||
|
|
2e0fc37d71 | ||
|
|
d15b869cae | ||
|
|
666a47735d | ||
|
|
7a539fd2cd | ||
|
|
d6a6216270 | ||
|
|
039eecabbc |
@@ -230,6 +230,7 @@ class UserConfig(ExtensibleDbtClassMixin, Replaceable, UserConfigContract):
|
||||
use_colors: Optional[bool] = None
|
||||
partial_parse: Optional[bool] = None
|
||||
printer_width: Optional[int] = None
|
||||
#TODO: add the yaml config options here
|
||||
|
||||
def set_values(self, cookie_dir):
|
||||
if self.send_anonymous_usage_stats:
|
||||
|
||||
@@ -18,6 +18,8 @@ WRITE_JSON = None
|
||||
PARTIAL_PARSE = None
|
||||
USE_COLORS = None
|
||||
STORE_FAILURES = None
|
||||
LOGGER_MAX_SIZE = None
|
||||
LOGGER_BACKUP_COUNT = None
|
||||
|
||||
|
||||
def env_set_truthy(key: str) -> Optional[str]:
|
||||
@@ -56,7 +58,7 @@ MP_CONTEXT = _get_context()
|
||||
def reset():
|
||||
global STRICT_MODE, FULL_REFRESH, USE_CACHE, WARN_ERROR, TEST_NEW_PARSER, \
|
||||
USE_EXPERIMENTAL_PARSER, WRITE_JSON, PARTIAL_PARSE, MP_CONTEXT, USE_COLORS, \
|
||||
STORE_FAILURES
|
||||
STORE_FAILURES, LOGGER_MAX_SIZE, LOGGER_BACKUP_COUNT
|
||||
|
||||
STRICT_MODE = False
|
||||
FULL_REFRESH = False
|
||||
@@ -69,12 +71,14 @@ def reset():
|
||||
MP_CONTEXT = _get_context()
|
||||
USE_COLORS = True
|
||||
STORE_FAILURES = False
|
||||
LOGGER_MAX_SIZE = 10 # default 10mb
|
||||
LOGGER_BACKUP_COUNT = 5
|
||||
|
||||
|
||||
def set_from_args(args):
|
||||
global STRICT_MODE, FULL_REFRESH, USE_CACHE, WARN_ERROR, TEST_NEW_PARSER, \
|
||||
USE_EXPERIMENTAL_PARSER, WRITE_JSON, PARTIAL_PARSE, MP_CONTEXT, USE_COLORS, \
|
||||
STORE_FAILURES
|
||||
STORE_FAILURES, LOGGER_MAX_SIZE, LOGGER_BACKUP_COUNT
|
||||
|
||||
USE_CACHE = getattr(args, 'use_cache', USE_CACHE)
|
||||
|
||||
@@ -98,7 +102,8 @@ def set_from_args(args):
|
||||
if use_colors_override is not None:
|
||||
USE_COLORS = use_colors_override
|
||||
|
||||
STORE_FAILURES = getattr(args, 'store_failures', STORE_FAILURES)
|
||||
LOGGER_MAX_SIZE = getattr(args, 'logger_max_size', LOGGER_MAX_SIZE)
|
||||
LOGGER_BACKUP_COUNT = getattr(args, 'logger_backup_count', LOGGER_BACKUP_COUNT)
|
||||
|
||||
|
||||
# initialize everything to the defaults on module load
|
||||
|
||||
@@ -389,8 +389,8 @@ class DelayedFileHandler(logbook.RotatingFileHandler, FormatterMixin):
|
||||
level=logbook.DEBUG,
|
||||
filter=None,
|
||||
bubble=True,
|
||||
max_size=10 * 1024 * 1024, # 10 mb
|
||||
backup_count=5,
|
||||
max_size=dbt.flags.LOGGER_MAX_SIZE * 1024 * 1024, # in megabytes
|
||||
backup_count=dbt.flags.LOGGER_BACKUP_COUNT,
|
||||
) -> None:
|
||||
self.disabled = False
|
||||
self._msg_buffer: Optional[List[logbook.LogRecord]] = []
|
||||
@@ -537,6 +537,12 @@ class LogManager(logbook.NestedSetup):
|
||||
|
||||
def set_path(self, path):
|
||||
self._file_handler.set_path(path)
|
||||
|
||||
def set_logger_file_configs(self):
|
||||
""""Refreshes the log file configs with newly defined global flags"""
|
||||
self._file_handler = DelayedFileHandler(
|
||||
max_size=(dbt.flags.LOGGER_MAX_SIZE * 1024 * 1024),
|
||||
backup_count=dbt.flags.LOGGER_BACKUP_COUNT)
|
||||
|
||||
def initialized(self):
|
||||
return self._file_handler.initialized
|
||||
|
||||
@@ -1057,6 +1057,26 @@ def parse_args(args, cls=DBTArgumentParser):
|
||||
'''
|
||||
)
|
||||
|
||||
p.add_argument(
|
||||
'--logger-max-size',
|
||||
default=10,
|
||||
type=int,
|
||||
dest='logger_max_size',
|
||||
help='''
|
||||
Set the max size of individual log files. Currently defaults to 10mb per file.
|
||||
'''
|
||||
)
|
||||
|
||||
p.add_argument(
|
||||
'--logger-backup-count',
|
||||
default=5,
|
||||
type=int,
|
||||
dest='logger_backup_count',
|
||||
help='''
|
||||
Set the max number of individual backup log files generated. Currently defaults to 5 backups.
|
||||
'''
|
||||
)
|
||||
|
||||
subs = p.add_subparsers(title="Available sub-commands")
|
||||
|
||||
base_subparser = _build_base_subparser()
|
||||
|
||||
@@ -68,6 +68,7 @@ class BaseTask(metaclass=ABCMeta):
|
||||
log_manager.format_json()
|
||||
else:
|
||||
log_manager.format_text()
|
||||
log_manager.set_logger_file_configs()
|
||||
|
||||
@classmethod
|
||||
def from_args(cls, args):
|
||||
|
||||
Reference in New Issue
Block a user