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

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

payment/migrations/0003_paymenthistory_is_webhook.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.1.5 on 2023-05-25 13:47
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+ dependencies = [
+ ("payment", "0002_rename_user_id_payment_user_payment_status_and_more"),
+ ]
+
+ operations = [
+ migrations.AddField(
+ model_name="paymenthistory",
+ name="is_webhook",
+ field=models.BooleanField(default=False),
+ ),
+ ]

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,8 +47,30 @@ def post(self, request):
  return Response(dto)
 
 
-@api_view(["GET"])
-def get__generate_payment_key(request):
+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(["POST"])
+def post__generate_payment_key(request):
 
  request_ticket_type = TicketType.objects.get(id=request.data["ticket_type"])