KCGL/inventory-backend/app/models/system.py

# 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
        }