discordBot.py

from os import path
import json
import sqlite3

import asyncio

import discord
from discord.ext.commands import Bot

class ManejoArchivos():
    rutaRaiz = path.dirname(__file__)
    rutaPizzas = path.join(rutaRaiz, 'pizzas.db')
    
    def leerToken(self):
        with open(path.join(self.rutaRaiz, 'config.json')) as archivo:
            config = json.load(archivo)
            archivo.close()
        
        token = config['token']
        return token
    
    def leerOpciones(self):
        with open(path.join(self.rutaRaiz, 'config.json')) as archivo:
            config = json.load(archivo)
            archivo.close()
        
        opciones = {
            'si': config['opciones']['si'],
            'no': config['opciones']['no']
        }
        return opciones
    
    def leerPizzas(self, servidor):
        conexion = sqlite3.connect(self.rutaPizzas)
        cursor = conexion.cursor()
        
        dictPizzas = {}
        
        try:#CREATE TABLE IF NOT EXISTS PIZZAS
            cursor.execute('''
                CREATE TABLE s{0}(
                    ID     INT               PRIMARY KEY NOT NULL,
                    VALOR  TINYINT UNSIGNED              NOT NULL
                )
            '''.format(servidor))
            conexion.commit()
            print('[DISC] Tabla {0} creada exitosamente'.format(servidor))
        except:
            print('[DISC] Leyendo la tabla {0} en {1}'.format(servidor, self.rutaPizzas))
            cursor.execute('''
                SELECT ID, VALOR FROM s{0}
            '''.format(servidor))
            for pareja in cursor:
                dictPizzas[pareja[0]] = pareja[1]
        
        conexion.close()
        return dictPizzas
    
    def escribirPizzas(self, servidor, usuario, valor):
        conexion = sqlite3.connect(self.rutaPizzas)
        cursor = conexion.cursor()

        celda = cursor.execute('''
            SELECT 1 FROM s{1} WHERE ID={0}
        '''.format(usuario, servidor)).fetchone()
        if celda == None:
            print('[DISC] Añadiendo {0} a la tabla {1} en {2}'.format(usuario, servidor, self.rutaPizzas))
            cursor.execute('''
                INSERT INTO s{2}(ID, VALOR)
                VALUES({0}, {1})    
            '''.format(usuario, valor, servidor))
        else:
            if valor > 0:
                print('[DISC] Actualizando {0} en la tabla {1} en {2}'.format(usuario, servidor, self.rutaPizzas))   
                cursor.execute('''
                    UPDATE s{2} SET VALOR={1} WHERE ID={0}
                '''.format(usuario, valor, servidor))
            else:
                print('[DISC] Removiendo {0} de la tabla {1} en {2}'.format(usuario, servidor, self.rutaPizzas))
                cursor.execute('''
                    DELETE FROM s{1} WHERE ID={0}
                '''.format(usuario, servidor))

        conexion.commit()
        conexion.close()

cliente = Bot(command_prefix='$')

@cliente.event
async def on_ready():
    print('[CLNT] Conectado exitosamente')

'''
@cliente.event
async def on_reaction_add(reaccion, usuario):
    if reaccion.message.author.id != cliente.user.id:
        return
    
    print('[LOG ] {0}'.format(reaccion.emoji.encode('ascii', 'namereplace')))
'''

@cliente.command()
async def pizzas(contexto):
    dictPizzas = sorted(list(archivos.leerPizzas(contexto.guild.id).items()), key=lambda x: x[1], reverse=True)

    if len(dictPizzas) > 0:
        mensaje = ''
        for tupla in dictPizzas:
            usuario = cliente.get_user(tupla[0])
            valor = tupla[1]
            mensaje += '{0}: {1}🍕\n'.format(usuario.mention, valor)
        totalPizzas = sum(map(lambda x: x[1], dictPizzas))
        mensaje += '{0}🧑, {1}🍕, {2}🍕per🧑'.format(len(dictPizzas), totalPizzas, round(totalPizzas/len(dictPizzas), 2))
    else:
        mensaje = 'Nadie debe pizzas'
    await contexto.send(mensaje)

@cliente.command()
async def añadirpizza(contexto):
    dictPizzas = archivos.leerPizzas(contexto.guild.id)
    
    for destino in contexto.message.mentions:
        idUsuario = destino.id
        if idUsuario == cliente.user.id:
            await contexto.send('🤨')
            return
        
        if idUsuario not in dictPizzas:
            cantidad = 1
        else:
            cantidad = dictPizzas[idUsuario] + 1
        miMensaje = await contexto.send('{0}, ¿Confirmas?'.format(destino.mention))
        await miMensaje.add_reaction(opciones['si']),
        await miMensaje.add_reaction(opciones['no'])

        reaccion = None
        try:
            reaccion, emisor = await cliente.wait_for('reaction_add', check=lambda reaction, user: user == destino, timeout=tiempoOpciones)
        except asyncio.TimeoutError:
            await miMensaje.delete()
            await contexto.send(content='Lo siento {0}, parece que esta vez no habrá pizza 😥'.format(contexto.author.mention))

        if reaccion != None:
            if reaccion.message.id == miMensaje.id and reaccion.emoji == opciones['si']:
                archivos.escribirPizzas(contexto.guild.id, idUsuario, cantidad)
                await contexto.send('{0} ahora debe {1}🍕'.format(destino.mention, cantidad))
            else:
                await miMensaje.delete()
                await contexto.send(content='Lo siento {0}, parece que esta vez no habrá pizza 😥'.format(contexto.author.mention))

@cliente.command()
async def removerpizza(contexto):
    dictPizzas = archivos.leerPizzas(contexto.guild.id)

    for persona in contexto.message.mentions:
        mensaje = '{0} no debe 🍕!'
        cantidad = 0
        
        idUsuario = persona.id
        if idUsuario == cliente.user.id:
            mensaje = '🤨'

        if idUsuario not in dictPizzas:
            pass
        else:
            if dictPizzas[idUsuario] == 0:
                pass
            else:
                mensaje = '{0} ahora debe {1}🍕'
                cantidad = dictPizzas[idUsuario] - 1
            archivos.escribirPizzas(contexto.guild.id, idUsuario, cantidad)
        
            await contexto.send(mensaje.format(persona.mention, cantidad))

@cliente.command()
async def buenardo(contexto):
    await contexto.send('https://www.youtube.com/watch?v=3OGYzegOhF4')

@cliente.command()
async def malardo(contexto):
    await contexto.send('https://cdn.discordapp.com/attachments/436279597139755012/714958150939312168/malardo.png')

@cliente.command()
async def raw(contexto):
    mensajeOriginal = contexto.message
    mensaje = ' '.join(mensajeOriginal.content.split(' ')[1:]).replace('*', '\*').replace('~', '\~')
    
    await contexto.send(mensaje)

archivos = ManejoArchivos()
#⏯⏭
opciones = archivos.leerOpciones()
tiempoOpciones = 300

token = archivos.leerToken()
cliente.run(token)