712 lines
27 KiB
Python
712 lines
27 KiB
Python
import logging
|
||
import traceback
|
||
import os
|
||
import pandas as pd
|
||
from django.http import FileResponse, HttpResponse
|
||
from django.conf import settings
|
||
from rest_framework import status
|
||
from rest_framework.decorators import api_view, permission_classes
|
||
from rest_framework.permissions import IsAuthenticated
|
||
from rest_framework.response import Response
|
||
|
||
from user_management.models import (
|
||
GmailTalentMapping, ChatHistory, ConversationSummary
|
||
)
|
||
from user_management.gmail_integration import GmailIntegration
|
||
|
||
from feishu.feishu_ai_chat import (
|
||
fetch_table_records, find_duplicate_email_creators,
|
||
process_duplicate_emails, auto_chat_session,
|
||
check_goal_achieved, export_feishu_creators_to_excel,
|
||
export_matching_emails_to_excel
|
||
)
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
@api_view(['POST'])
|
||
@permission_classes([IsAuthenticated])
|
||
def process_feishu_table(request):
|
||
"""
|
||
从飞书多维表格读取数据，处理重复邮箱
|
||
|
||
请求参数:
|
||
table_id: 表格ID
|
||
view_id: 视图ID
|
||
app_token: 飞书应用TOKEN (可选)
|
||
access_token: 用户访问令牌 (可选)
|
||
app_id: 应用ID (可选)
|
||
app_secret: 应用密钥 (可选)
|
||
goal_template: 目标内容模板 (可选)
|
||
auto_chat: 是否自动执行AI对话 (可选)
|
||
turns: 自动对话轮次 (可选)
|
||
"""
|
||
try:
|
||
# 检查用户权限 - 只允许组长使用
|
||
if request.user.role != 'leader':
|
||
return Response(
|
||
{"error": "只有组长角色的用户可以使用此功能"},
|
||
status=status.HTTP_403_FORBIDDEN
|
||
)
|
||
|
||
# 获取参数
|
||
table_id = request.data.get("table_id", "tbl3oikG3F8YYtVA") # 默认表格ID
|
||
view_id = request.data.get("view_id", "vewSOIsmxc") # 默认视图ID
|
||
app_token = request.data.get("app_token", "XYE6bMQUOaZ5y5svj4vcWohGnmg")
|
||
access_token = request.data.get("access_token", "u-fK0HvbXVte.G2xzYs5oxV6k1nHu1glvFgG00l0Ma24VD")
|
||
app_id = request.data.get("app_id", "cli_a5c97daacb9e500d")
|
||
app_secret = request.data.get("app_secret", "fdVeOCLXmuIHZVmSV0VbJh9wd0Kq1o5y")
|
||
goal_template = request.data.get(
|
||
"goal_template",
|
||
"与达人{handle}(邮箱:{email})建立联系并了解其账号情况，评估合作潜力，处理合作需求，最终目标是达成合作并签约。"
|
||
)
|
||
auto_chat = request.data.get("auto_chat", False)
|
||
turns = request.data.get("turns", 5)
|
||
|
||
logger.info(f"处理飞书表格数据: table_id={table_id}, view_id={view_id}, app_id={app_id}")
|
||
|
||
# 从飞书表格获取记录
|
||
records = fetch_table_records(
|
||
app_token,
|
||
table_id,
|
||
view_id,
|
||
access_token,
|
||
app_id,
|
||
app_secret
|
||
)
|
||
|
||
if not records:
|
||
logger.warning("未获取到任何记录，可能是表格ID或视图ID不正确，或无权限访问")
|
||
|
||
# 尝试使用SDK中的search方法直接获取
|
||
try:
|
||
import lark_oapi as lark
|
||
from lark_oapi.api.bitable.v1 import (
|
||
SearchAppTableRecordRequest,
|
||
SearchAppTableRecordRequestBody
|
||
)
|
||
|
||
# 创建client
|
||
client = lark.Client.builder() \
|
||
.enable_set_token(True) \
|
||
.log_level(lark.LogLevel.DEBUG) \
|
||
.build()
|
||
|
||
# 构造请求对象
|
||
request = SearchAppTableRecordRequest.builder() \
|
||
.app_token(app_token) \
|
||
.table_id(table_id) \
|
||
.page_size(20) \
|
||
.request_body(SearchAppTableRecordRequestBody.builder().build()) \
|
||
.build()
|
||
|
||
# 发起请求
|
||
option = lark.RequestOption.builder().user_access_token(access_token).build()
|
||
response = client.bitable.v1.app_table_record.search(request, option)
|
||
|
||
if not response.success():
|
||
logger.error(f"直接搜索请求失败: {response.code}, {response.msg}")
|
||
return Response(
|
||
{"message": f"未获取到任何记录，错误: {response.msg}"},
|
||
status=status.HTTP_404_NOT_FOUND
|
||
)
|
||
|
||
# 获取记录
|
||
records = response.data.items
|
||
if not records:
|
||
return Response(
|
||
{"message": "未获取到任何记录"},
|
||
status=status.HTTP_404_NOT_FOUND
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"尝试直接搜索时出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
return Response(
|
||
{"message": f"未获取到任何记录，错误: {str(e)}"},
|
||
status=status.HTTP_404_NOT_FOUND
|
||
)
|
||
|
||
# 查找重复邮箱的创作者
|
||
duplicate_emails = find_duplicate_email_creators(records)
|
||
|
||
if not duplicate_emails:
|
||
return Response(
|
||
{"message": "未找到与系统中已有creator匹配的邮箱"},
|
||
status=status.HTTP_200_OK
|
||
)
|
||
|
||
# 处理匹配的邮箱记录
|
||
results = process_duplicate_emails(duplicate_emails, goal_template)
|
||
|
||
# 如果需要自动对话
|
||
chat_results = []
|
||
if auto_chat and results['success'] > 0:
|
||
# 为每个成功创建的记录执行自动对话
|
||
for detail in results['details']:
|
||
if detail['status'] == 'success':
|
||
email = detail['email']
|
||
chat_result = auto_chat_session(request.user, email, max_turns=turns)
|
||
chat_results.append({
|
||
'email': email,
|
||
'result': chat_result
|
||
})
|
||
|
||
# 返回处理结果
|
||
return Response({
|
||
'status': 'success',
|
||
'records_count': len(records),
|
||
'duplicate_emails_count': len(duplicate_emails),
|
||
'processing_results': results,
|
||
'chat_results': chat_results
|
||
}, status=status.HTTP_200_OK)
|
||
|
||
except Exception as e:
|
||
logger.error(f"处理飞书表格时出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
return Response(
|
||
{"error": str(e)},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
|
||
@api_view(['POST'])
|
||
@permission_classes([IsAuthenticated])
|
||
def run_auto_chat(request):
|
||
"""
|
||
为指定邮箱执行自动对话
|
||
|
||
请求参数:
|
||
email: 达人邮箱
|
||
force_send: 是否强制发送新邮件(即使没有新回复)(可选)
|
||
subject: 邮件主题(可选，仅当force_send=true时使用)
|
||
content: 邮件内容(可选，仅当force_send=true时使用)
|
||
"""
|
||
try:
|
||
# 检查用户权限 - 只允许组长使用
|
||
if request.user.role != 'leader':
|
||
return Response(
|
||
{"error": "只有组长角色的用户可以使用此功能"},
|
||
status=status.HTTP_403_FORBIDDEN
|
||
)
|
||
|
||
# 获取参数
|
||
email = request.data.get("email")
|
||
force_send = request.data.get("force_send", False)
|
||
subject = request.data.get("subject")
|
||
content = request.data.get("content")
|
||
|
||
# 验证必要参数
|
||
if not email:
|
||
return Response(
|
||
{"error": "缺少参数email"},
|
||
status=status.HTTP_400_BAD_REQUEST
|
||
)
|
||
|
||
# 如果强制发送且没有提供内容
|
||
if force_send and not content:
|
||
return Response(
|
||
{"error": "当force_send=true时，必须提供content参数"},
|
||
status=status.HTTP_400_BAD_REQUEST
|
||
)
|
||
|
||
# 首先尝试同步最新邮件
|
||
try:
|
||
# 创建Gmail集成实例
|
||
gmail_integration = GmailIntegration(request.user)
|
||
|
||
# 同步最新邮件
|
||
logger.info(f"正在同步与 {email} 的最新邮件...")
|
||
sync_result = gmail_integration.sync_talent_emails(email)
|
||
|
||
if sync_result.get('status') == 'success':
|
||
logger.info(f"成功同步邮件: {sync_result.get('message', 'No message')}")
|
||
else:
|
||
logger.warning(f"同步邮件警告: {sync_result.get('message', 'Unknown warning')}")
|
||
|
||
except Exception as e:
|
||
logger.error(f"同步邮件出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
# 仅记录错误，不中断流程
|
||
|
||
# 如果是强制发送模式
|
||
if force_send:
|
||
try:
|
||
# 创建Gmail集成实例
|
||
gmail_integration = GmailIntegration(request.user)
|
||
|
||
# 检查Gmail服务是否已正确初始化
|
||
if not hasattr(gmail_integration, 'gmail_service') or gmail_integration.gmail_service is None:
|
||
return Response(
|
||
{"error": "Gmail服务未正确初始化，请检查Gmail API配置"},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
|
||
# 获取知识库映射
|
||
mapping = GmailTalentMapping.objects.filter(
|
||
user=request.user,
|
||
talent_email=email,
|
||
is_active=True
|
||
).first()
|
||
|
||
if not mapping:
|
||
return Response(
|
||
{"error": f"找不到与邮箱 {email} 的映射关系"},
|
||
status=status.HTTP_404_NOT_FOUND
|
||
)
|
||
|
||
# 直接发送邮件
|
||
mail_subject = subject if subject else "关于合作的洽谈"
|
||
mail_result = gmail_integration.send_email(
|
||
to_email=email,
|
||
subject=mail_subject,
|
||
body=content,
|
||
conversation_id=mapping.conversation_id
|
||
)
|
||
|
||
if mail_result['status'] != 'success':
|
||
return Response(
|
||
{"error": f"邮件发送失败: {mail_result.get('message', 'Unknown error')}"},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
|
||
# 保存发送的内容到对话历史
|
||
ChatHistory.objects.create(
|
||
user=request.user,
|
||
knowledge_base=mapping.knowledge_base,
|
||
conversation_id=mapping.conversation_id,
|
||
role='assistant',
|
||
content=content
|
||
)
|
||
|
||
return Response({
|
||
'status': 'success',
|
||
'message': f"已强制发送邮件到 {email}",
|
||
'email_sent': True,
|
||
'conversation_id': mapping.conversation_id
|
||
}, status=status.HTTP_200_OK)
|
||
|
||
except Exception as e:
|
||
logger.error(f"强制发送邮件时出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
return Response(
|
||
{"error": str(e)},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
|
||
# 执行自动对话
|
||
result = auto_chat_session(request.user, email)
|
||
|
||
# 返回结果
|
||
return Response(result, status=status.HTTP_200_OK)
|
||
|
||
except Exception as e:
|
||
logger.error(f"执行自动对话时出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
return Response(
|
||
{"error": str(e)},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
|
||
@api_view(['GET', 'POST'])
|
||
@permission_classes([IsAuthenticated])
|
||
def feishu_user_goal(request):
|
||
"""
|
||
设置或获取用户总目标
|
||
|
||
GET 请求:
|
||
获取当前用户总目标
|
||
|
||
POST 请求参数:
|
||
email: 达人邮箱
|
||
goal: 目标内容
|
||
"""
|
||
try:
|
||
# 检查用户权限 - 只允许组长使用
|
||
if request.user.role != 'leader':
|
||
return Response(
|
||
{"error": "只有组长角色的用户可以使用此功能"},
|
||
status=status.HTTP_403_FORBIDDEN
|
||
)
|
||
|
||
if request.method == 'GET':
|
||
# 创建Gmail集成实例
|
||
gmail_integration = GmailIntegration(request.user)
|
||
# 获取总目标
|
||
result = gmail_integration.manage_user_goal()
|
||
return Response(result, status=status.HTTP_200_OK)
|
||
|
||
elif request.method == 'POST':
|
||
# 获取参数
|
||
email = request.data.get("email")
|
||
goal = request.data.get("goal")
|
||
|
||
# 验证必要参数
|
||
if not email:
|
||
return Response(
|
||
{"error": "缺少参数email"},
|
||
status=status.HTTP_400_BAD_REQUEST
|
||
)
|
||
if not goal:
|
||
return Response(
|
||
{"error": "缺少参数goal"},
|
||
status=status.HTTP_400_BAD_REQUEST
|
||
)
|
||
|
||
# 设置用户总目标
|
||
gmail_integration = GmailIntegration(request.user)
|
||
result = gmail_integration.manage_user_goal(goal)
|
||
|
||
return Response(result, status=status.HTTP_200_OK)
|
||
|
||
except Exception as e:
|
||
logger.error(f"管理用户总目标时出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
return Response(
|
||
{"error": str(e)},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
|
||
@api_view(['GET'])
|
||
@permission_classes([IsAuthenticated])
|
||
def check_goal_status(request):
|
||
"""
|
||
检查目标完成状态
|
||
|
||
请求参数:
|
||
email: 达人邮箱
|
||
"""
|
||
try:
|
||
# 检查用户权限 - 只允许组长使用
|
||
if request.user.role != 'leader':
|
||
return Response(
|
||
{"error": "只有组长角色的用户可以使用此功能"},
|
||
status=status.HTTP_403_FORBIDDEN
|
||
)
|
||
|
||
# 获取参数
|
||
email = request.query_params.get("email")
|
||
|
||
# 验证必要参数
|
||
if not email:
|
||
return Response(
|
||
{"error": "缺少参数email"},
|
||
status=status.HTTP_400_BAD_REQUEST
|
||
)
|
||
|
||
# 查找Gmail映射关系
|
||
mapping = GmailTalentMapping.objects.filter(
|
||
user=request.user,
|
||
talent_email=email,
|
||
is_active=True
|
||
).first()
|
||
|
||
if not mapping:
|
||
return Response(
|
||
{"error": f"找不到与邮箱 {email} 的映射关系"},
|
||
status=status.HTTP_404_NOT_FOUND
|
||
)
|
||
|
||
# 获取对话历史中最后的AI回复
|
||
last_ai_message = ChatHistory.objects.filter(
|
||
user=request.user,
|
||
knowledge_base=mapping.knowledge_base,
|
||
conversation_id=mapping.conversation_id,
|
||
role='assistant',
|
||
is_deleted=False
|
||
).order_by('-created_at').first()
|
||
|
||
if not last_ai_message:
|
||
return Response(
|
||
{"error": f"找不到与邮箱 {email} 的对话历史"},
|
||
status=status.HTTP_404_NOT_FOUND
|
||
)
|
||
|
||
# 检查目标是否已达成
|
||
goal_achieved = check_goal_achieved(last_ai_message.content)
|
||
|
||
# 获取对话总结
|
||
summary = ConversationSummary.objects.filter(
|
||
user=request.user,
|
||
talent_email=email,
|
||
is_active=True
|
||
).order_by('-updated_at').first()
|
||
|
||
result = {
|
||
'status': 'success',
|
||
'email': email,
|
||
'goal_achieved': goal_achieved,
|
||
'last_message_time': last_ai_message.created_at.strftime('%Y-%m-%d %H:%M:%S'),
|
||
'last_message': last_ai_message.content,
|
||
'summary': summary.summary if summary else None
|
||
}
|
||
|
||
return Response(result, status=status.HTTP_200_OK)
|
||
|
||
except Exception as e:
|
||
logger.error(f"检查目标状态时出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
return Response(
|
||
{"error": str(e)},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
|
||
@api_view(['POST'])
|
||
@permission_classes([IsAuthenticated])
|
||
def export_creators_data(request):
|
||
"""
|
||
导出匹配的FeishuCreator数据到Excel文件
|
||
|
||
请求参数:
|
||
table_id: 表格ID
|
||
view_id: 视图ID
|
||
app_token: 飞书应用TOKEN (可选)
|
||
access_token: 用户访问令牌 (可选)
|
||
app_id: 应用ID (可选)
|
||
app_secret: 应用密钥 (可选)
|
||
export_type: 导出类型，'creators'或'feishu'，默认为'creators'
|
||
format: 格式，支持'excel'或'csv'，默认为'excel'
|
||
"""
|
||
try:
|
||
# 检查用户权限 - 只允许组长使用
|
||
if request.user.role != 'leader':
|
||
return Response(
|
||
{"error": "只有组长角色的用户可以使用此功能"},
|
||
status=status.HTTP_403_FORBIDDEN
|
||
)
|
||
|
||
# 获取参数
|
||
table_id = request.data.get("table_id", "tbl3oikG3F8YYtVA") # 默认表格ID
|
||
view_id = request.data.get("view_id", "vewSOIsmxc") # 默认视图ID
|
||
app_token = request.data.get("app_token", "XYE6bMQUOaZ5y5svj4vcWohGnmg")
|
||
access_token = request.data.get("access_token", "u-fK0HvbXVte.G2xzYs5oxV6k1nHu1glvFgG00l0Ma24VD")
|
||
app_id = request.data.get("app_id", "cli_a5c97daacb9e500d")
|
||
app_secret = request.data.get("app_secret", "fdVeOCLXmuIHZVmSV0VbJh9wd0Kq1o5y")
|
||
export_type = request.data.get("export_type", "creators") # 导出类型：creators或feishu
|
||
export_format = request.data.get("format", "excel") # 导出格式：excel或csv
|
||
|
||
if export_format not in ["excel", "csv"]:
|
||
return Response(
|
||
{"error": "当前支持的格式有: excel, csv"},
|
||
status=status.HTTP_400_BAD_REQUEST
|
||
)
|
||
|
||
logger.info(f"导出飞书数据: table_id={table_id}, view_id={view_id}, type={export_type}, format={export_format}")
|
||
|
||
# 从飞书表格获取记录
|
||
records = fetch_table_records(
|
||
app_token,
|
||
table_id,
|
||
view_id,
|
||
access_token,
|
||
app_id,
|
||
app_secret
|
||
)
|
||
|
||
if not records:
|
||
logger.warning("未获取到任何记录，可能是表格ID或视图ID不正确，或无权限访问")
|
||
return Response(
|
||
{"message": "未获取到任何记录"},
|
||
status=status.HTTP_404_NOT_FOUND
|
||
)
|
||
|
||
# 查找重复邮箱的创作者
|
||
duplicate_emails = find_duplicate_email_creators(records)
|
||
|
||
if not duplicate_emails:
|
||
return Response(
|
||
{"message": "未找到与系统中已有creator匹配的邮箱"},
|
||
status=status.HTTP_404_NOT_FOUND
|
||
)
|
||
|
||
# 创建存储导出文件的目录
|
||
export_dir = os.path.join(settings.MEDIA_ROOT, 'exports')
|
||
os.makedirs(export_dir, exist_ok=True)
|
||
|
||
# 根据导出类型和格式选择输出文件名
|
||
if export_type == "creators":
|
||
file_prefix = "feishu_creators"
|
||
else:
|
||
file_prefix = "feishu_data"
|
||
|
||
if export_format == "excel":
|
||
file_ext = ".xlsx"
|
||
else:
|
||
file_ext = ".csv"
|
||
|
||
output_filename = f"{file_prefix}_{request.user.id}{file_ext}"
|
||
output_path = os.path.join(export_dir, output_filename)
|
||
|
||
# 根据导出类型选择导出函数
|
||
if export_type == "creators":
|
||
# 导出FeishuCreator数据
|
||
if export_format == "excel":
|
||
excel_path = export_feishu_creators_to_excel(duplicate_emails, output_path)
|
||
|
||
if not excel_path:
|
||
return Response(
|
||
{"error": "导出FeishuCreator数据失败"},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
else:
|
||
# CSV格式导出直接处理
|
||
try:
|
||
# 获取所有匹配的邮箱
|
||
emails = list(duplicate_emails.keys())
|
||
|
||
# 从数据库获取FeishuCreator记录
|
||
from user_management.models import FeishuCreator
|
||
creators = FeishuCreator.objects.filter(email__in=emails)
|
||
|
||
if not creators.exists():
|
||
return Response(
|
||
{"error": "没有找到匹配的FeishuCreator记录"},
|
||
status=status.HTTP_404_NOT_FOUND
|
||
)
|
||
|
||
# 创建数据列表
|
||
data = []
|
||
for creator in creators:
|
||
# 处理datetime字段，移除时区信息
|
||
created_at = creator.created_at
|
||
if hasattr(created_at, 'tzinfo') and created_at.tzinfo is not None:
|
||
created_at = created_at.replace(tzinfo=None)
|
||
|
||
updated_at = creator.updated_at
|
||
if hasattr(updated_at, 'tzinfo') and updated_at.tzinfo is not None:
|
||
updated_at = updated_at.replace(tzinfo=None)
|
||
|
||
row = {
|
||
'id': str(creator.id),
|
||
'handle': creator.handle,
|
||
'email': creator.email,
|
||
'phone': creator.phone,
|
||
'created_at': created_at,
|
||
'updated_at': updated_at,
|
||
# 其他需要的字段...
|
||
}
|
||
data.append(row)
|
||
|
||
# 创建DataFrame并导出到CSV
|
||
df = pd.DataFrame(data)
|
||
df.to_csv(output_path, index=False, encoding='utf-8-sig') # 使用BOM标记以支持中文
|
||
excel_path = output_path
|
||
except Exception as e:
|
||
logger.error(f"导出CSV时出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
return Response(
|
||
{"error": f"导出CSV失败: {str(e)}"},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
|
||
elif export_type == "feishu":
|
||
# 导出飞书原始数据
|
||
if export_format == "excel":
|
||
excel_path = export_matching_emails_to_excel(duplicate_emails, records, output_path)
|
||
|
||
if not excel_path:
|
||
return Response(
|
||
{"error": "导出飞书数据失败"},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
else:
|
||
# CSV格式导出
|
||
try:
|
||
# 创建数据列表
|
||
data = []
|
||
for email, email_records in duplicate_emails.items():
|
||
for record in email_records:
|
||
fields = record.fields
|
||
row = {
|
||
'Email': email,
|
||
'RecordID': record.record_id
|
||
}
|
||
|
||
# 提取所有字段
|
||
for field_name, field_value in fields.items():
|
||
row[field_name] = extract_field_value(field_value)
|
||
|
||
data.append(row)
|
||
|
||
# 创建DataFrame并导出到CSV
|
||
df = pd.DataFrame(data)
|
||
df.to_csv(output_path, index=False, encoding='utf-8-sig')
|
||
excel_path = output_path
|
||
except Exception as e:
|
||
logger.error(f"导出CSV时出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
return Response(
|
||
{"error": f"导出CSV失败: {str(e)}"},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
else:
|
||
return Response(
|
||
{"error": f"不支持的导出类型: {export_type}，可选值为'creators'或'feishu'"},
|
||
status=status.HTTP_400_BAD_REQUEST
|
||
)
|
||
|
||
# 获取服务器域名，考虑各种情况
|
||
domain = request.build_absolute_uri('/').rstrip('/')
|
||
# 如果是本地开发环境，使用127.0.0.1:8000
|
||
if 'localhost' in domain or '127.0.0.1' in domain:
|
||
# 从请求头获取Host
|
||
host = request.META.get('HTTP_HOST', '127.0.0.1:8000')
|
||
if ':' not in host:
|
||
host = f"{host}:8000" # 添加默认端口
|
||
domain = f"http://{host}"
|
||
|
||
# 构建下载URL
|
||
file_url = f"{domain}/api/feishu/download/{output_filename}"
|
||
|
||
return Response({
|
||
"status": "success",
|
||
"message": f"成功导出{export_type}数据，格式为{export_format}",
|
||
"matched_emails": len(duplicate_emails),
|
||
"file_url": file_url,
|
||
"file_name": output_filename
|
||
}, status=status.HTTP_200_OK)
|
||
|
||
except Exception as e:
|
||
logger.error(f"导出数据时出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
return Response(
|
||
{"error": str(e)},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
)
|
||
|
||
@api_view(['GET'])
|
||
@permission_classes([IsAuthenticated])
|
||
def download_exported_file(request, filename):
|
||
"""
|
||
下载已导出的Excel文件
|
||
|
||
URL参数:
|
||
filename: 文件名
|
||
"""
|
||
try:
|
||
# 检查用户权限 - 只允许组长使用
|
||
if request.user.role != 'leader':
|
||
return Response(
|
||
{"error": "只有组长角色的用户可以使用此功能"},
|
||
status=status.HTTP_403_FORBIDDEN
|
||
)
|
||
|
||
# 构建文件路径
|
||
file_path = os.path.join(settings.MEDIA_ROOT, 'exports', filename)
|
||
|
||
# 检查文件是否存在
|
||
if not os.path.exists(file_path):
|
||
return Response(
|
||
{"error": f"文件不存在: {filename}"},
|
||
status=status.HTTP_404_NOT_FOUND
|
||
)
|
||
|
||
# 返回文件
|
||
response = FileResponse(open(file_path, 'rb'))
|
||
response['Content-Disposition'] = f'attachment; filename="{filename}"'
|
||
return response
|
||
|
||
except Exception as e:
|
||
logger.error(f"下载文件时出错: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
return Response(
|
||
{"error": str(e)},
|
||
status=status.HTTP_500_INTERNAL_SERVER_ERROR
|
||
) |