brain.py

import json
import aiohttp
import logging
import random

logging.getLogger('asyncio').setLevel(logging.CRITICAL)

class Brain:
    def __init__(self, user, key, nick, loop):
        self.user = user
        self.key = key
        self.nick = nick
        self.body = {
                    'user': self.user,
                    'key': self.key,
                    'nick': self.nick
                }
        self.loop = loop
        self.sess = aiohttp.ClientSession(loop=self.loop, raise_for_status=True, headers={"user-Agent": "HyperAI/2.0"})
        print("Initiated Brain ClientSession.")

    async def create(self):
        try:
            async with self.sess.post('https://cleverbot.io/1.0/create', json=self.body, timeout=5) as resp:
                if resp.status == 200:
                    r = await resp.json()
                    if r["status"] is not "success":
                        self.nick += '-{}'.format(random.randint(1,1000))
                        self.body['nick'] = self.nick
                        await self.create()
                    return ("API is online. Using clever mode.", 200)
                else:
                    return ("API is down. Will notify the users.", 500)
        except:
            return ("API is down. Will notify the users.", 500)
        
    async def query(self, text):
        self.body['text'] = text
        try:
            async with self.sess.post('https://cleverbot.io/1.0/ask', json=self.body) as resp:
                r = await resp.json()

            if r['status'] == 'success':
                return r['response'], 200
        except:
            return "Looks like the CleverBot.IO api is down. Please try again later.", 500