352 lines
17 KiB
Python
352 lines
17 KiB
Python
|
from django.db import models
|
|||
|
|
|
|||
|
|
# Create your models here.
|
|||
|
|
from datetime import datetime
|
|||
|
|
from django.db import models
|
|||
|
|
from django.contrib.auth.hashers import make_password,is_password_usable
|
|||
|
|
from django.utils import timezone
|
|||
|
|
|
|||
|
|
# 管理员模型
|
|||
|
|
class Administrator(models.Model):
|
|||
|
|
username = models.CharField(max_length=150, unique=True) # 用户名，确保唯一性
|
|||
|
|
password = models.CharField(max_length=128) # 密码，将会加密存储
|
|||
|
|
role = models.CharField(max_length=100) # 角色
|
|||
|
|
is_active = models.BooleanField(default=True) # 状态，启用/禁用，默认启用
|
|||
|
|
last_login = models.DateTimeField(null=True, blank=True)
|
|||
|
|
created_at = models.DateTimeField(auto_now_add=True) # 创建时间，自动设置为当前时间
|
|||
|
|
# 保存模型之前对密码进行加密
|
|||
|
|
def save(self, *args, **kwargs):
|
|||
|
|
if not is_password_usable(self.password):
|
|||
|
|
self.password = make_password(self.password)
|
|||
|
|
super().save(*args, **kwargs)
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "Administrator"
|
|||
|
|
|
|||
|
|
from django.db import models
|
|||
|
|
|
|||
|
|
class MembershipType(models.Model):
|
|||
|
|
type = models.CharField(max_length=50, unique=True, verbose_name="卡类型") # 卡类型，如天卡、周卡等 coins 额度卡
|
|||
|
|
title = models.CharField(max_length=100, verbose_name="标题") # 卡的标题
|
|||
|
|
description = models.TextField(verbose_name="描述") # 卡的描述
|
|||
|
|
price = models.DecimalField(max_digits=10, decimal_places=2, verbose_name="价格") # 卡的价格
|
|||
|
|
duration_days = models.IntegerField(verbose_name="有效天数") # 卡的有效期，以天为单位
|
|||
|
|
coins = models.IntegerField(default=0, verbose_name="赠送金币数量") # 赠送金币的数量
|
|||
|
|
is_quota = models.BooleanField(default=False, verbose_name="是否为额度充值") # 是否为额度充值
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "membership_type" # 设置数据库表名
|
|||
|
|
verbose_name = "会员卡类型"
|
|||
|
|
verbose_name_plural = "会员卡类型"
|
|||
|
|
|
|||
|
|
def __str__(self):
|
|||
|
|
return self.title
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 用户模型
|
|||
|
|
class User(models.Model):
|
|||
|
|
openid = models.CharField(max_length=150, unique=True) # 用户唯一标识
|
|||
|
|
wxid = models.CharField(max_length=150, blank=True, null=True) # 微信ID
|
|||
|
|
wechat_number = models.CharField(max_length=150, blank=True, null=True, verbose_name='微信号') # 微信号
|
|||
|
|
nickname = models.CharField(max_length=150, blank=True, null=True) # 用户标识
|
|||
|
|
gender = models.CharField(max_length=10, blank=True, null=True) # 性别
|
|||
|
|
region = models.CharField(max_length=150, blank=True, null=True) # 地区
|
|||
|
|
email = models.EmailField(blank=True, null=True) # 邮箱
|
|||
|
|
phone = models.CharField(max_length=20, blank=True, null=True) # 号码
|
|||
|
|
password = models.CharField(max_length=128, blank=True, null=True) # 密码
|
|||
|
|
scene = models.CharField(max_length=100, blank=True, null=True) # 用户来源
|
|||
|
|
is_member = models.BooleanField(default=False) # 是否为会员
|
|||
|
|
member_start_time = models.BigIntegerField(blank=True, null=True) # 存储会员开始时间的时间戳
|
|||
|
|
member_end_time = models.BigIntegerField(blank=True, null=True) # 存储会员到期时间的时间戳
|
|||
|
|
is_active = models.BooleanField(default=True) # 状态，启用/禁用，默认启用
|
|||
|
|
usage_count = models.IntegerField(default=0, blank=True, null=True) # 用户使用次数
|
|||
|
|
coins = models.IntegerField(default=3, verbose_name='金币数量') # 用户金币数量
|
|||
|
|
balance = models.DecimalField(max_digits=10, decimal_places=2, default=0.00, verbose_name='用户余额') # 用户余额
|
|||
|
|
inviter_nickname = models.CharField(max_length=150, blank=True, null=True, verbose_name='邀请人昵称') # 用户邀请人昵称
|
|||
|
|
invitees_count = models.IntegerField(default=0, verbose_name='邀请人数量') # 邀请人数量
|
|||
|
|
created_at = models.DateTimeField(auto_now_add=True) # 用户创建时间
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "User"
|
|||
|
|
|
|||
|
|
#添加好友表
|
|||
|
|
class FriendRequest(models.Model):
|
|||
|
|
fromusername = models.CharField(max_length=255, verbose_name='微信ID') # 发起请求的微信ID
|
|||
|
|
alias = models.CharField(max_length=255, blank=True, null=True, verbose_name='微信号') # 微信号
|
|||
|
|
fromnickname = models.CharField(max_length=255, blank=True, null=True, verbose_name='昵称') # 昵称
|
|||
|
|
country = models.CharField(max_length=255, blank=True, null=True, verbose_name='国家') # 国家
|
|||
|
|
province = models.CharField(max_length=255, blank=True, null=True, verbose_name='省份') # 省份
|
|||
|
|
city = models.CharField(max_length=255, blank=True, null=True, verbose_name='城市') # 城市
|
|||
|
|
sex = models.CharField(max_length=1, blank=True, null=True, verbose_name='性别') # 性别
|
|||
|
|
scene = models.CharField(max_length=255, blank=True, null=True, verbose_name='来源') # 来源
|
|||
|
|
time = models.DateTimeField(verbose_name='申请时间') # 申请时间
|
|||
|
|
towxid = models.CharField(max_length=255, verbose_name='目标微信ID') # 接收好友请求的微信ID
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "FriendRequest"
|
|||
|
|
verbose_name = '添加好友请求'
|
|||
|
|
verbose_name_plural = '添加好友请求'
|
|||
|
|
|
|||
|
|
def __str__(self):
|
|||
|
|
return self.fromnickname or self.fromusername
|
|||
|
|
|
|||
|
|
#文案库
|
|||
|
|
from django.db import models
|
|||
|
|
|
|||
|
|
class Copywriting(models.Model):
|
|||
|
|
openid = models.CharField(max_length=150, blank=True, null=True, verbose_name="openid")
|
|||
|
|
nickname = models.CharField(max_length=255, blank=True, null=True, verbose_name="用户标识")
|
|||
|
|
text_content = models.TextField(verbose_name="文本内容")
|
|||
|
|
source = models.CharField(max_length=255, verbose_name="来源")
|
|||
|
|
tag = models.CharField(max_length=255, default=None, verbose_name="标签")
|
|||
|
|
popularity = models.IntegerField(default=0, verbose_name="热度")
|
|||
|
|
added_time = models.DateTimeField(auto_now_add=True, verbose_name="添加时间")
|
|||
|
|
is_approved = models.BooleanField(default=False, verbose_name="审核通过") # 默认未审核
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "Copywriting"
|
|||
|
|
verbose_name = "文案库"
|
|||
|
|
verbose_name_plural = "文案库"
|
|||
|
|
|
|||
|
|
def __str__(self):
|
|||
|
|
return self.text_content[:50] # 返回文案的前50个字符作为字符串表示
|
|||
|
|
|
|||
|
|
|
|||
|
|
#视频提取记录
|
|||
|
|
class VideoExtractionRecord(models.Model):
|
|||
|
|
extraction_link = models.URLField(verbose_name="提取链接", max_length=2048)
|
|||
|
|
source = models.CharField(max_length=100, choices=[('index', 'Index'), ('weixin', 'Weixin')], verbose_name="来源")
|
|||
|
|
openid = models.CharField(max_length=150, blank=True, null=True, verbose_name="openid")
|
|||
|
|
nickname = models.CharField(max_length=255, blank=True, null=True, verbose_name="用户标识")
|
|||
|
|
video_title = models.TextField(blank=True, null=True, verbose_name="视频标题")
|
|||
|
|
wxid = models.CharField(max_length=255, blank=True, null=True, verbose_name="微信id")
|
|||
|
|
wechat_alias = models.CharField(max_length=255, blank=True, null=True, verbose_name="微信号")
|
|||
|
|
is_successful = models.CharField(max_length=255,blank=True, null=True, verbose_name="是否提取成功")
|
|||
|
|
exception_reason = models.TextField(blank=True, null=True, verbose_name="异常原因")
|
|||
|
|
extraction_time = models.DateTimeField(auto_now_add=True, verbose_name="提取时间")
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "VideoExtractionRecord"
|
|||
|
|
verbose_name = "视频提取记录"
|
|||
|
|
verbose_name_plural = "视频提取记录"
|
|||
|
|
|
|||
|
|
def __str__(self):
|
|||
|
|
return f"{self.user}'s extraction record"
|
|||
|
|
|
|||
|
|
#文案提取任务
|
|||
|
|
class TranscriptionTask(models.Model):
|
|||
|
|
uuid = models.CharField(max_length=255)
|
|||
|
|
video_url = models.TextField()
|
|||
|
|
task_id = models.CharField(max_length=255)
|
|||
|
|
result = models.TextField(null=True, blank=True)
|
|||
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|||
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "TranscriptionTask"
|
|||
|
|
verbose_name = "文案提取任务"
|
|||
|
|
verbose_name_plural = "文案提取任务"
|
|||
|
|
def __str__(self):
|
|||
|
|
return self.task_id
|
|||
|
|
|
|||
|
|
#api调用日志
|
|||
|
|
class ApiCallLog(models.Model):
|
|||
|
|
# 用户来源，如'index', 'weixin'等
|
|||
|
|
source = models.CharField(max_length=100, blank=True, null=True, choices=[('index', 'Index'), ('weixin', 'Weixin')], verbose_name="来源")
|
|||
|
|
|
|||
|
|
# 用户的OpenID
|
|||
|
|
openid = models.CharField(max_length=150, blank=True, null=True, verbose_name="openid")
|
|||
|
|
|
|||
|
|
# 用户昵称
|
|||
|
|
nickname = models.CharField(max_length=255, blank=True, null=True, verbose_name="用户标识")
|
|||
|
|
|
|||
|
|
# 微信ID
|
|||
|
|
wxid = models.CharField(max_length=255, blank=True, null=True, verbose_name="微信id")
|
|||
|
|
|
|||
|
|
# 微信号
|
|||
|
|
wechat_alias = models.CharField(max_length=255, blank=True, null=True, verbose_name="微信号")
|
|||
|
|
|
|||
|
|
# 调用的API接口名称或路径
|
|||
|
|
api_name = models.CharField(max_length=255, verbose_name="调用接口")
|
|||
|
|
|
|||
|
|
# 调用结果，使用BooleanField表示成功或失败
|
|||
|
|
is_successful = models.BooleanField(default=False, verbose_name="是否成功")
|
|||
|
|
|
|||
|
|
# 备注信息，可以用来记录失败原因或其他注释信息
|
|||
|
|
remarks = models.TextField(blank=True, null=True, verbose_name="备注")
|
|||
|
|
|
|||
|
|
# 调用时间，使用自动设置为当前时间的DateTimeField
|
|||
|
|
call_time = models.DateTimeField(auto_now_add=True, verbose_name="调用时间")
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "api_call_log"
|
|||
|
|
verbose_name = "API调用日志"
|
|||
|
|
verbose_name_plural = "API调用日志"
|
|||
|
|
|
|||
|
|
def __str__(self):
|
|||
|
|
return f"{self.nickname} - {self.api_name} at {self.call_time}"
|
|||
|
|
|
|||
|
|
#卡密记录
|
|||
|
|
class RedemptionCard(models.Model):
|
|||
|
|
code = models.CharField(max_length=100, unique=True, verbose_name="卡密") # 卡密
|
|||
|
|
card_type = models.CharField(max_length=50, choices=(('member', '会员卡'), ('quota', '额度卡')), verbose_name="卡类型") # 卡类型
|
|||
|
|
created_date = models.DateTimeField(auto_now_add=True, verbose_name="生成日期") # 卡密生成日期
|
|||
|
|
expiry_date = models.DateTimeField(verbose_name="到期时间") # 卡密到期时间
|
|||
|
|
is_used = models.BooleanField(default=False, verbose_name="是否已使用") # 是否已被使用
|
|||
|
|
used_by_openid = models.CharField(max_length=150, blank=True, null=True, verbose_name="使用者OpenID") # 使用者的OpenID
|
|||
|
|
used_by_nickname = models.CharField(max_length=150, blank=True, null=True, verbose_name="使用者昵称") # 使用者的昵称
|
|||
|
|
card_creation_time = models.DateTimeField(auto_now_add=True, verbose_name="卡密创建时间") # 卡密创建时间
|
|||
|
|
validity_period = models.PositiveIntegerField(default=0, verbose_name="有效期") # 有效期天数或额度数量
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "redemption_card" # 设置数据库表名
|
|||
|
|
verbose_name = "兑换卡"
|
|||
|
|
verbose_name_plural = "兑换卡"
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 充值列表
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 订单记录
|
|||
|
|
class TransactionLog(models.Model):
|
|||
|
|
TRANSACTION_STATUS_CHOICES = [
|
|||
|
|
('pending', '待处理'),
|
|||
|
|
('completed', '完成'),
|
|||
|
|
('failed', '失败')
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
TRANSACTION_TYPE_CHOICES = [
|
|||
|
|
('member', '会员购买'),
|
|||
|
|
('quota', '续费会员')
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
|
|||
|
|
transaction_no = models.CharField(max_length=255, unique=True, verbose_name="交易编号")
|
|||
|
|
transaction_status = models.CharField(max_length=50, default='pending', choices=TRANSACTION_STATUS_CHOICES, verbose_name="交易状态")
|
|||
|
|
user_openid = models.CharField(max_length=150, verbose_name="用户OpenID")
|
|||
|
|
transaction_type = models.CharField(max_length=50, verbose_name="交易类型")
|
|||
|
|
transaction_amount = models.DecimalField(max_digits=10, decimal_places=2, verbose_name="交易金额")
|
|||
|
|
remark = models.TextField(blank=True, null=True, verbose_name="备注")
|
|||
|
|
created_at = models.DateTimeField(auto_now_add=True, verbose_name="创建时间")
|
|||
|
|
updated_at = models.DateTimeField(auto_now=True, verbose_name="更新时间")
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "transaction_log"
|
|||
|
|
verbose_name = "交易记录"
|
|||
|
|
verbose_name_plural = "交易记录"
|
|||
|
|
|
|||
|
|
def __str__(self):
|
|||
|
|
return f"{self.transaction_no} - {self.transaction_status}"
|
|||
|
|
|
|||
|
|
|
|||
|
|
class IDCounter(models.Model):
|
|||
|
|
count = models.IntegerField(default=10000) # 初始化为0或其他你想开始的数字
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def get_next_id(cls):
|
|||
|
|
# 这个方法获取下一个ID，并且自增存储的计数
|
|||
|
|
counter, _ = cls.objects.get_or_create(pk=1) # 假设我们只用一个计数器，其ID为1
|
|||
|
|
counter.count += 1
|
|||
|
|
counter.save()
|
|||
|
|
return counter.count
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "IDCounter"
|
|||
|
|
|
|||
|
|
class ChatRecord(models.Model):
|
|||
|
|
openid = models.CharField(max_length=150,default=None) # 用户唯一标识
|
|||
|
|
nickname = models.CharField(max_length=150, verbose_name='用户昵称') # 用户昵称
|
|||
|
|
conversation_id = models.CharField(max_length=100, verbose_name='会话ID') # 会话ID
|
|||
|
|
role = models.CharField(max_length=100, verbose_name='角色') # 角色
|
|||
|
|
message_content = models.TextField(verbose_name='消息内容') # 消息内容
|
|||
|
|
is_response = models.BooleanField(default=False, verbose_name='是否为AI回复') # 是否为AI回复
|
|||
|
|
timestamp = models.DateTimeField(default=timezone.now, verbose_name='时间戳') # 时间戳
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "chat_record" # 数据库表名
|
|||
|
|
ordering = ['-timestamp'] # 默认排序
|
|||
|
|
verbose_name = '聊天记录'
|
|||
|
|
verbose_name_plural = '聊天记录'
|
|||
|
|
|
|||
|
|
def __str__(self):
|
|||
|
|
return f"{self.nickname}: {self.message_content[:20]}"
|
|||
|
|
|
|||
|
|
class VideoTask(models.Model):
|
|||
|
|
TASK_STATUS_CHOICES = [
|
|||
|
|
('pending','待处理'),
|
|||
|
|
('failed', '已提交'),
|
|||
|
|
('running','运行中'),
|
|||
|
|
('success', '成功'),
|
|||
|
|
('failed', '失败'),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
openid = models.CharField(max_length=150) # 用户唯一标识
|
|||
|
|
nickname = models.CharField(max_length=150) # 用户昵称
|
|||
|
|
task_id = models.CharField(max_length=255, unique=True) # 任务ID
|
|||
|
|
task_type = models.CharField(max_length=50) # 任务类型，例如 'gen2'
|
|||
|
|
status = models.CharField(max_length=20, choices=TASK_STATUS_CHOICES, default='pending') # 任务状态
|
|||
|
|
created_at = models.DateTimeField(auto_now_add=True) # 任务创建时间
|
|||
|
|
updated_at = models.DateTimeField(auto_now=True) # 任务更新时间
|
|||
|
|
text_prompt = models.TextField(null=True) # 文本提示
|
|||
|
|
width = models.IntegerField(null=True) # 视频宽度
|
|||
|
|
height = models.IntegerField(null=True) # 视频高度
|
|||
|
|
motion_score = models.IntegerField(null=True) # 动态评分
|
|||
|
|
style = models.CharField(max_length=255, blank=True, null=True) # 风格
|
|||
|
|
seconds = models.IntegerField(null=True) # 视频时长
|
|||
|
|
image_url = models.TextField(blank=True, null=True) # 图生视频的图像URL
|
|||
|
|
result_url = models.TextField(blank=True, null=True) # 视频生成结果URL
|
|||
|
|
qiniu_url = models.TextField(blank=True, null=True) # 七牛云储存url
|
|||
|
|
gif_url = models.TextField(blank=True, null=True) # GIF图片链接
|
|||
|
|
progress = models.FloatField(default=0.0) # 进度百分比
|
|||
|
|
error_message = models.TextField(blank=True, null=True) # 错误信息
|
|||
|
|
description_zh = models.TextField(blank=True, null=True) # 中文描述
|
|||
|
|
is_processing = models.BooleanField(default=False) # 正在处理标志
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "video_task"
|
|||
|
|
ordering = ['-created_at']
|
|||
|
|
|
|||
|
|
def __str__(self):
|
|||
|
|
return self.task_id
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AssetLibrary(models.Model):
|
|||
|
|
STYLE_CHOICES = [
|
|||
|
|
('abandoned', '废弃'),
|
|||
|
|
('abstract_sculpture', '抽象'),
|
|||
|
|
('advertising', '广告'),
|
|||
|
|
('anime', '动漫'),
|
|||
|
|
('cine_lens', '电影镜头'),
|
|||
|
|
('cinematic', '电影'),
|
|||
|
|
('concept_art', '艺术'),
|
|||
|
|
('forestpunk', '赛博朋克'),
|
|||
|
|
('frost', '雪'),
|
|||
|
|
('graphite', '石墨'),
|
|||
|
|
('macro_photography', '宏观'),
|
|||
|
|
('pixel_art', '像素艺术'),
|
|||
|
|
('retro_photography', '复古'),
|
|||
|
|
('sci_fi_art', '科幻'),
|
|||
|
|
('thriller', '惊悚'),
|
|||
|
|
('35mm', '35mm'),
|
|||
|
|
('vector', '矢量'),
|
|||
|
|
('watercolor', '水彩'),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
original_url = models.TextField(verbose_name='原始URL', null=True) # 原始URL
|
|||
|
|
qiniu_url = models.TextField(blank=True, null=True) # 七牛云储存url
|
|||
|
|
duration = models.FloatField(verbose_name='时长') # 视频时长，以秒为单位
|
|||
|
|
category = models.CharField(max_length=50, choices=STYLE_CHOICES, verbose_name='分类', null=True) # 视频分类
|
|||
|
|
description = models.TextField(verbose_name='视频描述', blank=True, null=True) # 视频描述
|
|||
|
|
description_zh = models.TextField(verbose_name='中文描述', blank=True, null=True) # 中文描述
|
|||
|
|
generated_by = models.CharField(null=True, verbose_name='生成用户', max_length=50) # 生成用户
|
|||
|
|
generated_at = models.DateTimeField(default=timezone.now, verbose_name='生成时间', null=True) # 生成时间
|
|||
|
|
download_count = models.IntegerField(default=0, verbose_name='下载次数', null=True) # 下载次数
|
|||
|
|
is_approved = models.BooleanField(default=False, verbose_name='是否审核', null=True) # 是否审核
|
|||
|
|
gif_url = models.TextField(verbose_name='GIF图片URL', blank=True, null=True) # GIF图片URL
|
|||
|
|
|
|||
|
|
class Meta:
|
|||
|
|
db_table = "asset_library" # 数据库表名
|
|||
|
|
verbose_name = '素材库'
|
|||
|
|
verbose_name_plural = '素材库'
|
|||
|
|
ordering = ['-generated_at'] # 默认排序
|
|||
|
|
|
|||
|
|
def __str__(self):
|
|||
|
|
return self.original_url
|