# -*- coding: utf-8 -*-

__author__ = 'Traphix'


import time
import logging

from .database import db
from .database.models import MRouterLogTable


class MRDatabaseHandler(logging.Handler):

    def __init__(self, module_name, level=logging.NOTSET):
        logging.Handler.__init__(self, level)
        self.module_name = module_name

    def emit(self, record):
        try:
            db.session.add(
                MRouterLogTable(
                    module_name=self.module_name,
                    time=time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(record.created)),
                    level=record.levelname,
                    content=record.getMessage()
                )
            )
            db.session.commit()
        except Exception:
            self.handleError(record)