Разрешение ошибок входа в поддомен Django-Tenant с помощью токенов Rest Framework

Обработка проблем со входом в поддомен в арендаторах Django с помощью запросов оптимизированной схемы

Бэкэнд-решение с использованием 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)

Явное управление токенами с использованием схем с учетом арендаторов

Модульное и многократно используемое представление API Django для входа в систему в многопользовательской архитектуре.

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() из Джанго-арендаторы библиотека для завершения выполнения запроса и переключения на правильную схему.
5. Почему вход в панель администратора работает, но вход пользователя невозможен?
6. Администратор Django автоматически настраивает контексты схемы, но пользовательские представления с помощью authenticate() или Token.objects не может быть, если это явно не настроено.
7. Как получить и сохранить токен входа во внешнем интерфейсе?
8. Используйте API-интерфейс выборки для отправки учетных данных, а затем сохраните токен ответа, используя localStorage.setItem() для постоянной аутентификации.
9. Как я могу отображать более качественные сообщения об ошибках при неудачном входе в систему?
10. Реализуйте оповещения интерфейса, используя такие библиотеки, как SweetAlert2 для уведомления пользователей о неправильных учетных данных или проблемах с сервером.

Обеспечение плавного входа в поддомены арендаторов

Для устранения ошибок входа в мультитенантные приложения Django необходимо убедиться, что все запросы к базе данных выполняются в правильной схеме. Явно используя такие инструменты, как контекст схемы, мы можем гарантировать, что токены пользователей извлекаются из правильной базы данных арендатора, избегая конфликтов схемы.

Представьте себе, что вы работаете на платформе SaaS, где пользователи сталкиваются с ошибками входа только на поддоменах. При правильном переключении схемы эти проблемы решаются, обеспечивая бесперебойную аутентификацию. Принятие этого исправления не только улучшает пользовательский опыт но также гарантирует безопасный и эффективный доступ к данным для каждого арендатора. 🔧