Pull is_guest up into the RegistrationWorkerStore

synapse/storage/databases/main/registration.py

@@ -1744,6 +1744,18 @@ def _replace_refresh_token_txn(txn: LoggingTransaction) -> None:
  "replace_refresh_token", _replace_refresh_token_txn
  )
 
+ @cached()
+ async def is_guest(self, user_id: str) -> bool:
+ res = await self.db_pool.simple_select_one_onecol(
+ table="users",
+ keyvalues={"name": user_id},
+ retcol="is_guest",
+ allow_none=True,
+ desc="is_guest",
+ )
+
+ return res if res else False
+
 
 class RegistrationBackgroundUpdateStore(RegistrationWorkerStore):
  def __init__(
@@ -1886,18 +1898,6 @@ def set_user_deactivated_status_txn(self, txn, user_id: str, deactivated: bool):
  self._invalidate_cache_and_stream(txn, self.get_user_by_id, (user_id,))
  txn.call_after(self.is_guest.invalidate, (user_id,))
 
- @cached()
- async def is_guest(self, user_id: str) -> bool:
- res = await self.db_pool.simple_select_one_onecol(
- table="users",
- keyvalues={"name": user_id},
- retcol="is_guest",
- allow_none=True,
- desc="is_guest",
- )
-
- return res if res else False
-
 
 class RegistrationStore(StatsStore, RegistrationBackgroundUpdateStore):
  def __init__(