init

app/db/mysql.py

@@ -0,0 +1,60 @@
+from datetime import datetime
+from typing import Optional, List, Dict, Any
+from .models import Base, News
+
+# 移除对 SQLAlchemy 的依赖
+# from app.core.db import Base
+
+# 定义一个简单的数据类来替代 SQLAlchemy 模型
+class News:
+    """新闻数据模型"""
+    
+    def __init__(self, 
+                 title: str = "", 
+                 content: str = "", 
+                 url: str = "", 
+                 source: str = "", 
+                 publish_time: Optional[datetime] = None):
+        self.id: Optional[int] = None
+        self.title = title
+        self.content = content
+        self.url = url
+        self.source = source
+        self.publish_time = publish_time or datetime.now()
+        self.created_at = datetime.now()
+    
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> 'News':
+        """从字典创建新闻对象"""
+        news = cls(
+            title=data.get('title', ''),
+            content=data.get('content', ''),
+            url=data.get('url', ''),
+            source=data.get('source', ''),
+            publish_time=data.get('publish_time')
+        )
+        if 'id' in data:
+            news.id = data['id']
+        if 'created_at' in data:
+            news.created_at = data['created_at']
+        return news
+    
+    def to_dict(self) -> Dict[str, Any]:
+        """转换为字典"""
+        return {
+            'id': self.id,
+            'title': self.title,
+            'content': self.content,
+            'url': self.url,
+            'source': self.source,
+            'publish_time': self.publish_time,
+            'created_at': self.created_at
+        }
+
+def insert_news(news_list):
+    """将新闻列表插入数据库"""
+    from app.core import db
+    # 如果传入的是 News 对象列表，转换为字典列表
+    if news_list and isinstance(news_list[0], News):
+        news_list = [news.to_dict() for news in news_list]
+    return db.insert_news(news_list)