74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
# app/models/system.py
|
||
from app.extensions import db
|
||
from werkzeug.security import generate_password_hash, check_password_hash
|
||
from datetime import datetime
|
||
from sqlalchemy.sql import func
|
||
|
||
|
||
class SysUser(db.Model):
|
||
"""
|
||
系统用户表
|
||
对应数据库: sys_user
|
||
"""
|
||
__tablename__ = 'sys_user'
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
username = db.Column(db.String(100), nullable=False)
|
||
email = db.Column(db.String(100), unique=True)
|
||
department = db.Column(db.String(100))
|
||
role = db.Column(db.String(50)) # 存储 UserRole 的值，如 'SUPER_ADMIN'
|
||
status = db.Column(db.String(20), default='active')
|
||
password_hash = db.Column(db.Text)
|
||
|
||
# [关键] 对应数据库的 created_at 字段
|
||
# 如果数据库报错 column not found，请务必执行 SQL: ALTER TABLE sys_user ADD COLUMN created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP;
|
||
created_at = db.Column(db.DateTime, server_default=func.now(), default=datetime.now)
|
||
|
||
def set_password(self, password):
|
||
"""生成加密密码"""
|
||
self.password_hash = generate_password_hash(password)
|
||
|
||
def check_password(self, password):
|
||
"""验证密码"""
|
||
return check_password_hash(self.password_hash, password)
|
||
|
||
def to_dict(self):
|
||
"""序列化为字典，供接口返回使用"""
|
||
return {
|
||
'id': self.id,
|
||
'username': self.username,
|
||
'email': self.email,
|
||
'department': self.department,
|
||
'role': self.role,
|
||
'status': self.status,
|
||
'created_at': self.created_at.strftime('%Y-%m-%d %H:%M:%S') if self.created_at else ''
|
||
}
|
||
|
||
|
||
class SysLog(db.Model):
|
||
"""
|
||
系统操作日志表
|
||
对应数据库表: sys_log
|
||
"""
|
||
__tablename__ = 'sys_log'
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
op_time = db.Column(db.DateTime, default=datetime.now)
|
||
op_user_name = db.Column(db.String(100))
|
||
op_user_id = db.Column(db.String(50))
|
||
module_name = db.Column(db.String(100))
|
||
action_type = db.Column(db.String(50))
|
||
target_table = db.Column(db.String(100))
|
||
target_id = db.Column(db.Integer)
|
||
description = db.Column(db.Text)
|
||
ip_address = db.Column(db.String(50))
|
||
|
||
def to_dict(self):
|
||
return {
|
||
'id': self.id,
|
||
'op_time': self.op_time.isoformat() if self.op_time else None,
|
||
'op_user_name': self.op_user_name,
|
||
'module_name': self.module_name,
|
||
'action_type': self.action_type,
|
||
'description': self.description
|
||
} |