diff options
author | dvshkurko <dvshkurko@yandex-team.ru> | 2022-02-10 16:45:52 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:45:52 +0300 |
commit | c768a99151e47c3a4bb7b92c514d256abd301c4d (patch) | |
tree | 1a2c5ffcf89eb53ecd79dbc9bc0a195c27404d0c /library/python/pytest | |
parent | 321ee9bce31ec6e238be26dbcbe539cffa2c3309 (diff) | |
download | ydb-c768a99151e47c3a4bb7b92c514d256abd301c4d.tar.gz |
Restoring authorship annotation for <dvshkurko@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/python/pytest')
-rw-r--r-- | library/python/pytest/plugins/ya.py | 52 |
1 files changed, 26 insertions, 26 deletions
diff --git a/library/python/pytest/plugins/ya.py b/library/python/pytest/plugins/ya.py index 60bb08cfeb..1bde03042d 100644 --- a/library/python/pytest/plugins/ya.py +++ b/library/python/pytest/plugins/ya.py @@ -98,31 +98,31 @@ class YaTestLoggingFileHandler(logging.FileHandler): pass -class _TokenFilterFormatter(logging.Formatter): - def __init__(self, fmt): - super(_TokenFilterFormatter, self).__init__(fmt) - self._replacements = [] - if not self._replacements: - if six.PY2: - for k, v in os.environ.iteritems(): - if k.endswith('TOKEN') and v: - self._replacements.append(v) - elif six.PY3: - for k, v in os.environ.items(): - if k.endswith('TOKEN') and v: - self._replacements.append(v) - self._replacements = sorted(self._replacements) - - def _filter(self, s): - for r in self._replacements: - s = s.replace(r, "[SECRET]") - - return s - - def format(self, record): - return self._filter(super(_TokenFilterFormatter, self).format(record)) - - +class _TokenFilterFormatter(logging.Formatter): + def __init__(self, fmt): + super(_TokenFilterFormatter, self).__init__(fmt) + self._replacements = [] + if not self._replacements: + if six.PY2: + for k, v in os.environ.iteritems(): + if k.endswith('TOKEN') and v: + self._replacements.append(v) + elif six.PY3: + for k, v in os.environ.items(): + if k.endswith('TOKEN') and v: + self._replacements.append(v) + self._replacements = sorted(self._replacements) + + def _filter(self, s): + for r in self._replacements: + s = s.replace(r, "[SECRET]") + + return s + + def format(self, record): + return self._filter(super(_TokenFilterFormatter, self).format(record)) + + def setup_logging(log_path, level=logging.DEBUG, *other_logs): logs = [log_path] + list(other_logs) root_logger = logging.getLogger() @@ -133,7 +133,7 @@ def setup_logging(log_path, level=logging.DEBUG, *other_logs): for log_file in logs: file_handler = YaTestLoggingFileHandler(log_file) log_format = '%(asctime)s - %(levelname)s - %(name)s - %(funcName)s: %(message)s' - file_handler.setFormatter(_TokenFilterFormatter(log_format)) + file_handler.setFormatter(_TokenFilterFormatter(log_format)) file_handler.setLevel(level) root_logger.addHandler(file_handler) |