Merge pull request #64 from RestTest-App/develop

멤버십 추가

Author: qor0530

alembic/versions/a1b2c3d4e5f6_add_membership_and_payment_history.py

@@ -0,0 +1,73 @@
+"""add membership and payment_history
+
+Revision ID: a1b2c3d4e5f6
+Revises: f17e2374299a
+Create Date: 2025-11-30 00:00:00.000000
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'a1b2c3d4e5f6'
+down_revision: Union[str, None] = 'f17e2374299a'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # Add membership_tier to user table
+    op.add_column(
+        'user',
+        sa.Column(
+            'membership_tier',
+            sa.String(20),
+            nullable=False,
+            server_default='FREE',
+            comment="멤버십 등급 (FREE, PREMIUM)"
+        )
+    )
+
+    # Add subscription_expire_date to user table
+    op.add_column(
+        'user',
+        sa.Column(
+            'subscription_expire_date',
+            sa.DateTime(),
+            nullable=True,
+            comment="구독 만료일"
+        )
+    )
+
+    # Create payment_history table
+    op.create_table(
+        'payment_history',
+        sa.Column('id', sa.BigInteger(), nullable=False, comment="결제 이력 고유 식별값"),
+        sa.Column('user_id', sa.BigInteger(), nullable=False, comment="사용자 ID"),
+        sa.Column('membership_tier', sa.String(20), nullable=False, comment="구매한 멤버십 등급"),
+        sa.Column('payment_amount', sa.Integer(), nullable=False, comment="결제 금액 (원)"),
+        sa.Column('payment_method', sa.String(50), nullable=True, comment="결제 수단 (카드, 계좌이체 등)"),
+        sa.Column('payment_status', sa.String(20), nullable=False, comment="결제 상태 (SUCCESS, FAILED, REFUND)"),
+        sa.Column('subscription_start_date', sa.DateTime(), nullable=False, comment="구독 시작일"),
+        sa.Column('subscription_end_date', sa.DateTime(), nullable=False, comment="구독 종료일"),
+        sa.Column('created_at', sa.DateTime(), nullable=False, comment="결제 시간"),
+        sa.Column('transaction_id', sa.String(255), nullable=True, comment="결제 고유 ID (PG사 제공)"),
+        sa.Column('pg_provider', sa.String(50), nullable=True, comment="PG사 (토스페이, 네이버페이 등)"),
+        sa.PrimaryKeyConstraint('id'),
+        sa.ForeignKeyConstraint(['user_id'], ['user.id'], ),
+        sa.UniqueConstraint('transaction_id')
+    )
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # Drop payment_history table
+    op.drop_table('payment_history')
+
+    # Remove membership columns from user table
+    op.drop_column('user', 'subscription_expire_date')
+    op.drop_column('user', 'membership_tier')

api/v1/membership/__init__.py

@@ -0,0 +1,204 @@
+from fastapi import APIRouter, Depends
+from typing import Annotated
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.auth.dependency import get_current_user
+from database.dependency import get_db
+from domain.user.entity.user import User
+from domain.membership.service.membership_service import MembershipService
+from app.utils.dto.success import ok
+from pydantic import BaseModel, Field
+
+
+router = APIRouter()
+
+
+class UpgradeMembershipRequest(BaseModel):
+    """멤버십 업그레이드 요청 DTO"""
+    membership_tier: str = Field(..., description="멤버십 등급 (FREE, PREMIUM)")
+    payment_amount: int = Field(default=0, description="결제 금액 (원)")
+    duration_days: int = Field(default=30, description="구독 기간 (일)")
+
+
+class MembershipInfoResponse(BaseModel):
+    """멤버십 정보 응답 DTO"""
+    user_id: int
+    membership_tier: str
+    is_active: bool
+    is_expired: bool
+    subscription_expire_date: str | None
+
+
+@router.get("/info")
+async def get_membership_info(
+    current_user: Annotated[User, Depends(get_current_user)]
+):
+    """
+    멤버십 정보 조회
+
+    Returns:
+        {
+            "code": 200,
+            "message": "멤버십 정보 조회 성공",
+            "data": {
+                "user_id": 1,
+                "membership_tier": "FREE",
+                "is_active": false,
+                "is_expired": false,
+                "subscription_expire_date": null
+            }
+        }
+    """
+    subscription_status = await MembershipService.check_subscription_status(current_user)
+
+    membership_tier = getattr(current_user, 'membership_tier', 'FREE')
+
+    response = MembershipInfoResponse(
+        user_id=current_user.id,
+        membership_tier=membership_tier,
+        is_active=subscription_status["is_active"],
+        is_expired=subscription_status["is_expired"],
+        subscription_expire_date=subscription_status["expire_date"].isoformat() if subscription_status["expire_date"] else None
+    )
+
+    return ok(data=response.model_dump(), message="멤버십 정보 조회 성공")
+
+
+@router.patch("/upgrade")
+async def upgrade_membership(
+    request: UpgradeMembershipRequest,
+    current_user: Annotated[User, Depends(get_current_user)],
+    db: Annotated[AsyncSession, Depends(get_db)]
+):
+    """
+    멤버십 업그레이드 (테스트용)
+
+    Request Body:
+        {
+            "membership_tier": "PREMIUM",
+            "payment_amount": 9900,
+            "duration_days": 30
+        }
+
+    Returns:
+        {
+            "code": 200,
+            "message": "멤버십 업그레이드 성공",
+            "data": {
+                "user_id": 1,
+                "previous_tier": "FREE",
+                "new_tier": "PREMIUM",
+                "subscription_expire_date": "2026-01-11T12:00:00"
+            }
+        }
+    """
+    result = await MembershipService.upgrade_membership(
+        db=db,
+        user=current_user,
+        membership_tier=request.membership_tier,
+        payment_amount=request.payment_amount,
+        duration_days=request.duration_days
+    )
+
+    # subscription_expire_date가 datetime이면 문자열로 변환
+    if result["subscription_expire_date"]:
+        result["subscription_expire_date"] = result["subscription_expire_date"].isoformat()
+
+    return ok(data=result, message="멤버십 업그레이드 성공")
+
+
+@router.get("/payment-history")
+async def get_payment_history(
+    current_user: Annotated[User, Depends(get_current_user)],
+    db: Annotated[AsyncSession, Depends(get_db)]
+):
+    """
+    결제 이력 조회
+
+    Returns:
+        {
+            "code": 200,
+            "message": "결제 이력 조회 성공",
+            "data": [
+                {
+                    "id": 1,
+                    "membership_tier": "PREMIUM",
+                    "payment_amount": 9900,
+                    "payment_method": "TEST",
+                    "payment_status": "SUCCESS",
+                    "subscription_start_date": "2025-11-30T00:00:00",
+                    "subscription_end_date": "2025-12-30T00:00:00",
+                    "created_at": "2025-11-30T00:00:00",
+                    "transaction_id": "TEST_1_1234567890",
+                    "pg_provider": "TEST"
+                }
+            ]
+        }
+    """
+    payment_history = await MembershipService.get_payment_history(
+        db=db,
+        user_id=current_user.id
+    )
+
+    return ok(data=payment_history, message="결제 이력 조회 성공")
+
+
+@router.patch("/cancel")
+async def cancel_membership(
+    current_user: Annotated[User, Depends(get_current_user)],
+    db: Annotated[AsyncSession, Depends(get_db)]
+):
+    """
+    멤버십 구독 해지
+
+    구독을 해지합니다. 만료일까지는 프리미엄 기능을 계속 사용할 수 있습니다.
+
+    Returns:
+        {
+            "code": 200,
+            "message": "구독 해지가 완료되었습니다",
+            "data": {
+                "user_id": 1,
+                "membership_tier": "PREMIUM",
+                "subscription_expire_date": "2026-01-11T12:00:00",
+                "days_remaining": 29,
+                "message": "만료일까지 프리미엄 기능을 계속 사용할 수 있습니다."
+            }
+        }
+    """
+    from datetime import datetime
+    from exception.client_exception import BadRequestException
+
+    membership_tier = getattr(current_user, 'membership_tier', 'FREE')
+    subscription_expire_date = getattr(current_user, 'subscription_expire_date', None)
+
+    # 이미 FREE 등급이면 해지할 구독이 없음
+    if membership_tier == "FREE":
+        raise BadRequestException(message="해지할 구독이 없습니다.")
+
+    # 구독 상태 확인
+    subscription_status = await MembershipService.check_subscription_status(current_user)
+
+    if subscription_status["is_expired"]:
+        raise BadRequestException(message="이미 만료된 구독입니다.")
+
+    # 남은 일수 계산
+    if subscription_expire_date:
+        now = datetime.now()
+        days_remaining = (subscription_expire_date - now).days
+    else:
+        days_remaining = 0
+
+    # 실제 구독 해지 로직은 여기서 구현
+    # (예: auto_renewal 플래그 설정, 또는 즉시 FREE로 변경 등)
+    # 현재는 정보만 반환
+
+    response = {
+        "user_id": current_user.id,
+        "membership_tier": membership_tier,
+        "subscription_expire_date": subscription_expire_date.isoformat() if subscription_expire_date else None,
+        "days_remaining": days_remaining,
+        "message": f"만료일({subscription_expire_date.date()})까지 프리미엄 기능을 계속 사용할 수 있습니다." if subscription_expire_date else "구독 해지가 완료되었습니다."
+    }
+
+    return ok(data=response, message="구독 해지가 완료되었습니다")

api/v1/membership/membership_router.py

@@ -6,6 +6,7 @@
 from api.v1.studybook import studybook_router
 from api.v1.studybook import studybook_question_router
 from api.v1.test import test_router
+from api.v1.membership import membership_router
 router = APIRouter(prefix="/api/v1")
 
 router.include_router(review_router.router, tags=["review"], prefix="/review")
@@ -15,4 +16,4 @@
 router.include_router(studybook_question_router.router, tags=["studybook_question"], prefix="/studybook-question")
 router.include_router(test_router.router, tags=["test"], prefix="/test")
 router.include_router(certificate_router.router, tags=["certificate"], prefix="/user")
-router.include_router(review_router.router, tags=["review"], prefix="/review")
+router.include_router(membership_router.router, tags=["membership"], prefix="/membership")

api/v1/routers.py

@@ -1,6 +1,6 @@
 # api/v1/studybook/studybook_router.py
 
-from fastapi import APIRouter, UploadFile, File, Path, Depends
+from fastapi import APIRouter, UploadFile, File, Form, Path, Depends
 from typing import Annotated
 from datetime import datetime
 
@@ -27,7 +27,7 @@
 from app.utils.dummy_questions import dummy_questions
 from app.studybook.usecase.studybook_usecase import upload_my_studybook_by_dummy_usecase
 
-from app.auth.dependency import get_current_user
+from app.auth.dependency import get_current_user, check_api_rate_limit
 router = APIRouter()
 #
 # def get_dummy_user() -> User:
@@ -63,23 +63,34 @@ async def upload_my_studybook_by_dummy(
 @router.post("/upload-my-studybook", response_model=UploadStudybookResponseDTO)
 async def upload_my_studybook(
     files: list[UploadFile] = File(...),
-    answers: str = None,
-    question_count: int = None,
+    copyright_agreed: bool = Form(...),  # 저작권 동의 필수
+    answers: str = Form(None),
+    question_count: int = Form(None),
     *,
     current_user: Annotated[User, Depends(get_current_user)],
-    db: Annotated[Session, Depends(get_db)]
+    db: Annotated[Session, Depends(get_db)],
+    _: Annotated[None, Depends(check_api_rate_limit)]  # Rate Limit 체크
 ):
     """
     통합 문제집 업로드 API (PDF 또는 이미지)
 
     - PDF: 1개 파일만 업로드 가능
     - 이미지: 여러 개 파일 업로드 가능
 
-    Query Parameters:
+    Form Data Parameters:
         files: 업로드할 파일 (PDF 또는 이미지)
+        copyright_agreed: 저작권 동의 여부 (필수, true 필요)
         answers: JSON 형식의 정답 리스트 (선택사항)
         question_count: 예상 문제 개수 (PDF 전용, 선택사항)
     """
+    from exception.client_exception import BadRequestException
+
+    # 저작권 동의 확인
+    if not copyright_agreed:
+        raise BadRequestException(
+            message="저작권 관련 법률을 준수하기 위해 동의가 필요합니다. 타인의 저작물을 무단으로 업로드하지 마세요."
+        )
+
     answer_list = None
     if answers:
         import json