khawir commited on
Commit
a79c78b
·
1 Parent(s): 7180025

added endpoint for site hosts

Browse files
Files changed (2) hide show
  1. models.py +4 -0
  2. routers/profile.py +12 -1
models.py CHANGED
@@ -218,3 +218,7 @@ class SitePublicMeWith(SitePublicMe):
218
  visits: list[VisitPublicWithGuest] = []
219
  hosts: list[GuestPublicMe] = []
220
 
 
 
 
 
 
218
  visits: list[VisitPublicWithGuest] = []
219
  hosts: list[GuestPublicMe] = []
220
 
221
+ class SitePublicMeWithHosts(SitePublicMe):
222
+ hosts: list[GuestPublicMe] = []
223
+
224
+
routers/profile.py CHANGED
@@ -4,7 +4,7 @@ from fastapi import Depends, APIRouter
4
  from sqlmodel import Session
5
  from models import User, UserPublicMe, UserPublicMeWith, UserUpdateMe
6
  from models import Site, SiteCreate, SitePublicMe, SitePublicMeWith, SiteUpdate
7
- from models import Guest, GuestCreateMe, GuestPublicMe, GuestPublicWith, GuestUpdateMe
8
  from core import crud, utils
9
 
10
  router = APIRouter(
@@ -138,6 +138,17 @@ def create_host(*,
138
 
139
 
140
 
 
 
 
 
 
 
 
 
 
 
 
141
  @router.get("/sites/{site_id}/hosts/{host_id}", response_model=GuestPublicWith)
142
  def read_host(*,
143
  session: Session = Depends(utils.get_session),
 
4
  from sqlmodel import Session
5
  from models import User, UserPublicMe, UserPublicMeWith, UserUpdateMe
6
  from models import Site, SiteCreate, SitePublicMe, SitePublicMeWith, SiteUpdate
7
+ from models import Guest, GuestCreateMe, GuestPublicMe, GuestPublicWith, GuestUpdateMe, SitePublicMeWithHosts
8
  from core import crud, utils
9
 
10
  router = APIRouter(
 
138
 
139
 
140
 
141
+ @router.get("/sites/{site_id}/hosts", response_model=SitePublicMeWithHosts)
142
+ def read_hosts(*,
143
+ session: Session = Depends(utils.get_session),
144
+ current_user: Annotated[User, Depends(crud.get_current_active_user)],
145
+ site_id: int,
146
+ ):
147
+ current_site = crud.get_current_site(session, current_user, site_id)
148
+ return current_site
149
+
150
+
151
+
152
  @router.get("/sites/{site_id}/hosts/{host_id}", response_model=GuestPublicWith)
153
  def read_host(*,
154
  session: Session = Depends(utils.get_session),