django-fixture-utils 0.1.0__tar.gz

django_fixture_utils-0.1.0/PKG-INFO

@@ -0,0 +1,13 @@
+Metadata-Version: 2.4
+Name: django-fixture-utils
+Version: 0.1.0
+Summary: Utilities for Django fixtures, tests and database setup
+Author: Student
+Requires-Python: >=3.11
+Requires-Dist: Django<6,>=5
+Requires-Dist: djangorestframework<4,>=3.15
+Requires-Dist: django-filter<26,>=24
+Requires-Dist: psycopg2-binary<3,>=2.9
+Requires-Dist: python-dotenv<2,>=1
+Requires-Dist: pytest<9,>=8
+Requires-Dist: pytest-django<5,>=4

django_fixture_utils-0.1.0/api/admin.py

@@ -0,0 +1,64 @@
+from django.contrib import admin
+from .models import Location, LaundryRoom, WashingMachine, WashOrder
+from .signals import sync_room
+
+
+class MachineInline(admin.TabularInline):
+    model = WashingMachine
+    extra = 0
+    readonly_fields = ("machine_id", "status_changed_at")
+
+
+@admin.register(LaundryRoom)
+class LaundryRoomAdmin(admin.ModelAdmin):
+    list_display = ("name", "location", "number_free_machines", "created_at")
+    list_filter = ("location",)
+    search_fields = ("name",)
+    readonly_fields = ("number_free_machines",)
+    inlines = [MachineInline]
+
+    def save_formset(self, request, form, formset, change):
+        super().save_formset(request, form, formset, change)
+        for obj in formset.deleted_objects + formset.saved_objects:
+            sync_room(obj.room_id)
+
+
+@admin.register(WashingMachine)
+class WashingMachineAdmin(admin.ModelAdmin):
+    list_display = (
+        "machine_id",
+        "get_status_display",
+        "max_load_kg",
+        "room",
+        "building",
+        "status_changed_at",
+    )
+    list_filter = ("status", "room")
+    readonly_fields = ("status_changed_at",)
+    actions = ["to_service"]
+
+    @admin.action(description="В тех. обслуживание")
+    def to_service(self, request, queryset):
+        rooms = set(queryset.values_list("room_id", flat=True))
+        queryset.update(status="2")
+        for r_id in rooms:
+            sync_room(r_id)
+
+
+@admin.register(WashOrder)
+class WashOrderAdmin(admin.ModelAdmin):
+    list_display = (
+        "title",
+        "get_status_display",
+        "laundry_weight",
+        "user",
+        "machine",
+        "building",
+        "created_at",
+    )
+    list_filter = ("status", "building")
+    search_fields = ("title", "user__email")
+    date_hierarchy = "created_at"
+
+
+admin.site.register(Location)

django_fixture_utils-0.1.0/api/apps.py

@@ -0,0 +1,9 @@
+from django.apps import AppConfig
+
+
+class ApiConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "api"
+
+    def ready(self):
+        import api.signals

django_fixture_utils-0.1.0/api/filters.py

@@ -0,0 +1,13 @@
+import django_filters
+
+from .models import WashingMachine
+
+
+class WashingMachineFilter(django_filters.FilterSet):
+    status = django_filters.CharFilter(field_name="status")
+    room_id = django_filters.UUIDFilter(field_name="room_id")
+    min_load = django_filters.NumberFilter(field_name="max_load_kg", lookup_expr="gte")
+
+    class Meta:
+        model = WashingMachine
+        fields = ["status", "room_id", "min_load"]

django_fixture_utils-0.1.0/api/migrations/0001_initial.py

@@ -0,0 +1,84 @@
+# Generated by Django 6.0.6 on 2026-06-30 07:14
+
+import django.core.validators
+import django.db.models.deletion
+import uuid
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Location',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('title', models.CharField(db_index=True, max_length=125, unique=True)),
+                ('floor_count', models.IntegerField(validators=[django.core.validators.MinValueValidator(0)])),
+                ('address', models.CharField(max_length=125)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='LaundryRoom',
+            fields=[
+                ('room_id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('name', models.CharField(max_length=125, unique=True)),
+                ('number_free_machines', models.IntegerField(db_index=True, default=0, validators=[django.core.validators.MinValueValidator(0)])),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('location', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='api.location')),
+            ],
+        ),
+        migrations.CreateModel(
+            name='WashingMachine',
+            fields=[
+                ('machine_id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('max_load_kg', models.IntegerField(validators=[django.core.validators.MinValueValidator(0)])),
+                ('status', models.CharField(choices=[('0', 'Свободна'), ('1', 'Стирает'), ('2', 'Тех. обслуживание')], db_index=True, default='0', max_length=1)),
+                ('status_changed_at', models.DateTimeField(auto_now=True)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('building', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='machine_buildings', to='api.location')),
+                ('pickup_point', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='machine_pickup_points', to='api.location')),
+                ('room', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='api.laundryroom')),
+            ],
+        ),
+        migrations.CreateModel(
+            name='WashOrder',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('title', models.CharField(max_length=125, unique=True)),
+                ('laundry_weight', models.IntegerField(validators=[django.core.validators.MinValueValidator(0)])),
+                ('idempotency_key', models.CharField(blank=True, max_length=256, null=True, unique=True)),
+                ('status', models.CharField(choices=[('0', 'created'), ('1', 'assigned'), ('2', 'completed'), ('3', 'cancelled')], default='0', max_length=1)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('building', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='order_buildings', to='api.location')),
+                ('machine', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='order_machines', to='api.washingmachine')),
+                ('pickup_point', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='order_pickup_points', to='api.location')),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+        migrations.AddField(
+            model_name='washingmachine',
+            name='order',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='machine_orders', to='api.washorder'),
+        ),
+        migrations.AddIndex(
+            model_name='laundryroom',
+            index=models.Index(fields=['location', 'name'], name='room_location_name_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='washorder',
+            index=models.Index(fields=['status', 'created_at'], name='order_status_created_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='washingmachine',
+            index=models.Index(fields=['room', 'status'], name='machine_room_status_idx'),
+        ),
+    ]

django_fixture_utils-0.1.0/api/migrations/0002_alter_laundryroom_room_id_and_more.py

@@ -0,0 +1,30 @@
+# Generated by Django 6.0.6 on 2026-06-30 17:04
+
+import django.core.validators
+import uuid
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('api', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='laundryroom',
+            name='room_id',
+            field=models.UUIDField(default=uuid.uuid4, primary_key=True, serialize=False),
+        ),
+        migrations.AlterField(
+            model_name='washingmachine',
+            name='max_load_kg',
+            field=models.IntegerField(validators=[django.core.validators.MinValueValidator(1)]),
+        ),
+        migrations.AlterField(
+            model_name='washorder',
+            name='laundry_weight',
+            field=models.IntegerField(validators=[django.core.validators.MinValueValidator(1)]),
+        ),
+    ]

django_fixture_utils-0.1.0/api/models.py

@@ -0,0 +1,139 @@
+from django.db import models
+from django.core.validators import MinValueValidator
+from django.contrib.auth import get_user_model
+from rest_framework.exceptions import ValidationError
+from uuid import uuid4
+
+User = get_user_model()
+
+
+class Location(models.Model):
+    title = models.CharField(max_length=125, unique=True, db_index=True)
+    floor_count = models.IntegerField(validators=[MinValueValidator(0)])
+    address = models.CharField(max_length=125)
+    created_at = models.DateTimeField(auto_now_add=True)
+
+    def __str__(self):
+        return self.title
+
+
+class LaundryRoom(models.Model):
+    room_id = models.UUIDField(default=uuid4, primary_key=True)
+    name = models.CharField(max_length=125, unique=True)
+    location = models.ForeignKey(Location, on_delete=models.CASCADE)
+    number_free_machines = models.IntegerField(
+        default=0, validators=[MinValueValidator(0)], db_index=True
+    )
+    created_at = models.DateTimeField(auto_now_add=True)
+
+    class Meta:
+        indexes = [
+            models.Index(fields=["location", "name"], name="room_location_name_idx")
+        ]
+
+    def __str__(self):
+        return self.name
+
+
+class WashingMachine(models.Model):
+    class STATUS(models.TextChoices):
+        FREE = "0", "Свободна"
+        LAUNDRY = "1", "Стирает"
+        SERVICE = "2", "Тех. обслуживание"
+
+    machine_id = models.UUIDField(default=uuid4, primary_key=True, editable=False)
+    max_load_kg = models.IntegerField(validators=[MinValueValidator(1)])
+    status = models.CharField(
+        max_length=1, choices=STATUS.choices, default=STATUS.FREE, db_index=True
+    )
+    order = models.ForeignKey(
+        "WashOrder",
+        null=True,
+        blank=True,
+        on_delete=models.SET_NULL,
+        related_name="machine_orders",
+    )
+    room = models.ForeignKey(LaundryRoom, on_delete=models.CASCADE, db_index=True)
+    building = models.ForeignKey(
+        Location, on_delete=models.CASCADE, related_name="machine_buildings"
+    )
+    pickup_point = models.ForeignKey(
+        Location, on_delete=models.CASCADE, related_name="machine_pickup_points"
+    )
+    status_changed_at = models.DateTimeField(auto_now=True)
+    created_at = models.DateTimeField(auto_now_add=True)
+
+    class Meta:
+        indexes = [
+            models.Index(fields=["room", "status"], name="machine_room_status_idx")
+        ]
+
+    def _set_status(self, new_status, label):
+        if self.status == new_status:
+            raise ValidationError(f"Статус уже '{label}'")
+        self.status = new_status
+        self.save(update_fields=["status", "status_changed_at"])
+
+    def mark_washing(self):
+        self._set_status(self.STATUS.LAUNDRY, "Стирает")
+
+    def mark_free(self):
+        self._set_status(self.STATUS.FREE, "Свободна")
+
+        if self.order_id:
+            WashOrder.objects.filter(pk=self.order_id).update(
+                status=WashOrder.STATUS.COMPLETED
+            )
+            self.order = None
+            self.save(update_fields=["order"])
+
+    def mark_service(self):
+        self._set_status(self.STATUS.SERVICE, "Тех. обслуживание")
+
+    def can_take_laundry(self, weight: int) -> bool:
+        return self.max_load_kg >= weight
+
+    def __str__(self):
+        return str(self.machine_id)
+
+
+class WashOrder(models.Model):
+    class STATUS(models.TextChoices):
+        CREATED = "0", "created"
+        ASSIGNED = "1", "assigned"
+        COMPLETED = "2", "completed"
+        CANCELLED = "3", "cancelled"
+
+    title = models.CharField(max_length=125, unique=True)
+    laundry_weight = models.IntegerField(validators=[MinValueValidator(1)])
+    user = models.ForeignKey(User, on_delete=models.CASCADE, db_index=True)
+    machine = models.ForeignKey(
+        WashingMachine,
+        on_delete=models.SET_NULL,
+        null=True,
+        blank=True,
+        related_name="order_machines",
+    )
+    building = models.ForeignKey(
+        Location, on_delete=models.CASCADE, related_name="order_buildings"
+    )
+    pickup_point = models.ForeignKey(
+        Location, on_delete=models.CASCADE, related_name="order_pickup_points"
+    )
+    idempotency_key = models.CharField(
+        max_length=256, unique=True, null=True, blank=True
+    )
+    status = models.CharField(
+        max_length=1, choices=STATUS.choices, default=STATUS.CREATED
+    )
+    created_at = models.DateTimeField(auto_now_add=True)
+
+    class Meta:
+        indexes = [
+            models.Index(
+                fields=["status", "created_at"], name="order_status_created_idx"
+            )
+        ]
+
+    def __str__(self):
+        return self.title

django_fixture_utils-0.1.0/api/serializers.py

@@ -0,0 +1,94 @@
+from rest_framework import serializers
+from .models import Location, LaundryRoom, WashingMachine, WashOrder
+
+
+def validate_locations(building, pickup_point):
+    if building and pickup_point and building.pk == pickup_point.pk:
+        raise serializers.ValidationError("building and pickup_point must differ")
+    
+def validate_id(building_id, pickup_point_id):
+        if building_id == pickup_point_id:
+            raise serializers.ValidationError("building and pickup_point must differ")
+        return True
+
+
+class LocationSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Location
+        fields = "__all__"
+        read_only_fields = ["id", "created_at"]
+
+
+class LaundryRoomSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = LaundryRoom
+        fields = "__all__"
+        read_only_fields = ["room_id", "number_free_machines", "created_at"]
+
+
+class WashingMachineSerializer(serializers.ModelSerializer):
+    status_label = serializers.CharField(source="get_status_display", read_only=True)
+
+    class Meta:
+        model = WashingMachine
+        fields = "__all__"
+        read_only_fields = ["machine_id", "status", "status_changed_at", "created_at"]
+
+    def validate(self, attrs):
+        building = attrs.get("building") or getattr(self.instance, "building", None)
+        pickup = attrs.get("pickup_point") or getattr(
+            self.instance, "pickup_point", None
+        )
+        validate_locations(building, pickup)
+        return attrs
+
+
+class WashOrderSerializer(serializers.ModelSerializer):
+    building_id = serializers.PrimaryKeyRelatedField(
+        queryset=Location.objects.all(), source="building", write_only=True
+    )
+    pickup_point_id = serializers.PrimaryKeyRelatedField(
+        queryset=Location.objects.all(), source="pickup_point", write_only=True
+    )
+
+    class Meta:
+        model = WashOrder
+        fields = "__all__"
+        read_only_fields = [
+            "id",
+            "user",
+            "machine",
+            "building",
+            "pickup_point",
+            "idempotency_key",
+            "status",
+            "created_at",
+        ]
+
+    def validate(self, attrs):
+        validate_locations(attrs.get("building"), attrs.get("pickup_point"))
+        weight = attrs.get("laundry_weight")
+        if weight is not None and weight <= 0:
+            raise serializers.ValidationError(
+                {"laundry_weight": ["Must be greater than 0"]}
+            )
+        return attrs
+
+
+class AllocateSerializer(serializers.Serializer):
+    laundry_weight = serializers.IntegerField(min_value=1)
+    building_id = serializers.IntegerField()
+    pickup_point_id = serializers.IntegerField()
+    
+    def validate(self, attrs):
+        validate_id(attrs["building_id"], attrs["pickup_point_id"])
+        return attrs
+
+
+class MachineWashingSerializer(serializers.Serializer):
+    building_id = serializers.IntegerField()
+    pickup_point_id = serializers.IntegerField()
+
+    def validate(self, attrs):
+        validate_id(attrs["building_id"], attrs["pickup_point_id"])
+        return attrs

django_fixture_utils-0.1.0/api/services.py

@@ -0,0 +1,85 @@
+from django.db import transaction
+from django.shortcuts import get_object_or_404
+from rest_framework.exceptions import ValidationError
+
+from config.exceptions import NoFreeMachines
+from config.logging_utils import log_event
+from .models import Location, WashingMachine, WashOrder
+
+
+@transaction.atomic
+def allocate_machine(laundry_weight, building_id, pickup_point_id):
+    if laundry_weight <= 0:
+        raise ValidationError({"laundry_weight": ["Must be greater than 0"]})
+
+    if building_id == pickup_point_id:
+        raise ValidationError(
+            {"building_id": ["building and pickup_point must differ"]}
+        )
+
+    building = get_object_or_404(Location, pk=building_id)
+    pickup_point = get_object_or_404(Location, pk=pickup_point_id)
+
+    machine = (
+        WashingMachine.objects.select_for_update()
+        .select_related("room")
+        .filter(
+            status=WashingMachine.STATUS.FREE,
+            max_load_kg__gte=laundry_weight,
+        )
+        .order_by("-room__number_free_machines", "status_changed_at")
+        .first()
+    )
+
+    if not machine:
+        log_event("allocate_conflict", laundry_weight=laundry_weight)
+        raise NoFreeMachines()
+
+    order = (
+        WashOrder.objects.select_for_update()
+        .filter(
+            status=WashOrder.STATUS.CREATED,
+            machine__isnull=True,
+            laundry_weight=laundry_weight,
+            building_id=building_id,
+            pickup_point_id=pickup_point_id,
+        )
+        .order_by("created_at")
+        .first()
+    )
+
+    machine.status = WashingMachine.STATUS.LAUNDRY
+    machine.building = building
+    machine.pickup_point = pickup_point
+
+    if order:
+        machine.order = order
+        order.machine = machine
+        order.status = WashOrder.STATUS.ASSIGNED
+        order.save(update_fields=["machine", "status"])
+        machine.save(
+            update_fields=[
+                "status",
+                "status_changed_at",
+                "building",
+                "pickup_point",
+                "order",
+            ]
+        )
+    else:
+        machine.save(
+            update_fields=[
+                "status",
+                "status_changed_at",
+                "building",
+                "pickup_point",
+            ]
+        )
+
+    log_event(
+        "allocate",
+        machine_id=str(machine.machine_id),
+        order_id=order.id if order else None,
+        laundry_weight=laundry_weight,
+    )
+    return machine, order

django_fixture_utils-0.1.0/api/signals.py

@@ -0,0 +1,19 @@
+from django.db.models.signals import post_save, post_delete
+from django.dispatch import receiver
+from .models import WashingMachine, LaundryRoom
+
+
+def sync_room(room_id):
+    if room_id:
+        cnt = WashingMachine.objects.filter(room_id=room_id, status="0").count()
+        LaundryRoom.objects.filter(pk=room_id).update(number_free_machines=cnt)
+
+
+@receiver(post_save, sender=WashingMachine)
+def machine_save(sender, instance, created, **kwargs):
+    sync_room(instance.room_id)
+
+
+@receiver(post_delete, sender=WashingMachine)
+def machine_del(sender, instance, **kwargs):
+    sync_room(instance.room_id)