90 lines
4.0 KiB
Python
90 lines
4.0 KiB
Python
from django.db import models
|
||
from apps.accounts.models import User
|
||
import json
|
||
import os
|
||
from django.utils import timezone
|
||
|
||
class GmailCredential(models.Model):
|
||
user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='gmail_credentials')
|
||
email = models.EmailField(unique=True, help_text="Gmail email address")
|
||
credentials = models.TextField(help_text="Serialized OAuth2 credentials (JSON)")
|
||
is_default = models.BooleanField(default=False, help_text="Default Gmail account for user")
|
||
created_at = models.DateTimeField(auto_now_add=True)
|
||
updated_at = models.DateTimeField(auto_now=True)
|
||
is_valid = models.BooleanField(default=True, help_text="Whether the credential is valid")
|
||
|
||
class Meta:
|
||
unique_together = ('user', 'email')
|
||
|
||
def set_credentials(self, credentials):
|
||
self.credentials = json.dumps({
|
||
'token': credentials.token,
|
||
'refresh_token': credentials.refresh_token,
|
||
'token_uri': credentials.token_uri,
|
||
'client_id': credentials.client_id,
|
||
'client_secret': credentials.client_secret,
|
||
'scopes': credentials.scopes
|
||
})
|
||
self.is_valid = True
|
||
|
||
def get_credentials(self):
|
||
from google.oauth2.credentials import Credentials
|
||
creds_data = json.loads(self.credentials)
|
||
return Credentials(
|
||
token=creds_data['token'],
|
||
refresh_token=creds_data['refresh_token'],
|
||
token_uri=creds_data['token_uri'],
|
||
client_id=creds_data['client_id'],
|
||
client_secret=creds_data['client_secret'],
|
||
scopes=creds_data['scopes']
|
||
)
|
||
|
||
def __str__(self):
|
||
return f"{self.user.username} - {self.email}"
|
||
|
||
class GmailConversation(models.Model):
|
||
"""Gmail对话记录，跟踪用户和达人之间的邮件交互"""
|
||
user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='gmail_conversations')
|
||
user_email = models.EmailField(help_text="用户Gmail邮箱")
|
||
influencer_email = models.EmailField(help_text="达人Gmail邮箱")
|
||
conversation_id = models.CharField(max_length=100, unique=True, help_text="关联到chat_history的会话ID")
|
||
title = models.CharField(max_length=100, default="Gmail对话", help_text="对话标题")
|
||
last_sync_time = models.DateTimeField(default=timezone.now, help_text="最后同步时间")
|
||
created_at = models.DateTimeField(auto_now_add=True)
|
||
updated_at = models.DateTimeField(auto_now=True)
|
||
is_active = models.BooleanField(default=True)
|
||
|
||
def __str__(self):
|
||
return f"{self.user.username}: {self.user_email} - {self.influencer_email}"
|
||
|
||
class Meta:
|
||
ordering = ['-updated_at']
|
||
unique_together = ('user', 'user_email', 'influencer_email')
|
||
|
||
class GmailAttachment(models.Model):
|
||
"""Gmail附件记录"""
|
||
conversation = models.ForeignKey(GmailConversation, on_delete=models.CASCADE, related_name='attachments')
|
||
email_message_id = models.CharField(max_length=100, help_text="Gmail邮件ID")
|
||
attachment_id = models.CharField(max_length=100, help_text="Gmail附件ID")
|
||
filename = models.CharField(max_length=255, help_text="原始文件名")
|
||
file_path = models.CharField(max_length=255, help_text="保存在服务器上的路径")
|
||
content_type = models.CharField(max_length=100, help_text="MIME类型")
|
||
size = models.IntegerField(default=0, help_text="文件大小(字节)")
|
||
sender_email = models.EmailField(help_text="发送者邮箱")
|
||
chat_message_id = models.CharField(max_length=100, blank=True, null=True, help_text="关联到ChatHistory的消息ID")
|
||
created_at = models.DateTimeField(auto_now_add=True)
|
||
|
||
def __str__(self):
|
||
return f"{self.filename} ({self.size} bytes)"
|
||
|
||
def get_file_extension(self):
|
||
"""获取文件扩展名"""
|
||
_, ext = os.path.splitext(self.filename)
|
||
return ext.lower()
|
||
|
||
def get_absolute_url(self):
|
||
"""获取文件URL"""
|
||
return f"/media/gmail_attachments/{os.path.basename(self.file_path)}"
|
||
|
||
class Meta:
|
||
ordering = ['-created_at'] |