niming_backend/utils/logger.py
2024-11-18 18:19:25 +00:00

29 lines
785 B
Python

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()