TekkenBot.py

import discord
import asyncio
import logging
from discord.ext import commands

# self-created modules below
import lib.embedCreation as embedCreation #contains functions for creating an embed
import lib.tekkenFinder as tekkenFinder   #contains functions for finding character and move details

# Get token from local dir text file
tokenFile = open("token.txt", 'r')
token = tokenFile.read()
tokenFile.close()

description = 'A Tekken 7 Frame Data Bot made by Hann.'

prefix = '.'

discord_logger = logging.getLogger('discord')
discord_logger.setLevel(logging.CRITICAL)
log = logging.getLogger()
log.setLevel(logging.INFO)
handler = logging.FileHandler(filename='combot.log', encoding='utf-8', mode='w')
log.addHandler(handler)

bot = commands.Bot(command_prefix=prefix, description=description)

combot_gagged_channels_File = open("lib/gagged_channels.txt", 'r')
combot_gagged_channels = combot_gagged_channels_File.read().splitlines()
combot_gagged_channels_File.close()

# TODO: YOU LEFT OFF HERE
# TODO: MAYBE NEXT TIME FAM
# file = open('bot_settings.json', 'r+')
# content = file.read()
# file.close()
# stuff = content.loads(content)

@bot.event
async def on_ready():
    # Display Login Status in Console
    print('<---------------------------->')
    print('Logged in as')
    print(bot.user.name)
    print(bot.user.id)
    print('<---------------------------->')
    while True:
        await bot.change_presence(game=discord.Game(name='DAH'))
        await asyncio.sleep(30)
        await bot.change_presence(game=discord.Game(name='.help'))
        await asyncio.sleep(120)
        await bot.change_presence(game=discord.Game(name='Riri Toppu Tieru'))
        await asyncio.sleep(30)
        await bot.change_presence(game=discord.Game(name='TEKKEN 7'))
        await asyncio.sleep(30)

@bot.event
async def on_message(message):
    # check if message author is a bot
    if message.author.bot:
        # check if sent by self
        if message.author.id == bot.user.id:
            await bot_message_cleanup(message)
        return
    if await is_Gagged(message):
        return

    if message.content.startswith('!'):
        if message.content.startswith('!!'):
            case_sensitive_toggle = True
        else:
            case_sensitive_toggle = False

        # message content should look like this
        # ![character] [move]

        userMessage = message.content
        userMessage = userMessage.replace("!", "")
        user_message_list = userMessage.split(" ", 1)

        if len(user_message_list) <= 1:
            print('! command used, but character not found/move not given\n')
            return

        user_Chara_Name = user_message_list[0]
        user_Chara_Move = user_message_list[1]

        #TODO: IMPLEMENT CHARACTER SHORTHAND NAME CONVERTER, OR CHARACTER NAMELIST DISPLAY
        character_name_string = tekkenFinder.does_char_exist(user_Chara_Name)

        if character_name_string:
            user_Chara_Name = character_name_string.lower()
            move_attribute_dict = tekkenFinder.get_Move_Details(user_Chara_Name,
                                                                user_Chara_Move,
                                                                case_sensitive_toggle)

            if move_attribute_dict:  # if dictionary not empty, move found
                embed_MoveFound = await get_MoveFound_Embed(**move_attribute_dict)
                await bot.send_message(message.channel, embed=embed_MoveFound)

            else:  # dictionary is empty, move not found
                embed_SimilarMoves = await get_SimilarMoves_Embed(user_Chara_Name,user_Chara_Move)
                await bot.send_message(message.channel, embed=embed_SimilarMoves)

            await user_message_cleanup(message)
            return

        else:
            await bot.send_message(message.channel, 'Character not found: ' + '**' + user_Chara_Name + '**')
            return

    await bot.process_commands(message)


@bot.command(pass_context=True)
async def legend(ctx):
    """Displays commonly used abbreviations, notations and their corresponding input icons."""
    embed_legend = embedCreation.embed_legend()
    await bot.say(embed=embed_legend)
    await user_message_cleanup(ctx.message)


@bot.command(pass_context=True)
@commands.has_permissions(administrator=True)
async def gagcombot(ctx):
    """Gags Combot in this channel. Usable only by admin roles."""
    channel = ctx.message.channel.id
    combot_gagged_channels.append(channel)

    f = open("lib/gagged_channels.txt","a")
    f.write(channel + '\n')
    f.close()

    await bot.say('Mmmph! Gagging Combot.')


@bot.command(pass_context=True)
@commands.has_permissions(administrator=True)
async def ungagcombot(ctx):
    """Ungags Combot in this channel. Usable only by server admins."""
    channel = ctx.message.channel.id
    if channel in combot_gagged_channels:
        combot_gagged_channels.remove(channel)
    else:
        return
    # clear file contents and rewrite
    open("lib/gagged_channels.txt", "w").close()
    f = open("lib/gagged_channels.txt", "a")
    for channel in combot_gagged_channels:
        f.write(channel+'\n')
    f.close()

    await bot.say('Ungagged Combot. Beep Boop.')


@bot.command(pass_context=True)
async def printServers(ctx):
    """List servers with Combot. Cmd restricted to Bot Owner."""
    appinfo = await bot.application_info()
    owner = appinfo.owner.id

    if ctx.message.author.id != owner:
        print('Non-bot owner called print server.')
        await bot.say('Command restricted to bot owner only.')
        await user_message_cleanup(ctx.message)
        return
    else:
        print('Bot Owner called print server.')

    serverConctStr = ''
    for server in bot.servers:
        serverConctStr = serverConctStr + server.name + '\n'
    await bot.say('Server List: \n' + serverConctStr)
    await user_message_cleanup(ctx.message)


@bot.command(pass_context=True)
async def Frame_Data(ctx):
    """Use ![character] [move], !! for case-sensitive search"""
    await user_message_cleanup(ctx.message)
    return


@bot.command(pass_context=True)
async def invite(ctx):
    """Invite the bot to your server."""
    await bot.say('Use this link to add me to your server. \nhttps://discordapp.com/oauth2/authorize?client_id=302295833208946689&scope=bot&permissions=11264')
    await user_message_cleanup(ctx.message)
    return


# This block of code to be used when character html pages are updated, do not edit
@bot.command(pass_context=True)
async def convertAll(ctx):
    """Converts all """
    appinfo = await bot.application_info()
    owner = appinfo.owner.id

    if ctx.message.author.id != owner:
        return
    else:
        await bot.say('Converting all character htmls to json.')
    tekkenFinder.charJsonMassConverter()
    return


@bot.event
async def on_command_error(error, ctx):
    if isinstance(error, commands.CheckFailure):
        await bot.send_message(ctx.message.channel, "You don't have permissions to run this.")

# ==============================================
# ==========NON COMMAND FUNCTIONS===============
# ==============================================
async def bot_message_cleanup(message):
    TZ_Server_ID = '165310633884123137'
    TZ_FrameChannel_ID = '315052762947649536'
    TestServer_Server_ID = '302481884984639488'
    TestServer_ChannelID = '303175029884059649'
    Delay_Seconds = 10

    if message.channel.is_private or message.channel.id == TZ_FrameChannel_ID:
        # lazy workaround for TZ's frame data channel cuz ppl spam shit in chara channels
        # dont do shit
        return
    if message.server.id == TZ_Server_ID:
        # lazy workaround No.2
        await asyncio.sleep(Delay_Seconds)
        await bot.delete_message(message)
        return

    if message.channel.permissions_for(message.server.me).manage_messages:
        # self delete does not require server permissions,
        # but tying both cleanups to one check for now until I make a controllable toggle.
        await asyncio.sleep(Delay_Seconds)
        await bot.delete_message(message)
        return

async def user_message_cleanup(message):
    Delay_Seconds = 15
    if message.channel.is_private:
        return
    if message.channel.permissions_for(message.server.me).manage_messages:
        await asyncio.sleep(Delay_Seconds)
        await bot.delete_message(message)

async def is_Gagged(user_message):
    message = user_message
    # check if channel is gagged
    if message.content != '.ungagcombot':
        for channelID in combot_gagged_channels:
            if message.channel.id == channelID:
                return True

    return False

async def get_MoveFound_Embed(**move_attribute_dict):
    misc_details_Dict = tekkenFinder.get_Misc_Chara_Details(move_attribute_dict['char_name'])
    embedDict = {**move_attribute_dict, **misc_details_Dict}
    embed_MoveFound = embedCreation.embed_Move_Details(**embedDict)

    return embed_MoveFound

async def get_SimilarMoves_Embed(user_Chara_Name, user_Chara_Move):
    misc_details_Dict = tekkenFinder.get_Misc_Chara_Details(user_Chara_Name)
    similar_moves_list = tekkenFinder.get_Similar_Moves(user_Chara_Name, user_Chara_Move)
    embed_SimilarMoves = embedCreation.embed_Similar_Moves(similar_moves_list, user_Chara_Name, **misc_details_Dict)

    return embed_SimilarMoves

#Starts the bot
bot.run(token)

handlers = log.handlers[:]
for hdlr in handlers:
    hdlr.close()
    log.removeHandler(hdlr)