Vladless
83 lines
3.1 KiB
Python
83 lines
3.1 KiB
Python
from fastapi import APIRouter, Depends, Query
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from api.depends import get_session
|
|
from api.v2.schemas import TariffBase, TariffResponse, TariffUpdate
|
|
from api.v2.schemas.tariffs import TariffGroup, TariffPublic
|
|
from api.v2.base_crud import generate_crud_router
|
|
from database.models import Tariff
|
|
|
|
|
|
def _tariff_to_public(t: Tariff) -> TariffPublic:
|
|
return TariffPublic(
|
|
id=t.id,
|
|
name=t.name or "",
|
|
group_code=t.group_code or "",
|
|
duration_days=t.duration_days or 0,
|
|
price_rub=t.price_rub or 0,
|
|
traffic_limit=t.traffic_limit,
|
|
device_limit=t.device_limit,
|
|
subgroup_title=t.subgroup_title,
|
|
sort_order=t.sort_order,
|
|
vless=bool(getattr(t, "vless", False)),
|
|
)
|
|
|
|
|
|
public_router = APIRouter()
|
|
|
|
|
|
@public_router.get("/groups", response_model=list[TariffGroup])
|
|
async def get_tariff_groups(session: AsyncSession = Depends(get_session)):
|
|
"""Публичный список групп тарифов — уникальные значения колонки group_code."""
|
|
q = (
|
|
select(Tariff.group_code)
|
|
.where(Tariff.is_active == True, Tariff.group_code.isnot(None), Tariff.group_code != "")
|
|
.distinct()
|
|
.order_by(Tariff.group_code)
|
|
)
|
|
result = await session.execute(q)
|
|
values = result.scalars().all()
|
|
return [TariffGroup(group_code=v or "") for v in values]
|
|
|
|
|
|
@public_router.get("/public", response_model=list[TariffPublic])
|
|
async def get_tariffs_public(
|
|
group_code: str | None = Query(None, description="Фильтр по группе тарифов"),
|
|
tariff_ids: str | None = Query(None, description="ID тарифов через запятую (приоритет над группой)"),
|
|
filter_vless: str | None = Query(
|
|
None,
|
|
description="vless: только для роутера (vless=True), app: только для приложения (vless=False), иначе все",
|
|
),
|
|
session: AsyncSession = Depends(get_session),
|
|
):
|
|
"""Публичный список активных тарифов (без авторизации)."""
|
|
q = select(Tariff).where(Tariff.is_active == True).order_by(Tariff.sort_order.asc().nulls_last(), Tariff.price_rub.asc())
|
|
if tariff_ids:
|
|
try:
|
|
ids = [int(x.strip()) for x in tariff_ids.split(",") if x.strip()]
|
|
if ids:
|
|
q = q.where(Tariff.id.in_(ids))
|
|
except ValueError:
|
|
pass
|
|
elif group_code:
|
|
q = q.where(Tariff.group_code == group_code)
|
|
if filter_vless == "router":
|
|
q = q.where(Tariff.vless == True)
|
|
elif filter_vless == "app":
|
|
q = q.where(Tariff.vless == False)
|
|
result = await session.execute(q)
|
|
rows = result.scalars().all()
|
|
return [_tariff_to_public(t) for t in rows]
|
|
|
|
|
|
router = generate_crud_router(
|
|
model=Tariff,
|
|
schema_response=TariffResponse,
|
|
schema_create=TariffBase,
|
|
schema_update=TariffUpdate,
|
|
identifier_field="name",
|
|
parameter_name="name",
|
|
enabled_methods=["get_all", "get_one", "create", "update", "delete"],
|
|
)
|