34 lines
984 B
Python
34 lines
984 B
Python
|
from fastapi import APIRouter, Depends
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from app.core.database import get_db
|
||
|
|
from app.core.dependencies import get_current_business
|
||
|
|
from app.modules.billing import schemas, service
|
||
|
|
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/plan", response_model=schemas.PlanRead)
|
||
|
|
async def get_plan(
|
||
|
|
business_id: int = Depends(get_current_business),
|
||
|
|
db: AsyncSession = Depends(get_db),
|
||
|
|
):
|
||
|
|
return await service.get_plan(db, business_id)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/upgrade", response_model=schemas.PlanRead)
|
||
|
|
async def upgrade_plan(
|
||
|
|
body: schemas.UpgradeRequest,
|
||
|
|
business_id: int = Depends(get_current_business),
|
||
|
|
db: AsyncSession = Depends(get_db),
|
||
|
|
):
|
||
|
|
return await service.upgrade_plan(db, business_id, body.plan)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/usage", response_model=schemas.UsageRead)
|
||
|
|
async def get_usage(
|
||
|
|
business_id: int = Depends(get_current_business),
|
||
|
|
db: AsyncSession = Depends(get_db),
|
||
|
|
):
|
||
|
|
return await service.get_usage(db, business_id)
|