153 lines
5.6 KiB
Python
153 lines
5.6 KiB
Python
|
import logging
|
|||
|
|
import re
|
|||
|
|
import json
|
|||
|
|
from django.db import transaction
|
|||
|
|
from apps.gmail.models import GmailConversation
|
|||
|
|
from apps.accounts.models import User
|
|||
|
|
from ..models import FeishuTableMapping
|
|||
|
|
from .bitable_service import BitableService
|
|||
|
|
from apps.gmail.services.gmail_service import GmailService
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
class GmailExtractionService:
|
|||
|
|
"""
|
|||
|
|
从飞书多维表格中提取Gmail邮箱并创建对话的服务
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def extract_gmail_addresses(text):
|
|||
|
|
"""
|
|||
|
|
从文本中提取Gmail邮箱地址
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
text: 需要提取邮箱的文本
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
list: Gmail邮箱地址列表
|
|||
|
|
"""
|
|||
|
|
if not text:
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
# Gmail邮箱正则表达式模式
|
|||
|
|
gmail_pattern = r'[a-zA-Z0-9._%+-]+@gmail\.com'
|
|||
|
|
|
|||
|
|
# 查找所有匹配
|
|||
|
|
matches = re.findall(gmail_pattern, text.lower())
|
|||
|
|
|
|||
|
|
# 返回唯一的Gmail地址
|
|||
|
|
return list(set(matches))
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def find_duplicate_emails(db_table_name, feishu_table_url, access_token, email_field_name, user):
|
|||
|
|
"""
|
|||
|
|
查找数据库表和飞书多维表格中重复的Gmail邮箱
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
db_table_name: 数据库表名
|
|||
|
|
feishu_table_url: 飞书多维表格URL
|
|||
|
|
access_token: 访问令牌
|
|||
|
|
email_field_name: 包含Gmail邮箱的字段名
|
|||
|
|
user: 当前用户对象
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
tuple: (邮箱列表, 错误信息)
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
# 从URL中提取app_token和table_id
|
|||
|
|
app_token, table_id = BitableService.extract_params_from_url(feishu_table_url)
|
|||
|
|
|
|||
|
|
# 获取飞书表格中的所有记录
|
|||
|
|
feishu_records = BitableService.search_records(
|
|||
|
|
app_token=app_token,
|
|||
|
|
table_id=table_id,
|
|||
|
|
access_token=access_token,
|
|||
|
|
page_size=1000 # 获取足够多的记录
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if not feishu_records or 'items' not in feishu_records:
|
|||
|
|
return None, "无法获取飞书表格数据"
|
|||
|
|
|
|||
|
|
# 提取每条记录中的Gmail邮箱
|
|||
|
|
feishu_emails = []
|
|||
|
|
for record in feishu_records.get('items', []):
|
|||
|
|
field_data = record.get('fields', {})
|
|||
|
|
if email_field_name in field_data:
|
|||
|
|
email_value = field_data[email_field_name]
|
|||
|
|
if isinstance(email_value, str):
|
|||
|
|
gmail_addresses = GmailExtractionService.extract_gmail_addresses(email_value)
|
|||
|
|
feishu_emails.extend(gmail_addresses)
|
|||
|
|
|
|||
|
|
# 确保邮箱地址唯一
|
|||
|
|
feishu_emails = list(set(feishu_emails))
|
|||
|
|
|
|||
|
|
logger.info(f"从飞书表格中提取出 {len(feishu_emails)} 个Gmail邮箱")
|
|||
|
|
|
|||
|
|
return feishu_emails, None
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"查找重复的Gmail邮箱失败: {str(e)}")
|
|||
|
|
return None, f"查找重复的Gmail邮箱失败: {str(e)}"
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
@transaction.atomic
|
|||
|
|
def create_conversations_for_emails(user, user_email, emails, kb_id=None):
|
|||
|
|
"""
|
|||
|
|
为提取的Gmail邮箱创建对话
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
user: 当前用户对象
|
|||
|
|
user_email: 用户Gmail邮箱
|
|||
|
|
emails: 达人Gmail邮箱列表
|
|||
|
|
kb_id: 知识库ID
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
tuple: (成功创建的对话数量, 错误信息)
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
if not emails:
|
|||
|
|
return 0, "没有提供邮箱列表"
|
|||
|
|
|
|||
|
|
success_count = 0
|
|||
|
|
failed_emails = []
|
|||
|
|
|
|||
|
|
for email in emails:
|
|||
|
|
try:
|
|||
|
|
# 检查是否已存在对话
|
|||
|
|
existing_conversation = GmailConversation.objects.filter(
|
|||
|
|
user=user,
|
|||
|
|
user_email=user_email,
|
|||
|
|
influencer_email=email
|
|||
|
|
).exists()
|
|||
|
|
|
|||
|
|
if existing_conversation:
|
|||
|
|
logger.info(f"用户 {user.username} 与 {email} 的对话已存在，跳过")
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# 创建新对话
|
|||
|
|
conversation_id, error = GmailService.save_conversations_to_chat(
|
|||
|
|
user=user,
|
|||
|
|
user_email=user_email,
|
|||
|
|
influencer_email=email,
|
|||
|
|
kb_id=kb_id
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if conversation_id:
|
|||
|
|
success_count += 1
|
|||
|
|
logger.info(f"成功创建与 {email} 的对话，ID: {conversation_id}")
|
|||
|
|
else:
|
|||
|
|
logger.error(f"创建与 {email} 的对话失败: {error}")
|
|||
|
|
failed_emails.append(email)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"处理邮箱 {email} 时出错: {str(e)}")
|
|||
|
|
failed_emails.append(email)
|
|||
|
|
|
|||
|
|
if failed_emails:
|
|||
|
|
return success_count, f"成功创建 {success_count} 个对话，失败 {len(failed_emails)} 个: {', '.join(failed_emails[:5])}{' 等' if len(failed_emails) > 5 else ''}"
|
|||
|
|
else:
|
|||
|
|
return success_count, None
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"创建Gmail对话失败: {str(e)}")
|
|||
|
|
return 0, f"创建Gmail对话失败: {str(e)}"
|