Modify naming conventions that may cause confusion and consolidate code files.

2025-02-13 15:56:34 +08:00
parent 9106d8ed3d
commit dfe1a833f9
2 changed files with 1 additions and 1 deletions
@@ -0,0 +1,31 @@
+from sqlalchemy import Column, Integer, String, Float, DateTime, Text, JSON
+from sqlalchemy.ext.declarative import declarative_base
+from datetime import datetime
+
+Base = declarative_base()
+
+class AIAnalysis(Base):
+    __tablename__ = 'ai_analysis'
+    
+    id = Column(Integer, primary_key=True)
+    message_id = Column(Integer, nullable=False)
+    sentiment = Column(String(10), nullable=False)
+    sentiment_score = Column(Float, nullable=False)
+    keywords = Column(JSON, nullable=False)
+    key_points = Column(Text, nullable=False)
+    influence_analysis = Column(Text, nullable=False)
+    risk_level = Column(String(10), nullable=False)
+    created_at = Column(DateTime, default=datetime.now)
+    
+    def to_dict(self):
+        return {
+            'id': self.id,
+            'message_id': self.message_id,
+            'sentiment': self.sentiment,
+            'sentiment_score': f"{self.sentiment_score:.2%}",
+            'keywords': self.keywords,
+            'key_points': self.key_points,
+            'influence': self.influence_analysis,
+            'risk_level': self.risk_level,
+            'analysis_time': self.created_at.strftime('%Y-%m-%d %H:%M:%S')
+        }