戒酒的李白
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
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')
|
|
} |