20 lines
674 B
Python
20 lines
674 B
Python
|
from fastapi import APIRouter, HTTPException
|
||
|
|
|
||
|
|
from app.schemas import EntityResponse, ListResponse, NetObject
|
||
|
|
from app.services.store import OBJECTS
|
||
|
|
|
||
|
|
router = APIRouter(prefix='/objects', tags=['objects'])
|
||
|
|
|
||
|
|
|
||
|
|
@router.get('', response_model=ListResponse[NetObject])
|
||
|
|
def list_objects() -> ListResponse[NetObject]:
|
||
|
|
return ListResponse(count=len(OBJECTS), items=OBJECTS)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get('/{object_id}', response_model=EntityResponse[NetObject])
|
||
|
|
def get_object(object_id: str) -> EntityResponse[NetObject]:
|
||
|
|
for item in OBJECTS:
|
||
|
|
if item.id == object_id:
|
||
|
|
return EntityResponse(item=item)
|
||
|
|
raise HTTPException(status_code=404, detail='Object not found')
|