Edit ticket models

Author: emscb

ticket/migrations/0004_alter_tickettype_id.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.1.5 on 2023-05-24 13:06
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ticket', '0003_ticket_remove_tickettype_code_and_more'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='tickettype',
+            name='id',
+            field=models.UUIDField(primary_key=True, serialize=False),
+        ),
+    ]

ticket/migrations/0005_alter_tickettype_day.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.1.5 on 2023-05-24 13:07
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ticket', '0004_alter_tickettype_id'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='tickettype',
+            name='day',
+            field=models.CharField(choices=[('FRI', '금요일'), ('SAT', '토요일'), ('SUN', '일요일'), ('WEEKEND', '토/일요일')], max_length=10),
+        ),
+    ]

ticket/migrations/0006_tickettype_program.py

@@ -0,0 +1,20 @@
+# Generated by Django 4.1.5 on 2023-05-24 13:39
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('program', '0003_program'),
+        ('ticket', '0005_alter_tickettype_day'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='tickettype',
+            name='program',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.PROTECT, to='program.program'),
+        ),
+    ]

ticket/models.py

@@ -9,19 +9,21 @@
 
 
 class TicketType(models.Model):
+    id = models.UUIDField(primary_key=True)
     name = models.CharField(max_length=100)
     price = models.IntegerField()
     min_price = models.IntegerField(null=True, blank=True)
     desc = models.TextField(max_length=1000)
     day = models.CharField(
         max_length=10,
         choices=(
+            ("FRI", "금요일"),
             ("SAT", "토요일"),
             ("SUN", "일요일"),
             ("WEEKEND", "토/일요일"),
         ),
     )
-    # program = models.ForeignKey()     # TODO
+    program = models.ForeignKey("program.Program", on_delete=models.PROTECT, null=True)
     is_refundable = models.BooleanField(default=True)
 
     def __str__(self):
@@ -30,6 +32,12 @@ def __str__(self):
     @property
     def buyable(self) -> bool:
         """잔여 수량이 있는지"""
+        if self.day == "FRI":
+            ticket_count = Ticket.objects.filter(
+                models.Q(ticket_type=self) & models.Q(is_refunded=False)
+            ).count()
+            return ticket_count < self.program.capacity
+
         sat_ticket_count = Ticket.objects.filter(
             models.Q(ticket_type__day="SAT") | models.Q(ticket_type__day="WEEKEND")
         ).filter(is_refunded=False).count()

ticket/view_models.py

@@ -5,21 +5,24 @@
 
 
 @dataclass(init=False)
-class ConferenceTicketTypeViewModel:
-    code: str
+class TicketTypeViewModel:
+    id: str
     name: str
     price: int
     min_price: Optional[int]
     desc: str
-    day: str
+    day: str  # choice
+    # program: ...
+    is_refundable: bool
 
     def __init__(self, model: TicketType):
-        self.code = model.code
+        self.id = model.id
         self.name = model.name
         self.price = model.price
         self.min_price = model.min_price
         self.desc = model.desc
         self.day = model.day
+        self.is_refundable = model.is_refundable
 
     def to_dict(self):
         return asdict(self)

ticket/views.py

@@ -1,24 +1,24 @@
 import json
 import traceback
 from datetime import datetime
-from typing import Callable, Literal
+from typing import Callable, Literal, Dict, List
 
 from django.contrib.auth import get_user_model
 from django.http import HttpRequest, HttpResponse
 from django.shortcuts import get_object_or_404, render
-from django.views.decorators.csrf import csrf_exempt
 from django.views import View
+from django.views.decorators.csrf import csrf_exempt
 
+import payment.utils
+from program.models import CONFERENCE, TUTORIAL, SPRINT
 from .models import Ticket, TicketType
 from .requests import (
     AddConferenceTicketRequest,
     CheckConferenceTicketTypeBuyableRequest,
     GetConferenceTicketTypesRequest,
     RequestParsingException,
 )
-from .view_models import ConferenceTicketTypeViewModel
-
-import payment.utils
+from .view_models import TicketTypeViewModel
 
 User = get_user_model()
 
@@ -64,20 +64,27 @@ def get__get_ticket_types(request: HttpRequest, **kwargs) -> HttpResponse:
 
     ticket_types = TicketType.objects.all()
 
-    return HttpResponse(
-        json.dumps(
-            [
-                ConferenceTicketTypeViewModel(ticket_type).to_dict()
-                for ticket_type in ticket_types
-            ]
-        )
-    )
+    response: Dict[str, List[dict]] = {
+        "conference": [],
+        "tutorial": [],
+        "sprint": [],
+    }
+
+    for ticket_type in ticket_types:
+        if ticket_type.program.program_type == CONFERENCE:
+            response["conference"].append(TicketTypeViewModel(ticket_type).to_dict())
+        elif ticket_type.program.program_type == TUTORIAL:
+            response["tutorial"].append(TicketTypeViewModel(ticket_type).to_dict())
+        elif ticket_type.program.program_type == SPRINT:
+            response["sprint"].append(TicketTypeViewModel(ticket_type).to_dict())
+
+    return HttpResponse(json.dumps(response))
 
 
 @request_method("GET")
 @exception_wrapper
 def get__check_conference_ticket_type_buyable(
-    request: HttpRequest, **kwargs
+        request: HttpRequest, **kwargs
 ) -> HttpResponse:
     """특정 티켓 종류 구매 가능 여부 조회"""
     request = CheckConferenceTicketTypeBuyableRequest(request, **kwargs)
@@ -143,10 +150,10 @@ def post__add_conference_ticket(request: HttpRequest, **kwargs) -> HttpResponse:
 
     bought_tickets = Ticket.objects.filter(user=user)
     if any(
-        (
-            not bought_ticket.ticket_type.can_coexist(ticket_type)
-            for bought_ticket in bought_tickets
-        )
+            (
+                    not bought_ticket.ticket_type.can_coexist(ticket_type)
+                    for bought_ticket in bought_tickets
+            )
     ):
         return HttpResponse("Duplicate day", status=400)