126 lines
3.5 KiB
Python
126 lines
3.5 KiB
Python
import sys
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Query, Depends
|
|
from sqlmodel import SQLModel, Field
|
|
from sqlmodel import select
|
|
|
|
import alchemy
|
|
import utils
|
|
from dependencies import get_session
|
|
from utils import update_item, create_item
|
|
from .status_model import Status
|
|
from .user import ACTIVE_USER
|
|
|
|
PRIMARY_ANNOTATION = utils.make_primary_annotation('Country')
|
|
|
|
|
|
# ----------------------------------------------------------------
|
|
# Models
|
|
|
|
class CountryBase(SQLModel):
|
|
code: str = Field(max_length=2, unique=True)
|
|
name: str = Field(max_length=80)
|
|
notes: str | None
|
|
|
|
|
|
class Country(CountryBase):
|
|
id: int | None = Field(default=None, primary_key=True)
|
|
|
|
|
|
class CountryCreate(CountryBase):
|
|
notes: str | None = None
|
|
|
|
|
|
class CountryPublic(CountryBase):
|
|
id: int
|
|
status: Status
|
|
|
|
|
|
class CountryUpdate(CountryBase):
|
|
code: str | None = None
|
|
name: str | None = None
|
|
notes: str | None = None
|
|
|
|
|
|
# ----------------------------------------------------------------
|
|
# Routes
|
|
|
|
router = APIRouter(prefix="/country", tags=["country"])
|
|
|
|
|
|
@router.get("/", response_model=list[CountryPublic])
|
|
async def get_countries(
|
|
offset: int = 0,
|
|
limit: Annotated[int, Query] = 100,
|
|
session=Depends(get_session)):
|
|
"""Get list of all countries"""
|
|
if limit < 1:
|
|
limit = sys.maxsize
|
|
return session.exec(select(alchemy.Country).offset(offset).limit(limit)).all()
|
|
|
|
|
|
@router.get("/{country_id}",
|
|
response_model=CountryPublic,
|
|
responses={404: {"description": "Not found"}})
|
|
async def get_country(
|
|
country_id: PRIMARY_ANNOTATION,
|
|
session=Depends(get_session)):
|
|
return utils.get_single_record(session, alchemy.Country, country_id)
|
|
|
|
|
|
@router.post("/",
|
|
response_model=CountryPublic)
|
|
async def create_country(
|
|
country: CountryCreate,
|
|
current_user: ACTIVE_USER,
|
|
session=Depends(get_session)):
|
|
return create_item(
|
|
session=session,
|
|
cls=alchemy.Country,
|
|
current_user=current_user,
|
|
data=Country.model_validate(country))
|
|
|
|
|
|
@router.patch("/{country_id}",
|
|
response_model=CountryPublic,
|
|
responses={404: {"description": "Not found"}})
|
|
async def update_country(
|
|
country_id: PRIMARY_ANNOTATION,
|
|
country: CountryUpdate,
|
|
current_user: ACTIVE_USER,
|
|
session=Depends(get_session)):
|
|
return update_item(
|
|
session=session,
|
|
current_user=current_user,
|
|
item=utils.get_single_record(session, alchemy.Country, country_id),
|
|
data=country)
|
|
|
|
|
|
@router.put("/{country_id}/activate",
|
|
response_model=CountryPublic,
|
|
responses={404: {"description": "Not found"}})
|
|
async def activate_country(
|
|
country_id: PRIMARY_ANNOTATION,
|
|
current_user: ACTIVE_USER,
|
|
session=Depends(get_session)):
|
|
return utils.set_item_status(
|
|
session=session,
|
|
current_user=current_user,
|
|
item=utils.get_single_record(session, alchemy.Country, country_id),
|
|
status='A')
|
|
|
|
|
|
@router.put("/{country_id}/deactivate",
|
|
response_model=CountryPublic,
|
|
responses={404: {"description": "Not found"}})
|
|
async def deactivate_country(
|
|
country_id: PRIMARY_ANNOTATION,
|
|
current_user: ACTIVE_USER,
|
|
session=Depends(get_session)):
|
|
return utils.set_item_status(
|
|
session=session,
|
|
current_user=current_user,
|
|
item=utils.get_single_record(session, alchemy.Country, country_id),
|
|
status='I')
|