24024
d4b28a731a
- 新增 SystemNotification 模型，实现系统通知的数据存储 - 管理后台新增通知相关接口，支持通知的增删改查 - 用户端新增接口，获取最新激活通知并支持标记已读 - 在前端首页添加全局通知弹窗，实现通知自动轮询及已读同步 - 生成历史记录中兼容支持图片缩略图及新旧图片格式 - 优化后台图片同步逻辑，新增缩略图生成与存储 - 支持上传参考图的拖拽、粘贴、多文件上传及排序功能 - 增加购买积分页面入口及菜单项，调整菜单结构 - 日志系统由 Redis 列表迁移为有序集合，保留 30 天日志 - 优化日志页面样式，提升可读性及滚动体验 - 调整部分模板布局为自定义滚动条容器，增强视觉一致性
104 lines
4.4 KiB
Python
104 lines
4.4 KiB
Python
from extensions import db
|
||
from datetime import datetime
|
||
from werkzeug.security import generate_password_hash, check_password_hash
|
||
|
||
# 角色与权限的多对多关联表
|
||
role_permissions = db.Table('role_permissions',
|
||
db.Column('role_id', db.Integer, db.ForeignKey('roles.id'), primary_key=True),
|
||
db.Column('permission_id', db.Integer, db.ForeignKey('permissions.id'), primary_key=True)
|
||
)
|
||
|
||
class Permission(db.Model):
|
||
__tablename__ = 'permissions'
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
name = db.Column(db.String(50), unique=True, nullable=False) # 如: 'view_logs', 'manage_users'
|
||
description = db.Column(db.String(100))
|
||
|
||
class Role(db.Model):
|
||
__tablename__ = 'roles'
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
name = db.Column(db.String(50), unique=True, nullable=False) # 如: '超级管理员', '普通用户'
|
||
description = db.Column(db.String(100))
|
||
# 角色拥有的权限
|
||
permissions = db.relationship('Permission', secondary=role_permissions, backref=db.backref('roles', lazy='dynamic'))
|
||
|
||
class User(db.Model):
|
||
__tablename__ = 'users'
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
phone = db.Column(db.String(20), unique=True, nullable=False)
|
||
password_hash = db.Column(db.String(255), nullable=False)
|
||
api_key = db.Column(db.String(255)) # 存储用户的 API Key
|
||
points = db.Column(db.Integer, default=2) # 账户积分，默认赠送2次试用
|
||
# 关联角色 ID
|
||
role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))
|
||
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
||
|
||
# 关系映射
|
||
role = db.relationship('Role', backref=db.backref('users', lazy='dynamic'))
|
||
|
||
def has_permission(self, perm_name):
|
||
"""动态检查用户是否拥有某项权限"""
|
||
if not self.role:
|
||
return False
|
||
|
||
# 获取用户拥有的所有权限名称
|
||
perms = [p.name for p in self.role.permissions]
|
||
|
||
# 核心修复：如果是超级管理员（拥有 manage_system），则豁免所有具体权限检查
|
||
if 'manage_system' in perms:
|
||
return True
|
||
|
||
return perm_name in perms
|
||
|
||
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)
|
||
|
||
class SystemDict(db.Model):
|
||
"""通用字典管理系统"""
|
||
__tablename__ = 'system_dicts'
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
dict_type = db.Column(db.String(50), nullable=False) # 如: 'ai_model', 'aspect_ratio', 'prompt_tpl'
|
||
label = db.Column(db.String(100), nullable=False) # 显示名称
|
||
value = db.Column(db.Text, nullable=False) # 存储值或提示词内容
|
||
cost = db.Column(db.Integer, default=0) # 消耗积分 (仅针对 ai_model 有效)
|
||
is_active = db.Column(db.Boolean, default=True)
|
||
sort_order = db.Column(db.Integer, default=0) # 排序权重
|
||
|
||
class GenerationRecord(db.Model):
|
||
"""AI 生成记录"""
|
||
__tablename__ = 'generation_records'
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
|
||
prompt = db.Column(db.Text)
|
||
model = db.Column(db.String(100))
|
||
# 存储生成的图片 URL 列表 (JSON 字符串)
|
||
image_urls = db.Column(db.Text)
|
||
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
||
|
||
user = db.relationship('User', backref=db.backref('records', lazy='dynamic', order_by='GenerationRecord.created_at.desc()'))
|
||
|
||
# 用户已读通知关联表
|
||
notification_reads = db.Table('notification_reads',
|
||
db.Column('user_id', db.Integer, db.ForeignKey('users.id'), primary_key=True),
|
||
db.Column('notification_id', db.Integer, db.ForeignKey('system_notifications.id'), primary_key=True)
|
||
)
|
||
|
||
class SystemNotification(db.Model):
|
||
"""系统全局通知"""
|
||
__tablename__ = 'system_notifications'
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
title = db.Column(db.String(200), nullable=False)
|
||
content = db.Column(db.Text, nullable=False)
|
||
is_active = db.Column(db.Boolean, default=True)
|
||
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
||
|
||
# 哪些用户已读
|
||
read_by_users = db.relationship('User', secondary=notification_reads, backref=db.backref('read_notifications', lazy='dynamic'))
|