b00e52e1e0
核心功能： - 集成博查Web Search API自动获取网站相关新闻 - 智能新闻更新机制（每日首次访问触发） - 精确新闻搜索（使用引号强制匹配网站名称） - News模型扩展（source_name, source_icon字段） - 网站详情页新闻展示模块 - 新闻来源网站信息展示 - 自动去重防止重复新闻 技术实现： - NewsSearcher工具类封装博查API - 数据库迁移脚本migrate_news_fields.py - 测试脚本test_news_feature.py - 定期任务脚本fetch_news_cron.py - API路由：/api/fetch-site-news, /api/fetch-all-news 配置优化： - 修复manage.sh路径和启动命令 - 博查API配置（BOCHA_API_KEY, BOCHA_BASE_URL） - 新闻搜索参数配置 界面优化： - 详情页新闻模块（左侧主栏） - 相似推荐模块（右侧边栏） - 首页标签图标修复 - 后台添加修改密码功能 - 登录页面优化 部署信息： - 部署日期: 2025-12-30 - 部署方式: 手动上传文件 - 数据库: 已迁移（添加source_name和source_icon字段）
100 lines
2.9 KiB
Python
100 lines
2.9 KiB
Python
"""
|
||
数据库迁移脚本 - 为News表添加source_name和source_icon字段
|
||
版本：v2.2.0
|
||
日期：2025-01-30
|
||
"""
|
||
import pymysql
|
||
import os
|
||
from dotenv import load_dotenv
|
||
|
||
# 加载环境变量
|
||
load_dotenv()
|
||
|
||
def migrate():
|
||
"""执行数据库迁移"""
|
||
# 数据库配置
|
||
db_config = {
|
||
'host': os.environ.get('DB_HOST', 'localhost'),
|
||
'port': int(os.environ.get('DB_PORT', 3306)),
|
||
'user': os.environ.get('DB_USER', 'root'),
|
||
'password': os.environ.get('DB_PASSWORD', ''),
|
||
'database': os.environ.get('DB_NAME', 'ai_nav'),
|
||
'charset': 'utf8mb4'
|
||
}
|
||
|
||
try:
|
||
# 连接数据库
|
||
connection = pymysql.connect(**db_config)
|
||
cursor = connection.cursor()
|
||
|
||
print("=" * 60)
|
||
print("开始执行数据库迁移 v2.2.0")
|
||
print("=" * 60)
|
||
|
||
# 检查字段是否已存在
|
||
cursor.execute("""
|
||
SELECT COLUMN_NAME
|
||
FROM INFORMATION_SCHEMA.COLUMNS
|
||
WHERE TABLE_SCHEMA = %s
|
||
AND TABLE_NAME = 'news'
|
||
AND COLUMN_NAME IN ('source_name', 'source_icon')
|
||
""", (db_config['database'],))
|
||
|
||
existing_columns = [row[0] for row in cursor.fetchall()]
|
||
|
||
# 添加 source_name 字段
|
||
if 'source_name' not in existing_columns:
|
||
print("\n1. 添加 source_name 字段...")
|
||
cursor.execute("""
|
||
ALTER TABLE news
|
||
ADD COLUMN source_name VARCHAR(100)
|
||
COMMENT '新闻来源网站名称'
|
||
AFTER url
|
||
""")
|
||
print(">>> source_name 字段添加成功")
|
||
else:
|
||
print("\n1. source_name 字段已存在，跳过")
|
||
|
||
# 添加 source_icon 字段
|
||
if 'source_icon' not in existing_columns:
|
||
print("\n2. 添加 source_icon 字段...")
|
||
cursor.execute("""
|
||
ALTER TABLE news
|
||
ADD COLUMN source_icon VARCHAR(500)
|
||
COMMENT '新闻来源网站图标URL'
|
||
AFTER source_name
|
||
""")
|
||
print(">>> source_icon 字段添加成功")
|
||
else:
|
||
print("\n2. source_icon 字段已存在，跳过")
|
||
|
||
# 提交事务
|
||
connection.commit()
|
||
|
||
print("\n" + "=" * 60)
|
||
print(">>> 数据库迁移完成！")
|
||
print("=" * 60)
|
||
|
||
# 显示表结构
|
||
print("\n当前 news 表结构：")
|
||
cursor.execute("DESCRIBE news")
|
||
for row in cursor.fetchall():
|
||
print(f" - {row[0]}: {row[1]} {row[2]}")
|
||
|
||
except Exception as e:
|
||
print(f"\n>>> 迁移失败：{str(e)}")
|
||
if 'connection' in locals():
|
||
connection.rollback()
|
||
raise
|
||
|
||
finally:
|
||
if 'cursor' in locals():
|
||
cursor.close()
|
||
if 'connection' in locals():
|
||
connection.close()
|
||
print("\n数据库连接已关闭")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
migrate()
|