mirror of
https://github.com/more-tech4-magnum-opus/backend.git
synced 2024-11-22 11:36:33 +03:00
77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
from rest_framework import serializers
|
|
from rest_framework.generics import get_object_or_404
|
|
|
|
from events.models import Event, EventAttendance
|
|
from users.api.serializers import UserSerializer
|
|
from users.models import User
|
|
|
|
|
|
class EventSerializer(serializers.ModelSerializer):
|
|
creator = UserSerializer(many=False, read_only=True)
|
|
|
|
class Meta:
|
|
model = Event
|
|
fields = [
|
|
"name",
|
|
"about",
|
|
"slug",
|
|
"creator",
|
|
"starts",
|
|
"image",
|
|
"planning",
|
|
"attended",
|
|
]
|
|
extra_kwargs = {
|
|
"slug": {"read_only": True},
|
|
"creator": {"read_only": True},
|
|
"planning": {"read_only": True},
|
|
"attended": {"read_only": True},
|
|
}
|
|
|
|
def create(self, validated_data):
|
|
return Event.objects.create(
|
|
**validated_data, creator=self.context["request"].user
|
|
)
|
|
|
|
|
|
class EventAttendanceSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = EventAttendance
|
|
fields = ["id", "event_slug", "worker_username", "attended"]
|
|
extra_kwargs = {
|
|
"id": {"read_only": True},
|
|
"event_slug": {"read_only": True},
|
|
"worker_username": {"read_only": True},
|
|
"attended": {"read_only": True},
|
|
}
|
|
|
|
def create(self, validated_data):
|
|
return EventAttendance.objects.get_or_create(
|
|
worker=self.context["request"].user,
|
|
event=get_object_or_404(
|
|
Event, slug=self.context["request"].parser_context["kwargs"]["slug"]
|
|
),
|
|
)[0]
|
|
|
|
|
|
class SubmitUserAttendedSerializer(serializers.Serializer):
|
|
username = serializers.CharField(max_length=200)
|
|
|
|
def create(self, validated_data):
|
|
event = get_object_or_404(
|
|
Event, slug=self.context["request"].parser_context["kwargs"]["slug"]
|
|
)
|
|
ea = EventAttendance.objects.get_or_create(
|
|
event=event,
|
|
worker=get_object_or_404(User, username=validated_data["username"]),
|
|
)[0]
|
|
if not ea.attended:
|
|
ea.attended = True
|
|
ea.save()
|
|
ea.event.attended += 1
|
|
ea.event.save()
|
|
return ea
|
|
|
|
def update(self, instance, validated_data):
|
|
pass
|