niming_backend/utils/logger.py

from utils import pgclass
from sqlalchemy.orm import sessionmaker

def logger(engine, type, message):
    Session = sessionmaker(bind=engine)
    session = Session()
    table = pgclass.SQLlog

    flag = False
    # new post & del post
    if type == "newpost" or type == "delpost":
        flag = True
        log = table(source = "general", message = message)
    elif type in ["login", "user.create", "user.delete", "article.delete", "article.pend"]:
        flag = True
        log = table(source = "admin", message = message)
    elif type in ["server.start"]:
        flag = True
        log = table(source = "server", message = message)
    
    # session.add
    if flag:
        session.add(log)
        session.commit()
    
    # session.close
    session.close()