Fix result annotation for when certain fields do not exist

2 years ago · 159a0bd0a1
parent eb903fe50b
commit 159a0bd0a1
1 changed files with 9 additions and 6 deletions
--- a/core/lib/opensearch.py
+++ b/core/lib/opensearch.py
@ -56,12 +56,15 @@ def annotate_results(results_parsed):
            net, [x["nick"] for x in results_parsed if x["src"] == "irc"]
        )
        for item in results_parsed:
-            if item["nick"] in online_info:
-                item["online"] = online_info[item["nick"]]
-            if item["channel"] in num_users:
-                item["num_users"] = num_users[item["channel"]]
-            if item["nick"] in num_chans:
-                item["num_chans"] = num_chans[item["nick"]]
+            if "nick" in item:
+                if item["nick"] in online_info:
+                    item["online"] = online_info[item["nick"]]
+            if "channel" in item:
+                if item["channel"] in num_users:
+                    item["num_users"] = num_users[item["channel"]]
+            if "nick" in item:
+                if item["nick"] in num_chans:
+                    item["num_chans"] = num_chans[item["nick"]]


 def filter_blacklisted(user, response):