printer.py

import sys
import time
from typing import Optional

        
class BiliLogger:
    # 格式化数据
    @staticmethod
    def format(
            *objects,
            extra_info: Optional[str] = None,
            need_timestamp: bool = True):
        timestamp = time.strftime("[%Y-%m-%d %H:%M:%S]", time.localtime())\
            if need_timestamp else '>'
        extra_info = f' ({extra_info})' if extra_info is not None else ''
        if objects:
            first_value, *others = objects
            others = [f'# {i}' for i in others]
            return (f'{timestamp} {first_value}{extra_info}', *others)
        return f'{timestamp} NULL{extra_info}',

    def infos(
            self,
            list_objects,
            **kwargs):
        self.info(*list_objects, **kwargs)

    def info(
            self,
            *objects,
            extra_info: Optional[str] = None,
            need_timestamp: bool = True):
        texts = self.format(
            *objects,
            extra_info=extra_info,
            need_timestamp=need_timestamp)
        for i in texts:
            print(i)

    def warn(
            self,
            *objects,
            extra_info: Optional[str] = None):
        texts = self.format(
            *objects,
            extra_info=extra_info,
            need_timestamp=True)
        for i in texts:
            print(i, file=sys.stderr)
        
        with open('bili.log', 'a', encoding='utf-8') as f:
            for i in texts:
                f.write(f'{i}\n')

    def debug(
            self,
            *objects,
            **kwargs):
        self.warn(*objects, **kwargs)

    def error(
            self,
            *objects,
            **kwargs):
        self.warn(*objects, **kwargs)
        sys.exit(-1)


printer = BiliLogger()


def info(*objects, **kwargs):
    printer.info(*objects, **kwargs)


def infos(list_objects, **kwargs):
    printer.infos(list_objects, **kwargs)


def warn(*objects, **kwargs):
    printer.warn(*objects, **kwargs)


def error(*objects, **kwargs):
    printer.error(*objects, **kwargs)


def debug(*objects, **kwargs):
    printer.debug(*objects, **kwargs)