Merge pull request #90 from golony6449/feature/golony/payment

portone callback 및 payment-key 생성 API 추가

Author: golony6449

payment/enum.py

@@ -0,0 +1,9 @@
+from enum import Enum
+
+
+class PaymentStatus(Enum):
+    BEFORE_PAYMENT = 1
+    PAYMENT_FAILED = 2
+    PAYMENT_SUCCESS = 3
+    REFUND_FAILED = 4
+    REFUND_SUCCESS = 5

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=user,
+        ticket_type=ticket_type,
+        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,10 @@
 from django.contrib import admin
 from django.urls import include, path
 
-urlpatterns = []
+from payment.views import PortoneWebhookApi, get__generate_payment_key
+
+
+urlpatterns = [
+    path("portone/webhook/", PortoneWebhookApi.as_view(), name="portone-webhook"),
+    path("key/", get__generate_payment_key, name="get-payment-key"),
+]

payment/views.py

@@ -1,12 +1,52 @@
+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):
+from django.conf import settings
+
+
+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)
+        portone_ips = [
+            "52.78.100.19",
+            "52.78.48.223",
+            "52.78.5.241"   # (Webhook Test Only)
+        ]
+
+        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"])
+
+        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
+    )
 
-        pass
+    response_data = {
+        "msg": "ok",
+        "payment_key": payment_key,
+        "price": request_ticket_type.price
+    }
 
+    return Response(response_data)