File: utils.py

package info (click to toggle)
jupyter-cache 1.0.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 840 kB
  • sloc: python: 2,601; makefile: 40; sh: 9
file content (24 lines) | stat: -rw-r--r-- 530 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
 
import logging

import click


class ClickLogHandler(logging.Handler):
    _use_stderr = True

    def emit(self, record):
        try:
            msg = self.format(record)
            click.echo(msg, err=self._use_stderr)
        except Exception:
            self.handleError(record)


def setup_logger(logger: logging.Logger) -> None:
    """Add handler to log to click."""
    try:
        import click_log
    except ImportError:
        logger.addHandler(ClickLogHandler())
    else:
        click_log.basic_config(logger)