keybaser.py

import time
import traceback
import discord
import logging

from discord.ext import commands
import kbutils as utils
import config as kb_config

bot = commands.Bot(command_prefix=['kb!', 'Kb!', 'kB!', 'KB!'])
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger('keybaser')

_fields = ['basics', 'profile', 'proofs_summary', 'pictures']

def is_owner():
    return commands.check(lambda ctx: ctx.message.author.id == kb_config.owner_id)

@bot.event
async def on_ready():
    print('!!READY!!')
    print(bot.user.name)
    print(bot.user.id)
    print('------')

    g = discord.Game(name='kb!help', url='fucker')
    await bot.change_presence(game=g)

@bot.command(pass_context=True)
async def ping(ctx):
    """Ping the fucker."""
    tinit = time.monotonic()
    m = await bot.say('pong')
    tfinish = time.monotonic()

    delta = (tfinish - tinit) * 1000
    await bot.edit_message(m, '`%.2fms`' % (delta))

@bot.command(pass_context=True)
async def lookup(ctx, user : str, location : str = ''):
    """Find a keybase profile"""
    if len(location.strip()) < 1:
        location = None

    try:
        res = None
        if location is None:
            res = await utils.kblookup(user, 'usernames', _fields)
        else:
            res = await utils.kblookup(user, location, _fields)
    except Exception as err:
        await bot.say("fuck: %r", err)
        logger.error('lookup cmd', exc_info=True)
        return

    logger.info("[lookup] %d bytes", len(str(res)))

    if res is None:
        await bot.say("fuckfuckfuck this shouldn't happen")
        return

    try:
        if len(res['them']) < 1:
            await bot.say("No data found for `%s`" % user)
            return

        try:
            userdata = res['them'][0]
        except Exception as err:
            await bot.say(":warning: error getting `userdata` for `%s`\n`%r" % (user, err))
            return

        try:
            user_id = userdata['id']
        except Exception as err:
            await bot.say(":warning: error getting user_id for `%s`\n`%r" % (user, err))
            return

        basics = userdata.get('basics')
        profile = userdata.get('profile')
        proofs = userdata.get('proofs_summary')
        pictures = userdata.get('pictures')

        if basics is None:
            await bot.say("`%r`: No basic data." % user)
            return

        if proofs is None:
            await bot.say("`%r`: No proofs" % user)
            return

        username = basics.get('username')
        if username is None:
            await bot.say("username is None???????????")
            return

        em = discord.Embed(title=username, colour=utils.mkcolor(username))

        if pictures is not None:
            em.set_thumbnail(url=pictures['primary']['url'])

        em.set_footer(text='userid %s, got %d bytes from API' % (user_id, len(str(res))))

        if profile is not None:
            em.add_field(name='Name', value=profile['full_name'])
            em.add_field(name='Location', value=profile['location'])
            em.add_field(name='Bio', value=profile['bio'])

        # proofs
        for proof_key in proofs['by_proof_type']:
            proof = proofs['by_proof_type'][proof_key][0]

            em.add_field(name='{}'.format(proof_key.capitalize()), \
                value='[{}]({})'.format(proof['nametag'], proof['proof_url']))

        await bot.say(embed=em)
    except:
        await bot.say('```\n%s\n```' % traceback.format_exc())

@is_owner()
@bot.command()
async def apiping():
    """Ping the Keybase server"""

    delay = await utils.make_ping(bot.loop, 'keybase.io')
    await bot.say('%s transmitted/%s received: min/avg/max `%s/%s/%s`ms' % \
        (delay[0], delay[1], delay[3], delay[4], delay[5]))

@is_owner()
@bot.command()
async def avatar():
    """Change Keybaser's avatar to the current avatar.png file"""
    try:
        avatar_file = open('avatar.png', 'rb')
        data = avatar_file.read()
        await bot.say('`%r`' % avatar_file)
        await bot.edit_profile(avatar=data)
    except discord.HTTPException:
        await bot.say(":warning: Editing the profile failed.")
    except discord.InvalidArgument:
        await bot.say(":warning: Invalid Argument")
    else:
        await bot.say("Sucksés. :ok_hand:")

    return

@is_owner()
@bot.command()
async def shutdown():
    """Self-explanatory"""
    await bot.say(":wave: bye u bitch!!!!!!! :wave:")
    await bot.logout()

async def _say(m, string):
    await bot.send_message(m.channel, string)

@bot.event
async def on_message(message):
    try:
        await bot.process_commands(message)
    except Exception as err:
        await _say(message, "fuck, we got error: ```%s```" % (traceback.format_exc()))

print("==run this shit==")
bot.run(kb_config.token)
print("\nrip me\n")