From 3e18b36e130fc5b8e90abda5f471ca656be6901e Mon Sep 17 00:00:00 2001 From: gmankab Date: Thu, 20 Jun 2024 20:51:34 +0400 Subject: [PATCH] [+] planets and sectors api --- galaxy_backend/api/planets.py | 38 +++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) diff --git a/galaxy_backend/api/planets.py b/galaxy_backend/api/planets.py index 7377daf..17e6f80 100644 --- a/galaxy_backend/api/planets.py +++ b/galaxy_backend/api/planets.py @@ -17,3 +17,41 @@ async def sector_list() -> list[dict]: }) return sectors_list + +@api.routers.planet.get('/list') +async def planet_list() -> list[dict]: + ''' + list planets + ''' + planets_list: list[dict] = [] + planets_models = await models.db.Planet.all().prefetch_related('sector') + for planet in planets_models: + planets_list.append({ + 'id': planet.id, + 'sector': planet.sector.id, + 'total_resources': planet.total_resources, + 'mined_resources': planet.mined_resources, + 'available': planet.available, + }) + return planets_list + + +@api.routers.sector.get('/get') +async def sector_get( + sector_id: int +) -> list[dict]: + ''' + get planets from sector + ''' + planets_list: list[dict] = [] + planets_models = await models.db.Planet.filter(sector_id=sector_id).prefetch_related('sector') + for planet in planets_models: + planets_list.append({ + 'id': planet.id, + 'sector': planet.sector.id, + 'total_resources': planet.total_resources, + 'mined_resources': planet.mined_resources, + 'available': planet.available, + }) + return planets_list +