Rest Framework 토큰을 사용하여 Django-Tenant 하위 도메인 로그인 오류 해결

최적화된 스키마 쿼리를 사용하여 Django-Tenants의 하위 도메인 로그인 문제 처리

명시적인 스키마 선택 및 오류 처리 기능을 갖춘 Django ORM을 사용하는 백엔드 솔루션입니다.

# Import necessary libraries
from django.db import connection
from rest_framework.authtoken.models import Token
from django.contrib.auth import authenticate, login
from django.http import JsonResponse
from django_tenants.utils import schema_context
from django.views.decorators.csrf import csrf_exempt
@csrf_exempt
def dual_login_view(request):
    """Handle login for multi-tenant subdomains with correct schema."""
    if request.method == "POST":
        username = request.POST.get("login")
        password = request.POST.get("password")
        tenant_schema_name = connection.tenant.schema_name
        try:
            # Switch to the correct tenant schema
            with schema_context(tenant_schema_name):
                user = authenticate(request, username=username, password=password)
                if user is not None:
                    login(request, user)
                    # Generate or retrieve token
                    token, created = Token.objects.get_or_create(user=user)
                    return JsonResponse({"status": "success", "token": token.key})
                else:
                    return JsonResponse({"status": "error", "message": "Invalid credentials"}, status=400)
        except Exception as e:
            return JsonResponse({"status": "error", "message": str(e)}, status=500)
    return JsonResponse({"status": "error", "message": "Invalid request method"}, status=405)

테넌트 인식 스키마를 사용한 명시적 토큰 관리

다중 테넌트 아키텍처의 로그인을 위한 모듈화되고 재사용 가능한 Django API 보기입니다.

from rest_framework.views import APIView
from rest_framework.response import Response
from rest_framework import status
from django.contrib.auth import authenticate
from rest_framework.authtoken.models import Token
from django_tenants.utils import schema_context
class TenantAwareLoginAPIView(APIView):
    """Login endpoint that ensures tenant-aware schema handling."""
    def post(self, request):
        username = request.data.get("username")
        password = request.data.get("password")
        tenant_schema_name = request.tenant.schema_name
        if not username or not password:
            return Response({"error": "Username and password required"}, status=status.HTTP_400_BAD_REQUEST)
        try:
            with schema_context(tenant_schema_name):
                user = authenticate(request, username=username, password=password)
                if user is None:
                    return Response({"error": "Invalid credentials"}, status=status.HTTP_401_UNAUTHORIZED)
                # Generate or retrieve token for the user
                token, created = Token.objects.get_or_create(user=user)
                return Response({"token": f"Token {token.key}"}, status=status.HTTP_200_OK)
        except Exception as e:
            return Response({"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)

하위 도메인 로그인 요청 처리를 위한 프런트엔드 스크립트

양식 제출을 처리하고 테넌트 하위 도메인에 대한 토큰 기반 로그인을 처리하는 JavaScript 솔루션입니다.

<script>
document.querySelector('form').addEventListener('submit', function(event) {
    event.preventDefault();
    let form = event.target;
    let formData = new FormData(form);
    fetch("{% url 'tenant_aware_login' %}", {
        method: 'POST',
        body: JSON.stringify(Object.fromEntries(formData)),
        headers: {
            'Content-Type': 'application/json',
            'X-CSRFToken': formData.get('csrfmiddlewaretoken')
        }
    })
    .then(response => {
        if (!response.ok) throw new Error('Server Error');
        return response.json();
    })
    .then(data => {
        if (data.token) {
            localStorage.setItem('token', data.token);
            window.location.href = '/';
        } else {
            Swal.fire({
                icon: 'error',
                title: 'Login Failed',
                text: data.error || 'Invalid credentials'
            });
        }
    })
    .catch(error => {
        console.error('Error:', error);
    });
});
</script>

스키마 인식 토큰 인증을 확인하기 위한 단위 테스트

API가 스키마 전환을 올바르게 처리하는지 확인하기 위한 Python의 단위 테스트입니다.

from django.test import TestCase
from rest_framework.test import APIClient
from django_tenants.utils import schema_context
from django.contrib.auth.models import User
from rest_framework.authtoken.models import Token
class TenantLoginTest(TestCase):
    def setUp(self):
        self.client = APIClient()
        with schema_context('test_tenant'):  # Switch to tenant schema
            self.user = User.objects.create_user(username='testuser', password='testpass')
    def test_successful_login(self):
        with schema_context('test_tenant'):
            response = self.client.post('/api/login/', {
                'username': 'testuser',
                'password': 'testpass'
            })
            self.assertEqual(response.status_code, 200)
            self.assertIn('token', response.json())
    def test_invalid_login(self):
        with schema_context('test_tenant'):
            response = self.client.post('/api/login/', {
                'username': 'wronguser',
                'password': 'wrongpass'
            })
            self.assertEqual(response.status_code, 401)
            self.assertIn('error', response.json())

다중 테넌트 Django 로그인 문제에 대한 일반적인 질문

1. 원인은 무엇입니까? 500 내부 서버 오류 로그인 중에?
2. 오류가 발생하는 이유는 Token.objects.get_or_create() 잘못된 스키마를 쿼리하여 사용자 레코드를 조회할 때 불일치가 발생합니다.
3. 토큰 쿼리가 올바른 테넌트 스키마를 가리키는지 어떻게 확인하나요?
4. 사용 schema_context() 에서 django-세입자 쿼리 실행을 래핑하고 올바른 스키마로 전환하는 라이브러리입니다.
5. 관리자 패널 로그인은 작동하지만 사용자 로그인은 실패하는 이유는 무엇입니까?
6. Django 관리자는 스키마 컨텍스트를 자동으로 조정하지만 사용자 정의 보기는 authenticate() 또는 Token.objects 명시적으로 구성하지 않는 한 그렇지 않을 수 있습니다.
7. 프런트엔드에서 로그인 토큰을 어떻게 검색하고 저장합니까?
8. 가져오기 API를 사용하여 자격 증명을 보낸 다음 다음을 사용하여 응답 토큰을 저장합니다. localStorage.setItem() 지속적인 인증을 위해.
9. 로그인 실패에 대한 오류 메시지를 더 잘 표시하려면 어떻게 해야 합니까?
10. 다음과 같은 라이브러리를 사용하여 프런트엔드 알림을 구현합니다. SweetAlert2 사용자에게 잘못된 자격 증명이나 서버 문제를 알리기 위해.

테넌트 하위 도메인 전체에서 원활한 로그인 보장

Django 다중 테넌트 앱에서 로그인 실패를 해결하려면 모든 데이터베이스 쿼리가 적절한 스키마에서 작동하는지 확인해야 합니다. 스키마 컨텍스트와 같은 도구를 명시적으로 사용하면 스키마 충돌을 방지하면서 올바른 테넌트 데이터베이스에서 사용자 토큰을 가져오도록 보장할 수 있습니다.

사용자가 하위 도메인에서만 로그인 실패에 직면하는 SaaS 플랫폼에서 작업한다고 상상해 보십시오. 적절한 스키마 전환을 통해 이러한 문제가 해결되어 원활한 인증이 보장됩니다. 이 수정 사항을 채택하면 개선될 뿐만 아니라 사용자 경험 또한 각 테넌트에 대해 안전하고 효율적인 데이터 액세스를 보장합니다. 🔧