以插件方式新增 Cookie 管理模块,不改动 framework 业务逻辑。
数据层
- CookieUser / CookieSite 两个模型(applications/models/admin_cookie.py)
- 对应 ManageSchema(applications/schemas/admin_cookie.py)
- 模型与 schema 在 applications/{models,schemas}/__init__.py 中注册,便于 Alembic 跟踪
- 新增两个迁移脚本(cookie_site / cookie_user + cookie 站点表)
插件
- plugins/cookieManager/:入口 __init__.py + 三个视图(概览 / 账号 / 站点)
- 后台模板:cookie_user 与 cookie_site 各自的 main/add/edit
- 路由前缀 /system/cookie-user/*、/system/cookie-site/*
74 lines
3.6 KiB
Python
74 lines
3.6 KiB
Python
"""Cookie管理数据模型"""
|
|
import datetime
|
|
from applications.extensions import db
|
|
|
|
|
|
class CookieUser(db.Model):
|
|
"""登录用户 - 保存已登录用户的Cookie信息"""
|
|
__tablename__ = 'admin_cookie_user'
|
|
|
|
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
|
|
username = db.Column(db.String(100), nullable=False, comment='用户名')
|
|
nickname = db.Column(db.String(100), default='', comment='昵称')
|
|
website_url = db.Column(db.String(255), nullable=False, comment='网站URL')
|
|
website_name = db.Column(db.String(100), default='', comment='网站名称')
|
|
cookie_value = db.Column(db.Text, nullable=False, comment='Cookie值')
|
|
domain = db.Column(db.String(255), default='', comment='Cookie域名')
|
|
expiry = db.Column(db.DateTime, comment='过期时间')
|
|
enable = db.Column(db.Integer, default=1, comment='状态(1启用,0禁用)')
|
|
create_by = db.Column(db.String(50), default='admin', comment='创建者')
|
|
create_at = db.Column(db.DateTime, default=datetime.datetime.now, comment='创建时间')
|
|
update_at = db.Column(db.DateTime, default=datetime.datetime.now, onupdate=datetime.datetime.now, comment='更新时间')
|
|
|
|
def to_dict(self):
|
|
return {
|
|
'id': self.id,
|
|
'username': self.username,
|
|
'nickname': self.nickname,
|
|
'website_url': self.website_url,
|
|
'website_name': self.website_name,
|
|
'cookie_value': self.cookie_value,
|
|
'domain': self.domain,
|
|
'expiry': self.expiry.strftime('%Y-%m-%d %H:%M:%S') if self.expiry else '',
|
|
'enable': self.enable,
|
|
'create_by': self.create_by,
|
|
'create_at': self.create_at.strftime('%Y-%m-%d %H:%M:%S') if self.create_at else '',
|
|
'update_at': self.update_at.strftime('%Y-%m-%d %H:%M:%S') if self.update_at else '',
|
|
}
|
|
|
|
|
|
class CookieSite(db.Model):
|
|
"""登录网站 - 保存网站的Cookie信息"""
|
|
__tablename__ = 'admin_cookie_site'
|
|
|
|
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
|
|
name = db.Column(db.String(100), nullable=False, comment='网站名称')
|
|
url = db.Column(db.String(255), nullable=False, comment='网站URL')
|
|
cookie_value = db.Column(db.Text, nullable=False, comment='Cookie值')
|
|
domain = db.Column(db.String(255), default='', comment='Cookie域名')
|
|
path = db.Column(db.String(255), default='/', comment='Cookie路径')
|
|
expiry = db.Column(db.DateTime, comment='过期时间')
|
|
secure = db.Column(db.Integer, default=0, comment='是否安全')
|
|
http_only = db.Column(db.Integer, default=0, comment='是否HTTPOnly')
|
|
enable = db.Column(db.Integer, default=1, comment='状态(1启用,0禁用)')
|
|
create_by = db.Column(db.String(50), default='admin', comment='创建者')
|
|
create_at = db.Column(db.DateTime, default=datetime.datetime.now, comment='创建时间')
|
|
update_at = db.Column(db.DateTime, default=datetime.datetime.now, onupdate=datetime.datetime.now, comment='更新时间')
|
|
|
|
def to_dict(self):
|
|
return {
|
|
'id': self.id,
|
|
'name': self.name,
|
|
'url': self.url,
|
|
'cookie_value': self.cookie_value,
|
|
'domain': self.domain,
|
|
'path': self.path,
|
|
'expiry': self.expiry.strftime('%Y-%m-%d %H:%M:%S') if self.expiry else '',
|
|
'secure': self.secure,
|
|
'http_only': self.http_only,
|
|
'enable': self.enable,
|
|
'create_by': self.create_by,
|
|
'create_at': self.create_at.strftime('%Y-%m-%d %H:%M:%S') if self.create_at else '',
|
|
'update_at': self.update_at.strftime('%Y-%m-%d %H:%M:%S') if self.update_at else '',
|
|
}
|