Skip to content

Commit 31ffa59

Browse files
committed
add argument to log time
Signed-off-by: Zen <[email protected]>
1 parent be88c6d commit 31ffa59

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

src/zenlib/util/init_argparser.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ def init_argparser(prog=None, description=None):
88
argparser.add_argument('-v', '--version', action='store_true', help='Print the version and exit.')
99
argparser.add_argument('--log-file', type=str, help='Log file path.')
1010
argparser.add_argument('--log-level', type=str, help='Log level.')
11+
argparser.add_argument('--log-time', action='store_true', help='Log timestamps.')
1112
argparser.add_argument('--no-log-color', action='store_true', help='Disable log color.')
1213

1314
return argparser

src/zenlib/util/process_args.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,11 @@ def process_args(argparser, logger=None):
2121
log_level = 20
2222
logger.setLevel(log_level)
2323

24-
if log_level < 10:
25-
format_str = '%(levelname)s | %(name)-42s | %(message)s'
26-
elif log_level < 20:
27-
format_str = '%(levelname)s | %(name)-42s | %(message)s'
24+
format_str = '%(asctime)s | ' if args.log_time else ''
25+
if log_level < 20:
26+
format_str += '%(levelname)s | %(name)-42s | %(message)s'
2827
else:
29-
format_str = None
28+
format_str += '%(levelname)s | %(message)s'
3029
formatter = ColorLognameFormatter(format_str) if not args.no_log_color else Formatter(format_str)
3130

3231
# Add the formatter to the first handler, or add a new handler

0 commit comments

Comments
 (0)