Merge branch 'master' of gitlab.prodcontest.ru:team-15/project

services/backend/api/v1/competition/schemas.py

@@ -1,8 +1,9 @@
+from typing import Literal
 from uuid import UUID
 
-from ninja import ModelSchema
+from ninja import ModelSchema, Schema
 
-from apps.competition.models import Competition
+from apps.competition.models import Competition, State
 
 
 class CompetitionOut(ModelSchema):
@@ -13,11 +14,33 @@ class CompetitionOut(ModelSchema):
         fields = "__all__"
 
 
+class StateOut(ModelSchema):
+    class Meta:
+        model = State
+        fields = ("state",)
+
+
+class StateIn(Schema):
+    state: Literal["started", "not_started", "finished"]
+
+
 class CompetitionListInstanceOut(ModelSchema):
     id: UUID
     is_participating: bool
     completed: bool
 
+    @staticmethod
+    def resolve_is_participating(self, context):
+        user = context["request"].auth
+        return self.participants.filter(id=user.id).exists()
+
+    @staticmethod
+    def resolve_completed(self, context):
+        user = context["request"].auth
+        return State.objects.filter(
+            competition=self, user=user, state="finished"
+        ).exists()
+
     class Meta:
         model = Competition
         fields = (

services/backend/api/v1/competition/views.py

@@ -2,11 +2,12 @@ from http import HTTPStatus as status
 from uuid import UUID
 
 from django.http import HttpRequest
+from django.shortcuts import get_object_or_404
 from ninja import Router
 
 import api.v1.schemas as global_schemas
-from api.v1.auth import BearerAuth
 from api.v1.competition import schemas
+from apps.competition.models import Competition, State
 
 router = Router(tags=["competition"])
 
@@ -18,11 +19,12 @@ router = Router(tags=["competition"])
         status.BAD_REQUEST: global_schemas.BadRequestError,
         status.UNAUTHORIZED: global_schemas.UnauthorizedError,
     },
-    auth=BearerAuth(),
 )
 def get_competition(
     request: HttpRequest, competition_id: UUID
-) -> tuple[status, schemas.CompetitionOut]: ...
+) -> tuple[status, schemas.CompetitionOut]:
+    competition = get_object_or_404(Competition, id=competition_id)
+    return status.OK, competition
 
 
 @router.get(
@@ -32,8 +34,35 @@ def get_competition(
         status.BAD_REQUEST: global_schemas.BadRequestError,
         status.UNAUTHORIZED: global_schemas.UnauthorizedError,
     },
-    auth=BearerAuth(),
 )
 def list_competitions(
     request: HttpRequest, is_participating: bool
-) -> tuple[status, list[schemas.CompetitionListInstanceOut]]: ...
+) -> tuple[status, list[schemas.CompetitionListInstanceOut]]:
+    user = request.auth
+    if is_participating:
+        competitions = Competition.objects.filter(participants=user)
+    else:
+        competitions = Competition.objects.exclude(participants=user)
+    return status.OK, competitions
+
+
+@router.post(
+    "competitions/{competition_id}/state",
+    response={
+        status.OK: schemas.StateOut,
+        status.BAD_REQUEST: global_schemas.BadRequestError,
+        status.UNAUTHORIZED: global_schemas.UnauthorizedError,
+    },
+)
+def change_competition_state(
+    request: HttpRequest,
+    competition_id: UUID,
+    state: schemas.StateIn,
+) -> tuple[status, schemas.StateOut]:
+    user = request.auth
+    competition = get_object_or_404(Competition, id=competition_id)
+
+    state_obj, _ = State.objects.update_or_create(
+        user=user, competition=competition, state=state.state
+    )
+    return status.OK, schemas.StateOut.from_orm(state_obj)

services/backend/api/v1/router.py

@@ -3,6 +3,7 @@ from functools import partial
 from ninja import NinjaAPI
 
 from api.v1 import handlers
+from api.v1.auth import BearerAuth
 from api.v1.competition.views import router as competition_router
 from api.v1.ping.views import router as ping_router
 from api.v1.user.views import router as user_router
@@ -12,6 +13,7 @@ router = NinjaAPI(
     version="1",
     description="API docs for DataRush",
     openapi_url="/docs/openapi.json",
+    auth=BearerAuth(),
 )
 
 

services/backend/apps/competition/migrations/0002_competition_participants.py

@@ -0,0 +1,19 @@
+# Generated by Django 5.1.6 on 2025-02-28 22:40
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('competition', '0001_initial'),
+        ('user', '0002_user_status'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='competition',
+            name='participants',
+            field=models.ManyToManyField(related_name='participants', to='user.user'),
+        ),
+    ]

services/backend/apps/competition/migrations/0003_state.py

@@ -0,0 +1,28 @@
+# Generated by Django 5.1.6 on 2025-02-28 23:26
+
+import django.db.models.deletion
+import uuid
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('competition', '0002_competition_participants'),
+        ('user', '0003_alter_user_status'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='State',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('state', models.CharField(choices=[('not_started', 'Not Started'), ('started', 'Started'), ('finished', 'Finished')], max_length=11)),
+                ('competition', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='competition.competition')),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='user.user')),
+            ],
+            options={
+                'abstract': False,
+            },
+        ),
+    ]

services/backend/apps/competition/models.py

@@ -1,18 +1,17 @@
 from django.db import models
 
 from apps.core.models import BaseModel
-
-
-class CompetitionType(models.TextChoices):
-    SOLO = "solo"
-
-
-class CompetitionParticipationType(models.TextChoices):
-    EDU = "edu"
-    COMPETITIVE = "competitive"
+from apps.user.models import User
 
 
 class Competition(BaseModel):
+    class CompetitionType(models.TextChoices):
+        SOLO = "solo"
+
+    class CompetitionParticipationType(models.TextChoices):
+        EDU = "edu"
+        COMPETITIVE = "competitive"
+
     title = models.CharField(max_length=100, verbose_name="Название")
     description = models.TextField(verbose_name="Описание")
     image_url = models.FileField(
@@ -34,7 +33,19 @@ class Competition(BaseModel):
         choices=CompetitionParticipationType.choices,
         verbose_name="Тип соревнования",
     )
+    participants = models.ManyToManyField(User, related_name="participants")
 
     class Meta:
         verbose_name = "соревнование"
         verbose_name_plural = "соревнования"
+
+
+class State(BaseModel):
+    class StateChoices(models.TextChoices):
+        NOT_STARTED = "not_started"
+        STARTED = "started"
+        FINISHED = "finished"
+
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+    competition = models.ForeignKey(Competition, on_delete=models.CASCADE)
+    state = models.CharField(choices=StateChoices.choices, max_length=11)

services/backend/apps/user/migrations/0002_user_status.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.6 on 2025-02-28 22:40
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('user', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='user',
+            name='status',
+            field=models.CharField(choices=[('student', 'Student'), ('metodist', 'Metodist')], default='student', max_length=10),
+        ),
+    ]

services/backend/apps/user/migrations/0003_alter_user_status.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.6 on 2025-02-28 22:45
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('user', '0002_user_status'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='user',
+            name='status',
+            field=models.CharField(choices=[('student', 'Student'), ('metodist', 'Metodist')], default='student', max_length=10),
+        ),
+    ]

services/backend/apps/user/models.py

@@ -14,7 +14,7 @@ class User(BaseModel):
     password = models.TextField(verbose_name="Пароль")
 
     status = models.CharField(
-        max_length=10, choices=UserRole.choices, default=UserRole.STUDENT
+        max_length=10, choices=UserRole, default="student"
     )
 
     def __str__(self) -> str:

services/backend/apps/user/test.py

@@ -0,0 +1,32 @@
+from django.core.exceptions import ValidationError
+from django.test import TestCase
+
+from apps.user.models import User
+
+
+class TestSignUp(TestCase):
+    def test_correct_signup(self):
+        user = User(
+            email="123123@timka.su",
+            password="1321312",
+            username="123123",
+        )
+        user.full_clean()
+        user.save()
+
+    def test_incorrect_mail(self):
+        user = User(
+            email="123123",
+            password="1321312",
+            username="123123123",
+        )
+        with self.assertRaises(ValidationError):
+            user.full_clean()
+
+    def test_missing_params(self):
+        user = User(
+            password="123123",
+            username="132131232131"
+        )
+        with self.assertRaises(ValidationError):
+            user.full_clean()