From df8c0c44cf611917a3f9267f5ca0e9db29cd91e1 Mon Sep 17 00:00:00 2001
From: David Robertson <davidr@element.io>
Date: Fri, 29 Jul 2022 16:59:03 +0100
Subject: Handle events that we don't have the state for(?)

---
 scripts-dev/debug_state_res.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'scripts-dev')

diff --git a/scripts-dev/debug_state_res.py b/scripts-dev/debug_state_res.py
index 8896ea7e16..02b8414809 100755
--- a/scripts-dev/debug_state_res.py
+++ b/scripts-dev/debug_state_res.py
@@ -235,10 +235,14 @@ async def debug_specific_stateres(
         watch_func: Optional[Callable[[EventBase], Awaitable[str]]]
 
         async def watch_func(event: EventBase) -> str:
-            result = await hs.get_storage_controllers().state.get_state_ids_for_event(
-                event.event_id, filter
-            )
-            return f"\n{key_pair}: {result.get(key_pair, '<Missing>')}"
+            try:
+                result = await hs.get_storage_controllers().state.get_state_ids_for_event(
+                    event.event_id, filter
+                )
+            except RuntimeError:
+                return f"\n{key_pair}: <Event unavailable :(>"
+            else:
+                return f"\n{key_pair}: {result.get(key_pair, '<No event in state>')}"
 
     else:
         watch_func = None
-- 
cgit 1.5.1