Skip to content

Commit

Permalink
update versionner
Browse files Browse the repository at this point in the history
  • Loading branch information
AntSimi committed Jul 29, 2024
1 parent 80f8e14 commit 27bafc6
Show file tree
Hide file tree
Showing 2 changed files with 296 additions and 165 deletions.
12 changes: 8 additions & 4 deletions src/py_eddy_tracker/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,13 +32,13 @@
del get_versions


def start_logger():
def start_logger(color=True):
FORMAT_LOG = "%(levelname)-8s %(asctime)s %(module)s.%(funcName)s :\n\t%(message)s"
logger = logging.getLogger("pet")
if len(logger.handlers) == 0:
# set up logging to CONSOLE
console = logging.StreamHandler()
console.setFormatter(ColoredFormatter(FORMAT_LOG))
console.setFormatter(ColoredFormatter(FORMAT_LOG, color=color))
# add the handler to the root logger
logger.addHandler(console)
return logger
Expand All @@ -53,13 +53,14 @@ class ColoredFormatter(logging.Formatter):
DEBUG="\033[34m\t",
)

def __init__(self, message):
def __init__(self, message, color=True):
super().__init__(message)
self.with_color = color

def format(self, record):
color = self.COLOR_LEVEL.get(record.levelname, "")
color_reset = "\033[0m"
model = color + "%s" + color_reset
model = (color + "%s" + color_reset) if self.with_color else "%s"
record.msg = model % record.msg
record.funcName = model % record.funcName
record.module = model % record.module
Expand Down Expand Up @@ -696,3 +697,6 @@ def identify_time(str_date):
VAR_DESCR_inv[VAR_DESCR[key]["nc_name"]] = key
for key_old in VAR_DESCR[key].get("old_nc_name", list()):
VAR_DESCR_inv[key_old] = key

from . import _version
__version__ = _version.get_versions()['version']
Loading

0 comments on commit 27bafc6

Please sign in to comment.