8011e5bd4a
- 移除顶部热门工具排行榜模块 - 在标签下方添加三个tab（最新/热门/推荐） - 添加is_recommended字段到Site模型 - 创建数据库迁移脚本add_is_recommended.py - 更新后台管理界面支持推荐标记 - 更新分页链接保持tab状态 - 所有功能已本地测试验证通过 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
185 lines
8.1 KiB
Python
185 lines
8.1 KiB
Python
from datetime import datetime
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from flask_login import UserMixin
|
|
from werkzeug.security import generate_password_hash, check_password_hash
|
|
|
|
db = SQLAlchemy()
|
|
|
|
# 网站和标签的多对多关系表
|
|
site_tags = db.Table('site_tags',
|
|
db.Column('site_id', db.Integer, db.ForeignKey('sites.id'), primary_key=True),
|
|
db.Column('tag_id', db.Integer, db.ForeignKey('tags.id'), primary_key=True)
|
|
)
|
|
|
|
class Site(db.Model):
|
|
"""网站模型"""
|
|
__tablename__ = 'sites'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
code = db.Column(db.String(8), unique=True, nullable=False, comment='8位数字编码')
|
|
name = db.Column(db.String(100), nullable=False, comment='网站名称')
|
|
url = db.Column(db.String(500), nullable=False, comment='网站URL')
|
|
slug = db.Column(db.String(100), unique=True, nullable=True, comment='URL别名(SEO用)')
|
|
logo = db.Column(db.String(500), comment='Logo图片路径')
|
|
short_desc = db.Column(db.String(200), comment='简短描述')
|
|
description = db.Column(db.Text, comment='详细介绍')
|
|
features = db.Column(db.Text, comment='主要功能')
|
|
news_keywords = db.Column(db.String(200), comment='新闻获取关键词（用于精准匹配相关新闻）')
|
|
is_active = db.Column(db.Boolean, default=True, comment='是否启用')
|
|
is_recommended = db.Column(db.Boolean, default=False, nullable=False, comment='是否推荐')
|
|
view_count = db.Column(db.Integer, default=0, comment='浏览次数')
|
|
sort_order = db.Column(db.Integer, default=0, comment='排序权重')
|
|
created_at = db.Column(db.DateTime, default=datetime.now, comment='创建时间')
|
|
updated_at = db.Column(db.DateTime, default=datetime.now, onupdate=datetime.now, comment='更新时间')
|
|
|
|
# 关联标签
|
|
tags = db.relationship('Tag', secondary=site_tags, lazy='subquery',
|
|
backref=db.backref('sites', lazy=True))
|
|
|
|
def __repr__(self):
|
|
return f'<Site {self.name}>'
|
|
|
|
def to_dict(self):
|
|
"""转换为字典"""
|
|
return {
|
|
'id': self.id,
|
|
'code': self.code,
|
|
'name': self.name,
|
|
'url': self.url,
|
|
'slug': self.slug,
|
|
'logo': self.logo,
|
|
'short_desc': self.short_desc,
|
|
'description': self.description,
|
|
'features': self.features,
|
|
'news_keywords': self.news_keywords,
|
|
'is_active': self.is_active,
|
|
'is_recommended': self.is_recommended,
|
|
'view_count': self.view_count,
|
|
'tags': [tag.name for tag in self.tags],
|
|
'created_at': self.created_at.strftime('%Y-%m-%d %H:%M:%S') if self.created_at else None
|
|
}
|
|
|
|
class Tag(db.Model):
|
|
"""标签模型"""
|
|
__tablename__ = 'tags'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(50), unique=True, nullable=False, comment='标签名称')
|
|
slug = db.Column(db.String(50), unique=True, nullable=False, comment='URL别名')
|
|
description = db.Column(db.String(200), comment='标签描述')
|
|
seo_title = db.Column(db.String(100), comment='SEO标题(v2.4新增)')
|
|
seo_description = db.Column(db.String(300), comment='SEO页面描述(v2.4新增)')
|
|
seo_keywords = db.Column(db.String(200), comment='SEO关键词(v2.4新增)')
|
|
icon = db.Column(db.String(100), comment='图标')
|
|
sort_order = db.Column(db.Integer, default=0, comment='排序权重')
|
|
created_at = db.Column(db.DateTime, default=datetime.now, comment='创建时间')
|
|
|
|
def __repr__(self):
|
|
return f'<Tag {self.name}>'
|
|
|
|
def to_dict(self):
|
|
"""转换为字典"""
|
|
return {
|
|
'id': self.id,
|
|
'name': self.name,
|
|
'slug': self.slug,
|
|
'description': self.description,
|
|
'seo_title': self.seo_title,
|
|
'seo_description': self.seo_description,
|
|
'seo_keywords': self.seo_keywords,
|
|
'icon': self.icon
|
|
}
|
|
|
|
class News(db.Model):
|
|
"""新闻模型"""
|
|
__tablename__ = 'news'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
site_id = db.Column(db.Integer, db.ForeignKey('sites.id'), nullable=False, comment='关联网站ID')
|
|
title = db.Column(db.String(200), nullable=False, comment='新闻标题')
|
|
content = db.Column(db.Text, comment='新闻内容')
|
|
news_type = db.Column(db.String(50), default='Industry News', comment='新闻类型')
|
|
url = db.Column(db.String(500), comment='新闻链接')
|
|
source_name = db.Column(db.String(100), comment='新闻来源网站名称')
|
|
source_icon = db.Column(db.String(500), comment='新闻来源网站图标URL')
|
|
published_at = db.Column(db.DateTime, default=datetime.now, comment='发布时间')
|
|
is_active = db.Column(db.Boolean, default=True, comment='是否启用')
|
|
created_at = db.Column(db.DateTime, default=datetime.now, comment='创建时间')
|
|
updated_at = db.Column(db.DateTime, default=datetime.now, onupdate=datetime.now, comment='更新时间')
|
|
|
|
# 关联网站
|
|
site = db.relationship('Site', backref=db.backref('news', lazy='dynamic', order_by='News.published_at.desc()'))
|
|
|
|
def __repr__(self):
|
|
return f'<News {self.title}>'
|
|
|
|
def to_dict(self):
|
|
"""转换为字典"""
|
|
return {
|
|
'id': self.id,
|
|
'site_id': self.site_id,
|
|
'title': self.title,
|
|
'content': self.content,
|
|
'news_type': self.news_type,
|
|
'url': self.url,
|
|
'source_name': self.source_name,
|
|
'source_icon': self.source_icon,
|
|
'published_at': self.published_at.strftime('%Y-%m-%d') if self.published_at else None,
|
|
'created_at': self.created_at.strftime('%Y-%m-%d %H:%M:%S') if self.created_at else None
|
|
}
|
|
|
|
class Admin(UserMixin, db.Model):
|
|
"""管理员模型"""
|
|
__tablename__ = 'admins'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
username = db.Column(db.String(50), unique=True, nullable=False, comment='用户名')
|
|
password_hash = db.Column(db.String(255), nullable=False, comment='密码哈希')
|
|
email = db.Column(db.String(100), comment='邮箱')
|
|
is_active = db.Column(db.Boolean, default=True, comment='是否启用')
|
|
created_at = db.Column(db.DateTime, default=datetime.now, comment='创建时间')
|
|
last_login = db.Column(db.DateTime, comment='最后登录时间')
|
|
|
|
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 __repr__(self):
|
|
return f'<Admin {self.username}>'
|
|
|
|
class PromptTemplate(db.Model):
|
|
"""AI提示词模板模型"""
|
|
__tablename__ = 'prompt_templates'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
key = db.Column(db.String(50), unique=True, nullable=False, comment='唯一标识(tags/features/description)')
|
|
name = db.Column(db.String(100), nullable=False, comment='模板名称')
|
|
system_prompt = db.Column(db.Text, nullable=False, comment='系统提示词')
|
|
user_prompt_template = db.Column(db.Text, nullable=False, comment='用户提示词模板(支持变量)')
|
|
description = db.Column(db.String(200), comment='模板说明')
|
|
is_active = db.Column(db.Boolean, default=True, comment='是否启用')
|
|
created_at = db.Column(db.DateTime, default=datetime.now, comment='创建时间')
|
|
updated_at = db.Column(db.DateTime, default=datetime.now, onupdate=datetime.now, comment='更新时间')
|
|
|
|
def __repr__(self):
|
|
return f'<PromptTemplate {self.name}>'
|
|
|
|
def to_dict(self):
|
|
"""转换为字典"""
|
|
return {
|
|
'id': self.id,
|
|
'key': self.key,
|
|
'name': self.name,
|
|
'system_prompt': self.system_prompt,
|
|
'user_prompt_template': self.user_prompt_template,
|
|
'description': self.description,
|
|
'is_active': self.is_active,
|
|
'created_at': self.created_at.strftime('%Y-%m-%d %H:%M:%S') if self.created_at else None,
|
|
'updated_at': self.updated_at.strftime('%Y-%m-%d %H:%M:%S') if self.updated_at else None
|
|
}
|
|
|