vbatts/microblog.pub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Improve lookup and handle visibility in threads

Browse source
This commit is contained in:
Thomas Sileo 2022-08-15 21:34:57 +02:00
parent 59af633c6c
commit d1b4bd0181
4 changed files with 38 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

10
app/actor.py
View file

@ -168,6 +168,7 @@ async def save_actor(db_session: AsyncSession, ap_actor: ap.RawObject) -> "Actor
async def fetch_actor(
db_session: AsyncSession,
actor_id: str,
save_if_not_found: bool = True,
) -> "ActorModel":
if actor_id == LOCAL_ACTOR.ap_id:
raise ValueError("local actor should not be fetched")
@ -180,9 +181,12 @@ async def fetch_actor(
).one_or_none()
if existing_actor:
return existing_actor
ap_actor = await ap.fetch(actor_id)
return await save_actor(db_session, ap_actor)
else:
if save_if_not_found:
ap_actor = await ap.fetch(actor_id)
return await save_actor(db_session, ap_actor)
else:
raise ap.ObjectNotFoundError
@dataclass