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

Fix gino bugs, reimplement ban and revoke cmds, enforce postgresql

Browse Source
This commit is contained in:
Jeremy Zhang
2018-07-10 19:06:54 +00:00
parent ac225c84f5
commit de94f87c45
4 changed files with 72 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
discordbot/titanembeds/bot.py
View File

@ -45,7 +45,7 @@ class Titan(discord.AutoShardedClient):
def run(self):
try:
self.loop.run_until_complete(self.start(config["bot-token"]))
self.loop.run_until_complete(self.start())
except discord.errors.LoginFailure:
print("Invalid bot token in config!")
finally:
@ -54,6 +54,10 @@ class Titan(discord.AutoShardedClient):
except Exception as e:
print("Error in cleanup:", e)
self.loop.close()
async def start(self):
await self.database.connect(config["database-uri"])
await super().start(config["bot-token"])
async def on_ready(self):
print('Titan [DiscordBot]')
@ -66,14 +70,6 @@ class Titan(discord.AutoShardedClient):
game = discord.Game(name="Embed your Discord server! Visit https://TitanEmbeds.com/")
await self.change_presence(status=discord.Status.online, activity=game)
try:
await self.database.connect(config["database-uri"])
except Exception:
self.logger.error("Unable to connect to specified database!")
traceback.print_exc()
await self.logout()
return
self.discordBotsOrg = DiscordBotsOrg(self.user.id, config.get("discord-bots-org-token", None))
self.botsDiscordPw = BotsDiscordPw(self.user.id, config.get("bots-discord-pw-token", None))