first code

utils/rss_generator.py

@@ -0,0 +1,40 @@
+from datetime import datetime
+from typing import Dict, Any, List
+import xml.etree.ElementTree as ET
+
+class RSSGenerator:
+    """RSS XML生成器"""
+    
+    def __init__(self, title: str, description: str, link: str, language: str = "zh-cn"):
+        self.title = title
+        self.description = description
+        self.link = link
+        self.language = language
+    
+    def generate_rss(self, items: List[Dict[str, Any]]) -> str:
+        """生成RSS XML"""
+        # 创建根元素
+        rss = ET.Element("rss", version="2.0")
+        channel = ET.SubElement(rss, "channel")
+        
+        # 频道信息
+        ET.SubElement(channel, "title").text = self.title
+        ET.SubElement(channel, "description").text = self.description
+        ET.SubElement(channel, "link").text = self.link
+        ET.SubElement(channel, "language").text = self.language
+        ET.SubElement(channel, "lastBuildDate").text = datetime.now().strftime("%a, %d %b %Y %H:%M:%S +0800")
+        
+        # 添加文章项目
+        for item_data in items:
+            item = ET.SubElement(channel, "item")
+            ET.SubElement(item, "title").text = item_data.get("title", "")
+            ET.SubElement(item, "description").text = item_data.get("description", "")
+            ET.SubElement(item, "link").text = item_data.get("link", self.link)
+            ET.SubElement(item, "pubDate").text = item_data.get("pub_date", datetime.now().strftime("%a, %d %b %Y %H:%M:%S +0800"))
+            
+            if "guid" in item_data:
+                ET.SubElement(item, "guid").text = item_data["guid"]
+        
+        # 转换为字符串
+        xml_str = ET.tostring(rss, encoding='unicode', method='xml')
+        return f'<?xml version="1.0" encoding="UTF-8"?>\n{xml_str}'