FE에서 API로 전송할 결제 성공 API 작성

Author: golony6449

payment/models.py

@@ -33,5 +33,6 @@ class PaymentHistory(models.Model):
             (5, "환불 완료"),
         )
     )
+    is_webhook = models.BooleanField(default=False)
     create_at = models.DateTimeField(auto_now_add=True)
     update_at = models.DateTimeField(auto_now=True)

payment/views.py

@@ -1,15 +1,18 @@
+from django.db import transaction
 from rest_framework.decorators import api_view
 from rest_framework.response import Response
 from rest_framework.views import APIView
 
+from payment import enum
 from ticket.models import TicketType
 from payment.logic import generate_payment_key
-from payment.models import Payment
+from payment.models import Payment, PaymentHistory
 
 from django.conf import settings
 
 
 class PortoneWebhookApi(APIView):
+    @transaction.atomic
     def post(self, request):
         portone_ips = [
             "52.78.100.19",
@@ -20,9 +23,20 @@ def post(self, request):
         if settings.DEBUG is False and request.META.get("REMOTE_ADDR") not in portone_ips:
             raise ValueError("Not Allowed IP")
 
-        target_payment = Payment.objects.get(payment_key=request.data["merchant_uid"])
+        payment_key = request.data["merchant_uid"]
 
-        if request.data["status "] != "paid":
+        target_payment = Payment.objects.get(payment_key=payment_key)
+        target_payment.status = enum.PaymentStatus.PAYMENT_SUCCESS.value
+        target_payment.save()
+
+        payment_history = PaymentHistory(
+            payment_key=payment_key,
+            status=enum.PaymentStatus.PAYMENT_SUCCESS.value,
+            is_webhook=True
+        )
+        payment_history.save()
+
+        if request.data["status"] != "paid":
             raise ValueError("결제 승인건 이외의 요청")
 
         dto = {
@@ -33,6 +47,28 @@ def post(self, request):
         return Response(dto)
 
 
+class PaymentSuccessApi(APIView):
+    def post(self, request):
+        if not request.is_authenticated:
+            return Response({"msg": "not logged in user"}, status=400)
+
+        payment_key = request.data["merchant_uid"]
+
+        payment_history = PaymentHistory(
+            payment_key=payment_key,
+            status=enum.PaymentStatus.PAYMENT_SUCCESS.value,
+            is_webhook=False
+        )
+        payment_history.save()
+
+        dto = {
+            "msg": "ok",
+            "merchant_uid": request.data["merchant_uid"]
+        }
+
+        return Response(dto)
+
+
 @api_view(["GET"])
 def get__generate_payment_key(request):