Properly handle networks when looking up users' online status

This commit is contained in:
Mark Veidemanis 2022-07-29 17:51:19 +01:00
parent 0457cfbbed
commit 654c9960ba
Signed by: m
GPG Key ID: 5ACFCEED46C0904F
1 changed files with 21 additions and 18 deletions

View File

@ -44,18 +44,21 @@ def annotate_results(results_parsed):
for net in nets: for net in nets:
# Annotate the online attribute from Threshold # Annotate the online attribute from Threshold
online_info = annotate_online( nicks = [
net, [x["nick"] for x in results_parsed if x["src"] == "irc"] x["nick"] for x in results_parsed if x["src"] == "irc" and x["net"] == net
) ]
channels = [
x["channel"]
for x in results_parsed
if x["src"] == "irc" and x["net"] == net
]
online_info = annotate_online(net, nicks)
# Annotate the number of users in the channel # Annotate the number of users in the channel
num_users = annotate_num_users( num_users = annotate_num_users(net, channels)
net, [x["channel"] for x in results_parsed if x["src"] == "irc"]
)
# Annotate the number channels the user is on # Annotate the number channels the user is on
num_chans = annotate_num_chans( num_chans = annotate_num_chans(net, nicks)
net, [x["nick"] for x in results_parsed if x["src"] == "irc"]
)
for item in results_parsed: for item in results_parsed:
if item["net"] == net:
if "nick" in item: if "nick" in item:
if item["nick"] in online_info: if item["nick"] in online_info:
item["online"] = online_info[item["nick"]] item["online"] = online_info[item["nick"]]