Discordbot database setup inital/partial

discordbot/titanembeds/bot.py

@@ -1,17 +1,57 @@
 from config import config
+from titanembeds.database import DatabaseInterface
 import discord
+import aiohttp
+import asyncio
 
-client = discord.Client()
+class Titan(discord.Client):
+    def __init__(self):
+        super().__init__()
+        self.aiosession = aiohttp.ClientSession(loop=self.loop)
+        self.http.user_agent += ' TitanEmbeds-Bot'
+        self.database = DatabaseInterface(self)
 
-@client.event
-async def on_ready():
-    print('Titan -- DiscordBot')
-    print('Logged in as the following user:')
-    print(client.user.name)
-    print(client.user.id)
-    print('------')
-    await test()
+    def _cleanup(self):
+        try:
+            self.loop.run_until_complete(self.logout())
+        except: # Can be ignored
+            pass
+        pending = asyncio.Task.all_tasks()
+        gathered = asyncio.gather(*pending)
+        try:
+            gathered.cancel()
+            self.loop.run_until_complete(gathered)
+            gathered.exception()
+        except: # Can be ignored
+            pass
 
-async def test():
-    from titanembeds.database import db, Guilds, session
-    session.query(Guilds).all()
+    def run(self):
+        try:
+            self.loop.run_until_complete(self.start(config["bot-token"]))
+        except discord.errors.LoginFailure:
+            print("Invalid bot token in config!")
+        finally:
+            try:
+                self._cleanup()
+            except Exception as e:
+                print("Error in cleanup:", e)
+            self.loop.close()
+
+    async def on_ready(self):
+        print('Titan [DiscordBot]')
+        print('Logged in as the following user:')
+        print(self.user.name)
+        print(self.user.id)
+        print('------')
+
+        await self.change_presence(
+            game=discord.Game(name="Get your own @ https://TitanEmbeds.tk/"), status=discord.Status.online
+        )
+
+        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