lucasjensen/TheGrapefruitsDuo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added logging and reworked Series query

Browse Source
This commit is contained in:
Lucas Jensen
2024-05-01 10:37:41 -07:00
parent 04b3f390de
commit 8a24a5e979
13 changed files with 89 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
server/app/controllers/events.py
View File

@@ -19,34 +19,34 @@ class EventController(BaseController):
for event_series_row in data:
series_name: str = event_series_row["name"]
event = Event(**event_series_row)
if series_name not in all_series:
all_series[series_name] = EventSeries(**event_series_row, events=[])
all_series[series_name].events.append(event)
if event_series_row.get("event_id"):
all_series[series_name].events.append(Event(**event_series_row))
return [series for series in all_series.values()]
async def get_all_series(self) -> list[EventSeries]:
data = await self.db.get_all()
series_data = await self.db.select_all_series()
try:
return self._all_series(data)
return self._all_series(series_data)
except Exception as e:
ic(e)
self.log_error(e)
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
detail=f"Error retrieving event objects: {e}",
)
async def get_one_series(self, id: int) -> EventSeries:
if not (data := await self.db.get_one(id)):
async def get_one_series_by_id(self, series_id: int) -> EventSeries:
if not (data := await self.db.select_one_series_by_id(series_id)):
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND, detail="Event not found"
)
try:
event = EventSeries(
**data[0], events=[Event(**e) for e in data if e.get("event_id")]
return EventSeries(
**data[0], events=[Event(**e) for e in data if e["event_id"]]
)
return event
except Exception as e:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
@@ -58,7 +58,7 @@ class EventController(BaseController):
inserted_id = await self.db.insert_one_series(series)
for new_event in series.events:
await self.db.insert_one_event(new_event, inserted_id)
return await self.get_one_series(inserted_id)
return await self.get_one_series_by_id(inserted_id)
except IntegrityError as e:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -66,10 +66,10 @@ class EventController(BaseController):
)
async def add_series_poster(self, series_id, poster: UploadFile) -> EventSeries:
series = await self.get_one_series(series_id)
series = await self.get_one_series_by_id(series_id)
series.poster_id = await self._upload_poster(poster)
await self.db.update_series_poster(series)
return await self.get_one_series(series.series_id)
return await self.get_one_series_by_id(series.series_id)
async def _upload_poster(self, poster: UploadFile) -> str:
image_file = await self.verify_image(poster)
@@ -83,7 +83,7 @@ class EventController(BaseController):
)
async def delete_series(self, id: int) -> None:
series = await self.get_one_series(id)
series = await self.get_one_series_by_id(id)
await self.db.delete_one_series(series)
async def update_series(self, route_id: int, series: EventSeries) -> EventSeries:
@@ -93,7 +93,7 @@ class EventController(BaseController):
status_code=status.HTTP_400_BAD_REQUEST,
detail="ID in URL does not match ID in request body",
)
prev_series = await self.get_one_series(series.series_id)
prev_series = await self.get_one_series_by_id(series.series_id)
if series.poster_id != prev_series.poster_id:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -103,4 +103,4 @@ class EventController(BaseController):
await self.db.replace_series(series)
for event in series.events:
await self.db.insert_one_event(event, series.series_id)
return await self.get_one_series(series.series_id)
return await self.get_one_series_by_id(series.series_id)