python自定义控制台输出记录的格式和等级

# !/usr/bin/env python
# -*-coding:utf-8 -*-

from datetime import datetime

from colorama import Fore, Style


def pp(*values, sep=' ', end='\n', file=None, color=None, level=None):
time_str = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
str_values = [f'[{time_str}]:']

for value in values:
if level is None and str(value).startswith('[ERROR]'):
level = 'error'
elif level is None and str(value).startswith('[WARNING]'):
level = 'warning'
elif level is None and str(value).startswith('[INFO]'):
level = 'info'
elif level is None and str(value).startswith('[SUCCESS]'):
level = 'success'

if level == 'info':
color = Fore.BLUE
elif level == 'warning':
color = Fore.YELLOW
elif level == 'error':
color = Fore.RED
elif level == 'success':
color = Fore.GREEN

for value in values:
str_values.append(str(value))
string = sep.join(str_values) + end
if color:
string = f'{color}{string}{Style.RESET_ALL}'
if file is None:
print(string)
else:
with open(file, 'a') as f:
f.write(string)


if __name__ == "__main__":
# 自动设置日志级别
pp(level='error')
pp('[WARNING] Be careful!')
pp('[INFO] FYI')
pp(level='success')

pp('Hello', 'World!')
pp('Message')