callback, payment_key 생성 추가

Author: golony6449

payment/logic.py

@@ -0,0 +1,46 @@
+from django.contrib.auth import get_user_model
+from django.db import transaction
+
+from payment.enum import PaymentStatus
+from payment.models import Payment, PaymentHistory
+from ticket.models import TicketType, Ticket
+
+import shortuuid
+
+User = get_user_model()
+
+
+@transaction.atomic
+def generate_payment_key(user: User, ticket_type: TicketType):
+    new_payment = Payment(
+        payment_key=shortuuid.uuid(),
+        user_id=user,
+        # ticket_type=ticket_type,      # TODO
+        money=ticket_type.price,
+        status=PaymentStatus.BEFORE_PAYMENT.value
+    )
+
+    new_payment.save()
+
+    _save_history(new_payment.payment_key, PaymentStatus.BEFORE_PAYMENT.value)
+    return new_payment.payment_key
+
+
+@transaction.atomic
+def proceed_payment(payment_key: str, is_succeed: bool):
+    status_value = PaymentStatus.PAYMENT_SUCCESS.value if is_succeed else PaymentStatus.PAYMENT_FAILED.value
+
+    target_payment = Payment.objects.get(payment_key=payment_key)
+    target_payment.status = status_value
+    target_payment.save()
+
+    _save_history(payment_key, status_value)
+
+
+def _save_history(payment_key: str, status: int):
+    new_payment_history = PaymentHistory(
+        payment_key=payment_key,
+        status=status
+    )
+
+    new_payment_history.save()

payment/urls.py

@@ -1,4 +1,9 @@
 from django.contrib import admin
 from django.urls import include, path
 
-urlpatterns = []
+from payment.views import PortoneWebhookApi
+
+
+urlpatterns = [
+    path("portone/webhook", PortoneWebhookApi.as_view(), name="portone-webhook"),
+]

payment/views.py

@@ -1,12 +1,45 @@
+from rest_framework.decorators import api_view
+from rest_framework.response import Response
 from rest_framework.views import APIView
 
+from ticket.models import TicketType
+from payment.logic import generate_payment_key
+from payment.models import Payment
 
-class PortoneWebhookAPI(APIView):
+
+class PortoneWebhookApi(APIView):
     def post(self, request):
         # TODO: IP Filtering
         # 52.78.100.19
         # 52.78.48.223
         # 52.78.5.241 (Webhook Test Only)
 
-        pass
+        target_payment = Payment.objects.get(payment_key=request.data["merchant_uid"])
+
+        if request.data["status "] != "paid":
+            raise ValueError("결제 승인건 이외의 요청")
+
+        dto = {
+            "msg": "ok",
+            "merchant_uid": request.data["merchant_uid"]
+        }
+
+        return Response(dto)
+
+
+@api_view(["GET"])
+def get__generate_payment_key(request):
+
+    request_ticket_type = TicketType.objects.get(id=request.data["ticket_type"])
+
+    payment_key = generate_payment_key(
+        user=request.user,
+        ticket_type=request_ticket_type
+    )
+
+    response_data = {
+        "msg": "ok",
+        "payment_key": payment_key
+    }
 
+    return Response()