rohskiddo/UMBRA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Fix CompoundSelect chaining: use standalone union_all()

Browse Source 
SQLAlchemy 2.0's select().union_all() returns a CompoundSelect which
cannot chain another .union_all(). Use the standalone union_all()
function to combine all three queries.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Kyle 2026-03-16 20:39:40 +08:00
parent 8f087ccebf
commit 0401a71fce
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
backend/app/services/calendar_sharing.py
View File

@ -7,7 +7,7 @@ import logging
from datetime import datetime, timedelta from datetime import datetime, timedelta
from fastapi import HTTPException from fastapi import HTTPException
from sqlalchemy import delete, literal_column, select, text, update from sqlalchemy import delete, literal_column, select, text, union_all, update
from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy.ext.asyncio import AsyncSession
from app.models.calendar import Calendar from app.models.calendar import Calendar
@ -45,21 +45,19 @@ async def get_accessible_event_scope(
from app.models.event_invitation import EventInvitation from app.models.event_invitation import EventInvitation
result = await db.execute( result = await db.execute(
select(literal_column("'c'").label("kind"), Calendar.id.label("val")) union_all(
.where(Calendar.user_id == user_id) select(literal_column("'c'").label("kind"), Calendar.id.label("val"))
.union_all( .where(Calendar.user_id == user_id),
select(literal_column("'c'"), CalendarMember.calendar_id) select(literal_column("'c'"), CalendarMember.calendar_id)
.where( .where(
CalendarMember.user_id == user_id, CalendarMember.user_id == user_id,
CalendarMember.status == "accepted", CalendarMember.status == "accepted",
) ),
)
.union_all(
select(literal_column("'i'"), EventInvitation.event_id) select(literal_column("'i'"), EventInvitation.event_id)
.where( .where(
EventInvitation.user_id == user_id, EventInvitation.user_id == user_id,
EventInvitation.status != "declined", EventInvitation.status != "declined",
) ),
) )
) )
cal_ids: list[int] = [] cal_ids: list[int] = []