diff --git a/apps/gmail/views.py b/apps/gmail/views.py index 4efa10a..462e3ba 100644 --- a/apps/gmail/views.py +++ b/apps/gmail/views.py @@ -35,83 +35,32 @@ class GmailAuthInitiateView(APIView): API 视图，用于启动 Gmail OAuth2 认证流程。 """ permission_classes = [IsAuthenticated] # 限制访问，仅允许已认证用户 - def post(self, request): """ 处理 POST 请求，启动 Gmail OAuth2 认证并返回授权 URL。 - 支持两种方式提供客户端密钥: - 1. 在请求体中提供client_secret_json字段 - 2. 上传名为client_secret_file的JSON文件 + 直接使用系统中已配置的客户端密钥文件。 Args: - request: Django REST Framework 请求对象，包含客户端密钥 JSON 数据或文件。 + request: Django REST Framework 请求对象。 Returns: Response: 包含授权 URL 的 JSON 响应（成功时），或错误信息（失败时）。 Status Codes: 200: 成功生成授权 URL。 - 400: 请求数据无效。 500: 服务器内部错误（如认证服务失败）。 """ - logger.debug(f"Received auth initiate request: {request.data}") + logger.debug(f"Received auth initiate request from user {request.user.id}") - # 检查是否是文件上传方式 - client_secret_json = None - if 'client_secret_file' in request.FILES: - try: - # 读取上传的JSON文件内容 - client_secret_file = request.FILES['client_secret_file'] - client_secret_json = json.loads(client_secret_file.read().decode('utf-8')) - logger.info(f"从上传文件读取到客户端密钥JSON") - except json.JSONDecodeError as e: - logger.error(f"解析客户端密钥JSON文件失败: {str(e)}") - return Response({ - 'code': 400, - 'message': f'无效的JSON文件格式: {str(e)}', - 'data': None - }, status=status.HTTP_400_BAD_REQUEST) - except Exception as e: - logger.error(f"处理上传文件失败: {str(e)}") - return Response({ - 'code': 500, - 'message': f'处理上传文件失败: {str(e)}', - 'data': None - }, status=status.HTTP_500_INTERNAL_SERVER_ERROR) + # 直接使用系统中已配置的客户端密钥文件 + client_secret_path = 'media/secret/client_secret_266164728215-v84lngbp3vgr4ulql01sqkg5vaigf4a5.apps.googleusercontent.com.json' - # 如果不是文件上传，则尝试从请求数据中提取JSON - if not client_secret_json: - serializer = GmailCredentialSerializer(data=request.data, context={'request': request}) - if serializer.is_valid(): - try: - # 从请求数据中提取客户端密钥 JSON - client_secret_json = serializer.validated_data['client_secret_json'] - except Exception as e: - logger.error(f"未提供客户端密钥JSON: {str(e)}") - return Response({ - 'code': 400, - 'message': '请提供client_secret_json字段或上传client_secret_file文件', - 'data': None - }, status=status.HTTP_400_BAD_REQUEST) - else: - # 记录无效请求数据并返回错误响应 - logger.warning(f"Invalid request data: {serializer.errors}") - return Response({ - 'code': 400, - 'message': '请求数据无效，请提供client_secret_json字段或上传client_secret_file文件', - 'data': serializer.errors - }, status=status.HTTP_400_BAD_REQUEST) - - # 如果此时仍然没有client_secret_json，返回错误 - if not client_secret_json: - return Response({ - 'code': 400, - 'message': '请提供client_secret_json字段或上传client_secret_file文件', - 'data': None - }, status=status.HTTP_400_BAD_REQUEST) - try: + # 读取客户端密钥文件 + with open(client_secret_path, 'r') as f: + client_secret_json = json.load(f) + # 调用 GmailService 生成授权 URL auth_url = GmailService.initiate_authentication(request.user, client_secret_json) logger.info(f"Generated auth URL for user {request.user.id}") @@ -135,17 +84,14 @@ class GmailAuthCompleteView(APIView): API 视图，用于完成 Gmail OAuth2 认证流程。 """ permission_classes = [IsAuthenticated] # 限制访问，仅允许已认证用户 - def post(self, request): """ 处理 POST 请求，使用授权代码完成 Gmail OAuth2 认证并保存凭证。 - 支持两种方式提供客户端密钥: - 1. 在请求体中提供client_secret_json字段 - 2. 上传名为client_secret_file的JSON文件 + 直接使用系统中已配置的客户端密钥文件。 Args: - request: Django REST Framework 请求对象，包含授权代码和客户端密钥 JSON 或文件。 + request: Django REST Framework 请求对象，包含授权代码。 Returns: Response: 包含已保存凭证数据的 JSON 响应（成功时），或错误信息（失败时）。 @@ -155,72 +101,25 @@ class GmailAuthCompleteView(APIView): 400: 请求数据无效。 500: 服务器内部错误（如认证失败）。 """ - logger.debug(f"Received auth complete request: {request.data}") + logger.debug(f"Received auth complete request from user {request.user.id}") - # 检查是否是文件上传方式 - client_secret_json = None - if 'client_secret_file' in request.FILES: - try: - # 读取上传的JSON文件内容 - client_secret_file = request.FILES['client_secret_file'] - client_secret_json = json.loads(client_secret_file.read().decode('utf-8')) - logger.info(f"从上传文件读取到客户端密钥JSON") - except json.JSONDecodeError as e: - logger.error(f"解析客户端密钥JSON文件失败: {str(e)}") - return Response({ - 'code': 400, - 'message': f'无效的JSON文件格式: {str(e)}', - 'data': None - }, status=status.HTTP_400_BAD_REQUEST) - except Exception as e: - logger.error(f"处理上传文件失败: {str(e)}") - return Response({ - 'code': 500, - 'message': f'处理上传文件失败: {str(e)}', - 'data': None - }, status=status.HTTP_500_INTERNAL_SERVER_ERROR) + # 直接使用系统中已配置的客户端密钥文件 + client_secret_path = 'media/secret/client_secret_266164728215-v84lngbp3vgr4ulql01sqkg5vaigf4a5.apps.googleusercontent.com.json' - # 获取授权码，无论是哪种方式都需要 + # 获取授权代码 auth_code = request.data.get('auth_code') if not auth_code: return Response({ 'code': 400, - 'message': '必须提供授权码', - 'data': None - }, status=status.HTTP_400_BAD_REQUEST) - - # 如果不是文件上传，则尝试从请求数据中提取JSON - if not client_secret_json: - serializer = GmailCredentialSerializer(data=request.data, context={'request': request}) - if serializer.is_valid(): - try: - # 从请求数据中提取客户端密钥 JSON - client_secret_json = serializer.validated_data['client_secret_json'] - except Exception as e: - logger.error(f"未提供客户端密钥JSON: {str(e)}") - return Response({ - 'code': 400, - 'message': '请提供client_secret_json字段或上传client_secret_file文件', - 'data': None - }, status=status.HTTP_400_BAD_REQUEST) - else: - # 记录无效请求数据并返回错误响应 - logger.warning(f"Invalid request data: {serializer.errors}") - return Response({ - 'code': 400, - 'message': '请求数据无效，请提供client_secret_json字段或上传client_secret_file文件', - 'data': serializer.errors - }, status=status.HTTP_400_BAD_REQUEST) - - # 如果此时仍然没有client_secret_json，返回错误 - if not client_secret_json: - return Response({ - 'code': 400, - 'message': '请提供client_secret_json字段或上传client_secret_file文件', + 'message': '缺少必要的授权代码', 'data': None }, status=status.HTTP_400_BAD_REQUEST) try: + # 读取客户端密钥文件 + with open(client_secret_path, 'r') as f: + client_secret_json = json.load(f) + # 完成认证并保存凭证 credential = GmailService.complete_authentication(request.user, auth_code, client_secret_json) # 序列化凭证数据以返回 @@ -240,7 +139,6 @@ class GmailAuthCompleteView(APIView): 'data': None }, status=status.HTTP_500_INTERNAL_SERVER_ERROR) - class GmailCredentialViewSet(viewsets.ModelViewSet): """ Gmail凭证管理视图集，提供对Gmail账户凭证的完整CRUD操作 diff --git a/daren_project/settings.py b/daren_project/settings.py index fa4327f..45fdb22 100644 --- a/daren_project/settings.py +++ b/daren_project/settings.py @@ -28,7 +28,7 @@ SECRET_KEY = 'django-insecure-aie+z75u&tnnx8@g!2ie+q)qhq1!eg&ob!c1(e1vr!eclh+xv6 # SECURITY WARNING: don't run with debug turned on in production! DEBUG = True -ALLOWED_HOSTS = ['localhost', '127.0.0.1', '325a-180-159-100-165.ngrok-free.app'] +ALLOWED_HOSTS = ['*'] # Application definition @@ -180,7 +180,7 @@ AUTH_USER_MODEL = 'accounts.User' API_BASE_URL = 'http://81.69.223.133:48329' SILICON_CLOUD_API_KEY = 'sk-xqbujijjqqmlmlvkhvxeogqjtzslnhdtqxqgiyuhwpoqcjvf' -GMAIL_WEBHOOK_URL = 'https://325a-180-159-100-165.ngrok-free.app/api/gmail/webhook/' +GMAIL_WEBHOOK_URL = 'https://7403-180-159-100-165.ngrok-free.app/api/gmail/webhook/' APPLICATION_ID = 'd5d11efa-ea9a-11ef-9933-0242ac120006'