diff --git a/endpoints/countries.get.py b/endpoints/countries.get.py index 5ddc127..eb4fd13 100644 --- a/endpoints/countries.get.py +++ b/endpoints/countries.get.py @@ -1,21 +1,18 @@ # Entity: Country -from fastapi import APIRouter, HTTPException, status +from fastapi import APIRouter, Depends, HTTPException, status +from sqlalchemy.orm import Session from typing import List -import httpx +from core.database import get_db from core.models.country import Country from core.schemas.country import CountrySchema router = APIRouter() @router.get("/countries", status_code=200, response_model=List[CountrySchema]) -async def get_countries(): - """Get list of countries from external API""" - async with httpx.AsyncClient() as client: - response = await client.get("https://restcountries.com/v3.1/all") - if response.status_code != 200: - raise HTTPException( - status_code=status.HTTP_503_SERVICE_UNAVAILABLE, - detail="Unable to fetch countries from external API" - ) - return response.json() \ No newline at end of file +async def get_countries( + db: Session = Depends(get_db) +): + """Get all countries""" + countries = db.query(Country).all() + return countries \ No newline at end of file