Yadciel/Titan
1
0
Fork 0
mirror of https://github.com/TitanEmbeds/Titan.git synced 2025-07-04 20:25:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Show global header message when bot loses connection with server

Browse Source
This commit is contained in:
Jeremy Zhang
2017-06-13 01:39:49 +00:00
parent 926d4d14a9
commit 2647753007
8 changed files with 78 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
discordbot/titanembeds/bot.py
View File

@ -17,6 +17,9 @@ class Titan(discord.Client):
self.http.user_agent += ' TitanEmbeds-Bot'
self.database = DatabaseInterface(self)
self.command = Commands(self, self.database)
self.database_connected = False
self.loop.create_task(self.send_webserver_heartbeat())
def _cleanup(self):
try:
@ -31,6 +34,14 @@ class Titan(discord.Client):
gathered.exception()
except: # Can be ignored
pass
async def send_webserver_heartbeat(self):
await self.wait_until_ready()
while not self.database_connected:
await asyncio.sleep(1) # Wait until db is connected
while not self.is_closed:
await self.database.send_webserver_heartbeat()
await asyncio.sleep(60)
def run(self):
try:
@ -57,6 +68,7 @@ class Titan(discord.Client):
try:
await self.database.connect(config["database-uri"] + "?charset=utf8mb4")
self.database_connected = True
except Exception:
self.logger.error("Unable to connect to specified database!")
traceback.print_exc()