Skip to content

Commit 27bafc6

Browse files
committed
update versionner
1 parent 80f8e14 commit 27bafc6

File tree

2 files changed

+296
-165
lines changed

2 files changed

+296
-165
lines changed

src/py_eddy_tracker/__init__.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,13 +32,13 @@
3232
del get_versions
3333

3434

35-
def start_logger():
35+
def start_logger(color=True):
3636
FORMAT_LOG = "%(levelname)-8s %(asctime)s %(module)s.%(funcName)s :\n\t%(message)s"
3737
logger = logging.getLogger("pet")
3838
if len(logger.handlers) == 0:
3939
# set up logging to CONSOLE
4040
console = logging.StreamHandler()
41-
console.setFormatter(ColoredFormatter(FORMAT_LOG))
41+
console.setFormatter(ColoredFormatter(FORMAT_LOG, color=color))
4242
# add the handler to the root logger
4343
logger.addHandler(console)
4444
return logger
@@ -53,13 +53,14 @@ class ColoredFormatter(logging.Formatter):
5353
DEBUG="\033[34m\t",
5454
)
5555

56-
def __init__(self, message):
56+
def __init__(self, message, color=True):
5757
super().__init__(message)
58+
self.with_color = color
5859

5960
def format(self, record):
6061
color = self.COLOR_LEVEL.get(record.levelname, "")
6162
color_reset = "\033[0m"
62-
model = color + "%s" + color_reset
63+
model = (color + "%s" + color_reset) if self.with_color else "%s"
6364
record.msg = model % record.msg
6465
record.funcName = model % record.funcName
6566
record.module = model % record.module
@@ -696,3 +697,6 @@ def identify_time(str_date):
696697
VAR_DESCR_inv[VAR_DESCR[key]["nc_name"]] = key
697698
for key_old in VAR_DESCR[key].get("old_nc_name", list()):
698699
VAR_DESCR_inv[key_old] = key
700+
701+
from . import _version
702+
__version__ = _version.get_versions()['version']

0 commit comments

Comments
 (0)