liz/PluralKit
liz/PluralKit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remember to await webhook stat submission

Browse Source
This commit is contained in:
Ske 2018-07-25 23:06:04 +02:00
parent e5612a7b41
commit 034f7816e4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/pluralkit/bot/proxy.py
View File

@ -171,7 +171,7 @@ class Proxy:
message = await resp.json() message = await resp.json()
# Report webhook stats to Influx # Report webhook stats to Influx
stats.report_webhook(time.perf_counter() - time_before, True) await stats.report_webhook(time.perf_counter() - time_before, True)
await db.add_message(conn, message["id"], message["channel_id"], member.id, original_message.author.id, await db.add_message(conn, message["id"], message["channel_id"], member.id, original_message.author.id,
text or "") text or "")
@ -211,7 +211,7 @@ class Proxy:
message_id=message["id"]) message_id=message["id"])
elif resp.status == 404 and not has_already_retried: elif resp.status == 404 and not has_already_retried:
# Report webhook stats to Influx # Report webhook stats to Influx
stats.report_webhook(time.perf_counter() - time_before, False) await stats.report_webhook(time.perf_counter() - time_before, False)
# Webhook doesn't exist. Delete it from the DB, create, and add a new one # Webhook doesn't exist. Delete it from the DB, create, and add a new one
self.logger.warning("Webhook registered in DB doesn't exist, deleting hook from DB, re-adding, and trying again (channel={}, hook={})".format(original_message.channel.id, hook_id)) self.logger.warning("Webhook registered in DB doesn't exist, deleting hook from DB, re-adding, and trying again (channel={}, hook={})".format(original_message.channel.id, hook_id))
@ -222,7 +222,7 @@ class Proxy:
return await self.do_proxy_message(conn, member, original_message, text, attachment_url, has_already_retried=True) return await self.do_proxy_message(conn, member, original_message, text, attachment_url, has_already_retried=True)
else: else:
# Report webhook stats to Influx # Report webhook stats to Influx
stats.report_webhook(time.perf_counter() - time_before, False) await stats.report_webhook(time.perf_counter() - time_before, False)
raise discord.HTTPException(resp, await resp.text()) raise discord.HTTPException(resp, await resp.text())