使用 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>

用于验证架构感知令牌身份验证的单元测试

Python 中的单元测试可确保 API 正确处理架构切换。

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. 使用 fetch API 发送凭据，然后使用存储响应令牌 localStorage.setItem() 用于持久身份验证。
9. 如何更好地显示登录失败的错误消息？
10. 使用类似的库实现前端警报 SweetAlert2 通知用户凭据不正确或服务器问题。

确保跨租户子域顺利登录

解决 Django 多租户应用程序中的登录失败问题需要确保所有数据库查询都在正确的架构中运行。通过显式使用架构上下文等工具，我们可以保证从正确的租户数据库中获取用户令牌，从而避免架构冲突。

想象一下，在 SaaS 平台上工作，用户仅在子域上面临登录失败。通过正确的架构切换，这些问题都可以得到解决，从而确保无缝身份验证。采用此修复不仅可以改善 用户体验 还保证每个租户安全、高效的数据访问。 🔧