diff --git a/endpoints/country.get.py b/endpoints/country.get.py index 41c21fa..952c3a0 100644 --- a/endpoints/country.get.py +++ b/endpoints/country.get.py @@ -1,7 +1,19 @@ -from fastapi import APIRouter +# Entity: Country -router = APIRouter(); +from fastapi import APIRouter, Depends, HTTPException, status +from sqlalchemy.orm import Session +from typing import List +from core.database import get_db +from models.country import Country +from schemas.country import CountrySchema +from helpers.country_helpers import get_countries_by_name_start -@router.get("/country") -def get_countries(): - return ["USA", "Canada", "Germany", "France", "Japan", "Australia"] +router = APIRouter() + +@router.get("/country", response_model=List[CountrySchema]) +async def get_countries( + search: str = None, + db: Session = Depends(get_db) +): + countries = get_countries_by_name_start(db, search) + return countries \ No newline at end of file