39 lines
1.9 KiB
Python
39 lines
1.9 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
from datetime import datetime
|
|||
|
|
from src.flask_prompt_master import db
|
|||
|
|
|
|||
|
|
class Favorite(db.Model):
|
|||
|
|
"""收藏表"""
|
|||
|
|
__tablename__ = 'favorites'
|
|||
|
|
|
|||
|
|
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
|
|||
|
|
user_id = db.Column(db.String(50), nullable=False, comment='用户ID（可以是IP或session_id）')
|
|||
|
|
template_id = db.Column(db.Integer, nullable=True, comment='模板ID')
|
|||
|
|
original_text = db.Column(db.Text, nullable=False, comment='原始输入文本')
|
|||
|
|
generated_prompt = db.Column(db.Text, nullable=False, comment='生成的提示词')
|
|||
|
|
system_prompt = db.Column(db.Text, nullable=True, comment='系统提示词')
|
|||
|
|
category = db.Column(db.String(50), nullable=True, comment='分类')
|
|||
|
|
industry = db.Column(db.String(50), nullable=True, comment='行业')
|
|||
|
|
profession = db.Column(db.String(50), nullable=True, comment='职业')
|
|||
|
|
tags = db.Column(db.String(500), nullable=True, comment='标签（JSON格式）')
|
|||
|
|
notes = db.Column(db.Text, nullable=True, comment='用户备注')
|
|||
|
|
created_time = db.Column(db.DateTime, default=datetime.now, comment='创建时间')
|
|||
|
|
updated_time = db.Column(db.DateTime, default=datetime.now, onupdate=datetime.now, comment='更新时间')
|
|||
|
|
|
|||
|
|
def to_dict(self):
|
|||
|
|
"""转换为字典"""
|
|||
|
|
return {
|
|||
|
|
'id': self.id,
|
|||
|
|
'template_id': self.template_id,
|
|||
|
|
'original_text': self.original_text,
|
|||
|
|
'generated_prompt': self.generated_prompt,
|
|||
|
|
'system_prompt': self.system_prompt,
|
|||
|
|
'category': self.category,
|
|||
|
|
'industry': self.industry,
|
|||
|
|
'profession': self.profession,
|
|||
|
|
'tags': self.tags,
|
|||
|
|
'notes': self.notes,
|
|||
|
|
'created_time': self.created_time.strftime('%Y-%m-%d %H:%M:%S') if self.created_time else None,
|
|||
|
|
'updated_time': self.updated_time.strftime('%Y-%m-%d %H:%M:%S') if self.updated_time else None
|
|||
|
|
}
|