189 lines
6.5 KiB
Python
189 lines
6.5 KiB
Python
# inventory-backend/app/models/system.py
|
||
from app.extensions import db
|
||
from werkzeug.security import generate_password_hash, check_password_hash
|
||
from datetime import datetime
|
||
|
||
|
||
# ==========================================
|
||
# 1. 系统用户表
|
||
# ==========================================
|
||
class SysUser(db.Model):
|
||
"""
|
||
系统用户表
|
||
对应数据库: sys_user
|
||
username 字段存储格式约定: "真实姓名/登录账号" (例如: 张三/zhangsan01)
|
||
"""
|
||
__tablename__ = 'sys_user'
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
username = db.Column(db.String(100), nullable=False) # 存储 "张三/zhangsan"
|
||
email = db.Column(db.String(100), unique=True)
|
||
department = db.Column(db.String(100))
|
||
role = db.Column(db.String(50))
|
||
status = db.Column(db.String(20), default='active')
|
||
password_hash = db.Column(db.Text)
|
||
created_at = db.Column(db.DateTime, 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):
|
||
"""
|
||
序列化为字典
|
||
数据库存的是 '张三/zhangsan'
|
||
前端需要的是 '张三(zhangsan)'
|
||
"""
|
||
raw_name = self.username
|
||
display_name = raw_name
|
||
account_id = raw_name
|
||
|
||
# 解析存储格式: Name/ID
|
||
if '/' in raw_name:
|
||
parts = raw_name.split('/')
|
||
real_name = parts[0]
|
||
acc_id = parts[1]
|
||
# 格式化为前端展示格式: 张三(zhangsan01)
|
||
display_name = f"{real_name}({acc_id})"
|
||
# 单独提取账号ID (如果前端需要单独用)
|
||
account_id = acc_id
|
||
|
||
return {
|
||
'id': self.id,
|
||
'username': display_name, # 列表显示: 张三(zhangsan01)
|
||
'raw_username': self.username, # 原始数据
|
||
'account_id': account_id, # 纯账号ID: zhangsan01
|
||
'email': self.email,
|
||
'department': self.department,
|
||
'role': self.role,
|
||
'status': self.status,
|
||
'created_at': self.created_at.isoformat() if self.created_at else None
|
||
}
|
||
|
||
|
||
# ==========================================
|
||
# 2. 系统日志表
|
||
# ==========================================
|
||
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
|
||
}
|
||
|
||
|
||
# ==========================================
|
||
# 3. 权限管理模型 (RBAC) - [新增]
|
||
# ==========================================
|
||
|
||
class SysMenu(db.Model):
|
||
"""系统菜单/页面表"""
|
||
__tablename__ = 'sys_menu'
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
parent_id = db.Column(db.Integer, default=0)
|
||
name = db.Column(db.String(50), nullable=False)
|
||
code = db.Column(db.String(100), unique=True, nullable=False)
|
||
path = db.Column(db.String(200))
|
||
sort_order = db.Column(db.Integer, default=0)
|
||
is_visible = db.Column(db.Boolean, default=True)
|
||
|
||
def to_dict(self):
|
||
return {
|
||
'id': self.id,
|
||
'name': self.name,
|
||
'code': self.code,
|
||
'path': self.path,
|
||
'type': 'menu' # 前端树形控件图标判断用
|
||
}
|
||
|
||
|
||
class SysElement(db.Model):
|
||
"""页面元素/列定义表"""
|
||
__tablename__ = 'sys_element'
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
menu_code = db.Column(db.String(100), db.ForeignKey('sys_menu.code'))
|
||
name = db.Column(db.String(100), nullable=False)
|
||
code = db.Column(db.String(100), nullable=False) # 如: unit_price
|
||
element_type = db.Column(db.String(20), default='column')
|
||
|
||
def to_dict(self):
|
||
return {
|
||
'id': self.id,
|
||
'name': self.name,
|
||
'code': self.code,
|
||
'menu_code': self.menu_code,
|
||
'type': 'element',
|
||
'element_type': self.element_type
|
||
}
|
||
|
||
|
||
class SysRolePermission(db.Model):
|
||
"""角色权限关联表"""
|
||
__tablename__ = 'sys_role_permission'
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
role_code = db.Column(db.String(50), nullable=False)
|
||
target_code = db.Column(db.String(100), nullable=False) # menu_code 或 element_code
|
||
type = db.Column(db.String(20), nullable=False) # 'menu' 或 'element'
|
||
|
||
|
||
# ==========================================
|
||
# 4. 库位管理模型
|
||
# ==========================================
|
||
class SysWarehouseLocation(db.Model):
|
||
"""
|
||
库位字典表（支持无限层级树形结构）
|
||
对应数据库: sys_warehouse_location
|
||
"""
|
||
__tablename__ = 'sys_warehouse_location'
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
parent_id = db.Column(db.Integer, db.ForeignKey('sys_warehouse_location.id'), nullable=True)
|
||
name = db.Column(db.String(100), nullable=False)
|
||
full_path = db.Column(db.String(500)) # 完整路径，如 "A区/货架1/第3层"
|
||
level = db.Column(db.Integer, default=0) # 层级深度，顶级为0
|
||
is_enabled = db.Column(db.Boolean, default=True)
|
||
created_at = db.Column(db.DateTime, default=datetime.now)
|
||
# 注意：数据库表中没有 updated_at 字段，不要添加！
|
||
|
||
# 自关联 - 使用 backref 定义父节点的反向引用
|
||
children = db.relationship(
|
||
'SysWarehouseLocation',
|
||
backref=db.backref('parent', remote_side=[id]),
|
||
lazy='dynamic'
|
||
)
|
||
|
||
def to_dict(self):
|
||
return {
|
||
'id': self.id,
|
||
'parent_id': self.parent_id,
|
||
'name': self.name,
|
||
'full_path': self.full_path,
|
||
'level': self.level,
|
||
'is_enabled': self.is_enabled,
|
||
'created_at': self.created_at.isoformat() if self.created_at else None
|
||
} |