регистрация компании

apps/users/admin.py

@@ -39,6 +39,7 @@ class CustomUserAdmin(UserAdmin):
         "username",
         "email",
         "is_company",
+        "is_active",
     )
     list_filter = ("is_company",)
     empty_value_display = "-empty-"

apps/users/migrations/0002_alter_customuser_managers.py

@@ -0,0 +1,20 @@
+# Generated by Django 4.1 on 2023-08-14 16:43
+
+from django.db import migrations
+
+import apps.users.models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("users", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AlterModelManagers(
+            name="customuser",
+            managers=[
+                ("objects", apps.users.models.CustomUserManager()),
+            ],
+        ),
+    ]

apps/users/models.py

@@ -1,8 +1,9 @@
 import re
 
-from django.contrib.auth.models import AbstractUser
+from django.contrib.auth.models import AbstractUser, UserManager
 from django.core.exceptions import ValidationError
 from django.db import models
+from django.db.models import Q
 from django.utils.translation import gettext_lazy as _
 
 
@@ -128,6 +129,11 @@ def __str__(self):
         return f"{self.first_name} {self.last_name}"
 
 
+class CustomUserManager(UserManager):
+    def get_companies(self):
+        return self.filter(Q(is_company=True) & Q(is_active=True))
+
+
 class CustomUser(AbstractUser):
     email = models.EmailField(unique=True, blank=False, max_length=254, verbose_name="Email")
     username = models.CharField(
@@ -154,13 +160,16 @@ class CustomUser(AbstractUser):
         on_delete=models.SET_NULL,
     )
 
+    objects = CustomUserManager()
+
     class Meta:
         swappable = "AUTH_USER_MODEL"
         ordering = ("username",)
         verbose_name = _("User")
         verbose_name_plural = _("Users")
 
     def clean(self):
+        super().clean()
         if self.is_company and not self.company:
             raise ValidationError(_("Company field is required for companies!"))
         if not (self.is_superuser or self.is_staff) and not (self.is_company or self.personal):
@@ -178,3 +187,11 @@ def save(self, *args, **kwargs):
 
     def __str__(self):
         return self.username
+
+    def delete(self):
+        """Предотвращает удаление модели.
+
+        Вместо непосредственного удаления, помечает запись удалённой (is_active=True).
+        """
+        self.is_active = False
+        self.save()

apps/users/serializers/addresses.py

@@ -0,0 +1,11 @@
+from rest_framework import serializers
+
+from apps.users.models import Address
+
+
+class AddressSerializer(serializers.ModelSerializer):
+    """Сериализатор для получения данных оо адресе."""
+
+    class Meta:
+        model = Address
+        fields = ("id", "address")

apps/users/serializers/base.py

@@ -0,0 +1,58 @@
+from django.contrib.auth import get_user_model
+from django.db import transaction
+from djoser.conf import settings
+from rest_framework import serializers
+
+from apps.users.models import Address, PhoneNumber
+
+User = get_user_model()
+
+
+class BaseSerializer(serializers.ModelSerializer):
+    """Абстрактный сериализатор для создания нового пользователя."""
+
+    password = serializers.CharField(style={"input_type": "password"}, write_only=True)
+
+    def perform_create(self, validated_data, **fields):
+        """Создает пользователя."""
+        with transaction.atomic():
+            user = User.objects.create_user(**validated_data, **fields)
+            if settings.SEND_ACTIVATION_EMAIL:
+                user.is_active = False
+                user.save(update_fields=["is_active"])
+        return user
+
+    def _address_phone_attach(self, obj):
+        """Создает объекты адреса и номера телефона."""
+        address = obj.pop("address", None)
+        phone_number = obj.pop("phone_number", None)
+
+        address_obj = Address.objects.create(**address)
+        phone_number_obj = PhoneNumber.objects.create(**phone_number)
+
+        return {"address": address_obj, "phone_number": phone_number_obj}
+
+    def _extract_relations(self, validated_data):
+        """Извлекает один уровень вложенных отношений."""
+        reverse_relations = {}
+        for field_name, field in self.fields.items():
+            if isinstance(field, serializers.ModelSerializer | serializers.ListSerializer):
+                if field.source not in validated_data:
+                    continue
+
+                reverse_relations[field_name] = (
+                    self.fields[field_name].Meta.model,
+                    validated_data.pop(field_name),
+                )
+        return reverse_relations
+
+    def update_or_create(self, validated_data):
+        """Создает объекты компании или физ. лица."""
+        reverse_relations = {}
+        relations = self._extract_relations(validated_data)
+
+        for field_name, (model, data) in relations.items():
+            with transaction.atomic():
+                address_phone = self._address_phone_attach(data)
+                reverse_relations[field_name] = model.objects.create(**data, **address_phone)
+        return reverse_relations

apps/users/serializers/companies.py

@@ -0,0 +1,95 @@
+from django.contrib.auth import get_user_model
+from django.db import transaction
+from rest_framework import serializers
+
+from apps.users.models import Company
+from apps.users.serializers.addresses import AddressSerializer
+from apps.users.serializers.base import BaseSerializer
+from apps.users.serializers.phonenumbers import PhoneNumberSerializer
+
+User = get_user_model()
+
+
+class CompanyReadSerializer(serializers.ModelSerializer):
+    """Сериализатор для получения и отображения данных о компании.
+
+    Используется в безопасных http-методах.
+    """
+
+    class Meta:
+        model = Company
+        depth = 1
+        fields = (
+            "id",
+            "role",
+            "name",
+            "company_account",
+            "inn",
+            "ogrn",
+            "phone_number",
+            "address",
+        )
+
+
+class CompanyWriteSerializer(serializers.ModelSerializer):
+    """Сериализатор для создания компании."""
+
+    address = AddressSerializer()
+    phone_number = PhoneNumberSerializer()
+
+    class Meta:
+        model = Company
+        fields = (
+            "role",
+            "name",
+            "inn",
+            "phone_number",
+            "address",
+        )
+
+    def to_representation(self, instance):
+        serializer = CompanyReadSerializer(instance)
+        return serializer.data
+
+
+class UserReadSerializer(serializers.ModelSerializer):
+    """Сериализатор для получения и отображения данных о пользователях.
+
+    Используется в безопасных http-методах.
+    """
+
+    company = CompanyReadSerializer()
+
+    class Meta:
+        model = User
+        fields = (
+            "id",
+            "email",
+            "username",
+            "is_company",
+            "company",
+        )
+
+
+class CompanySerializer(BaseSerializer):
+    """Сериализатор для создания пользователя-компании."""
+
+    company = CompanyWriteSerializer()
+
+    class Meta:
+        model = User
+        fields = (
+            "email",
+            "username",
+            "password",
+            "company",
+        )
+
+    @transaction.atomic
+    def create(self, validated_data):
+        relations = self.update_or_create(validated_data)
+        return self.perform_create(validated_data, **relations, is_company=True)
+
+    def to_representation(self, instance):
+        serializer = UserReadSerializer(instance)
+        return serializer.data

apps/users/serializers/phonenumbers.py

@@ -0,0 +1,11 @@
+from rest_framework import serializers
+
+from apps.users.models import PhoneNumber
+
+
+class PhoneNumberSerializer(serializers.ModelSerializer):
+    """Сериализатор для получения данных о номере телефона."""
+
+    class Meta:
+        model = PhoneNumber
+        fields = ("id", "phone_number")

apps/users/tests/conftest.py

@@ -0,0 +1,38 @@
+import pytest
+from rest_framework.test import APIClient
+
+from apps.users.models import Address, Company, PhoneNumber
+
+
+@pytest.fixture
+def apiclient():
+    return APIClient()
+
+
+@pytest.fixture
+def company(django_user_model):
+    phone_number = PhoneNumber.objects.create(phone_number="1234567")
+    address = Address.objects.create(address="earth")
+    company_obj = Company.objects.create(
+        role="supplier",
+        name="best_company",
+        company_account="12345678901234567890",
+        inn="1234567890",
+        ogrn="1234567890123",
+        phone_number=phone_number,
+        address=address,
+    )
+    return django_user_model.objects.create_user(
+        email="[email protected]",
+        username="companyuser",
+        password="12345678",
+        is_company=True,
+        company=company_obj,
+    )
+
+
+@pytest.fixture
+def company_client(company):
+    client = APIClient()
+    client.force_authenticate(user=company)
+    return client