Adding more shit
This commit is contained in:
@@ -7,7 +7,7 @@ from ...models.models import MembershipTier
|
||||
from ...schemas import (
|
||||
MembershipTierCreate, MembershipTierUpdate, MembershipTierResponse, MessageResponse
|
||||
)
|
||||
from ...api.dependencies import get_current_active_user, get_admin_user
|
||||
from ...api.dependencies import get_current_active_user, get_admin_user, get_super_admin_user
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
@@ -47,10 +47,10 @@ async def get_membership_tier(
|
||||
@router.post("/", response_model=MembershipTierResponse, status_code=status.HTTP_201_CREATED)
|
||||
async def create_membership_tier(
|
||||
tier_data: MembershipTierCreate,
|
||||
current_user = Depends(get_admin_user),
|
||||
current_user = Depends(get_super_admin_user),
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
"""Create a new membership tier (admin only)"""
|
||||
"""Create a new membership tier (super admin only)"""
|
||||
# Check if tier with same name exists
|
||||
existing_tier = db.query(MembershipTier).filter(
|
||||
MembershipTier.name == tier_data.name
|
||||
@@ -74,10 +74,10 @@ async def create_membership_tier(
|
||||
async def update_membership_tier(
|
||||
tier_id: int,
|
||||
tier_update: MembershipTierUpdate,
|
||||
current_user = Depends(get_admin_user),
|
||||
current_user = Depends(get_super_admin_user),
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
"""Update membership tier (admin only)"""
|
||||
"""Update membership tier (super admin only)"""
|
||||
tier = db.query(MembershipTier).filter(MembershipTier.id == tier_id).first()
|
||||
if not tier:
|
||||
raise HTTPException(
|
||||
@@ -99,10 +99,10 @@ async def update_membership_tier(
|
||||
@router.delete("/{tier_id}", response_model=MessageResponse)
|
||||
async def delete_membership_tier(
|
||||
tier_id: int,
|
||||
current_user = Depends(get_admin_user),
|
||||
current_user = Depends(get_super_admin_user),
|
||||
db: Session = Depends(get_db)
|
||||
):
|
||||
"""Delete membership tier (admin only)"""
|
||||
"""Delete membership tier (super admin only)"""
|
||||
tier = db.query(MembershipTier).filter(MembershipTier.id == tier_id).first()
|
||||
if not tier:
|
||||
raise HTTPException(
|
||||
|
||||
Reference in New Issue
Block a user