bot.py

import telebot
from telebot import types
import json
import os
import sys
import sqlite3

## Descripción: Este bot te introducirá en la lista de amantes de Croissants y te notificará cada tanto para tener la oportunidad de recibir uno en la FI por la mañana.

# Base de datos
adictos_a_croissants = sqlite3.connect('/home/paloma/LeCroissantBot/data/subscribersDB.db', check_same_thread=False)
proveedores_de_felicidad = sqlite3.connect('/home/paloma/LeCroissantBot/data/providersDB.db', check_same_thread=False)

# cursorAdictos = adictos_a_croissants.cursor()
# cursorAdictos.execute('''
#     CREATE TABLE users(id TEXT PRIMARY KEY, name TEXT, numCroissants INTEGER)''')
# adictos_a_croissants.commit()


# cursorProveedor = proveedores_de_felicidad.cursor()
# # cursorProveedor.execute('''
# #     CREATE TABLE admins(id TEXT PRIMARY KEY, name TEXT)''')
# cursorProveedor.execute('''INSERT INTO admins(id, name)
#                   VALUES(?,?)''', ('6419832', 'Lord Voldemort'))
# proveedores_de_felicidad.commit()


# Abrir cosas
with open("/home/paloma/LeCroissantBot/bot.token", "r") as token:
  bot = telebot.TeleBot(token.readline().strip()) # Strip elimina mierda que se pueda colar en el token 

 # Lista en texto para admin
global lista_compra
lista_compra = []

# Lista para notificación
global adictos_en_lista_compra
adictos_en_lista_compra = []

# User tracking
global adictos_en_tramite
adictos_en_tramite = {} 

# Suscriptores que aun no han contestado a la pregunta
global adictos_no_contestados
adictos_no_contestados = []

def isAdmin_fromPrivate(message):
    if message.chat.type == 'private':
        cursor = proveedores_de_felicidad.cursor()
        cursor.execute('''SELECT 1 FROM admins WHERE id=?''', (message.from_user.id, ))
        esAdmin = cursor.fetchall()
        if not esAdmin:
            return False
        return True

def notificarAdictos():
  cursorAdictos = adictos_a_croissants.cursor()
  for row in cursorAdictos.execute('SELECT id FROM users'):
    replyToQuestion(row[0])  
  return

def notificarAdictosCompra():
  [avisar_final(adict) for adict in adictos_no_contestados]  
  [mensaje_compra(adict) for adict in adictos_en_lista_compra]  
  return

def replyToQuestion(adict):
  adictos_no_contestados.append(adict)
  markup = types.ReplyKeyboardMarkup()
  markup.one_time_keyboard = True
  markup.row('¡SÍ!')
  markup.row('¡NO, PESADA!')
  adictos_en_tramite[adict] = 1 # El adicto ha progresado al nivel 1
  bot.send_message(adict, "¿Croissant?", reply_markup=markup)

def mensaje_compra(adict):
  bot.send_message(adict, "Los Croissants van de camino ;)")

def avisar_final(adict):
  adictos_en_tramite[adict] = 0
  bot.send_message(adict, "Demasiado tarde, la compra ha acabado.")
  

### cosas de comandos
@bot.message_handler(commands=['start'])
def send_welcome(message):
  bot.send_message(message.chat.id, "¡Hola *" + message.from_user.first_name + "*!\nTe doy la bienvenida al mejor bot de la historia: Le Croissant. Disfruta la experiencia\n/help", parse_mode="Markdown")

  
# Lista de comandos      
@bot.message_handler(commands=['help'])
def send_help(message):
  bot.send_message(message.chat.id, "Hola!\n Los comandos que puedes usar son:\n/help\n/subscribe\n/horario\n/unsubscribe\n/historico")

  
@bot.message_handler(commands=['subscribe'])
def send_subscr(message):
  cursorAdictos = adictos_a_croissants.cursor()
  cursorAdictos.execute('''SELECT name FROM users WHERE id=?''', (message.from_user.id, ))
  oldSub = cursorAdictos.fetchall()

  if not oldSub:
    cursorAdictos.execute('''INSERT INTO users(id, name, numCroissants)
                  VALUES(?,?,?)''', (message.from_user.id, message.chat.first_name, 0))
    adictos_a_croissants.commit()
    bot.send_message(6419832, "Se ha unido " + message.from_user.first_name + " a la lista de adictos!")
    bot.send_message(message.chat.id, "Has sido añadido a la lista de suscriptores a noticias sobre croissants. ¡Enhorabuena!")   
  else:
    bot.send_message(message.chat.id, "Pero si ya estás en la lista, ¡Caesa!")

    
# Ya se verá si se implementa esto algún día, por ahora supongo que nadie querrá salir de la lista nunca.    
@bot.message_handler(commands=['unsubscribe'])
def send_goodbye(message):
  cursorAdictos = adictos_a_croissants.cursor()
  
  cursorAdictos.execute('''SELECT name FROM users WHERE id=?''', (message.from_user.id, ))
  oldSub = cursorAdictos.fetchall()
  
  if not oldSub:
    bot.send_message(message.chat.id, "Pero si aun no estás suscrito, ¡Caesa!\n)")
  else:        
    cursorAdictos.execute('''DELETE FROM users WHERE id=? ''', (message.from_user.id, ))
    bot.send_message(message.chat.id, "¡Los croissants no te echarán de menos!\n")
    bot.send_message(6419832, "¡Se ha ido el traidor " + message.from_user.first_name + " de la lista de adictos! D:")

    adictos_a_croissants.commit()  

# Esto aun no sirve de nada pero me gusta el nombre
@bot.message_handler(commands=['croissantDay'])
def send_msssg(message):
    bot.send_message(message.chat.id, "Día de Croissants")


# Horario de la panadería
@bot.message_handler(commands=['opening'])
def horarios(message):
    bot.send_message(message.chat.id, "Horario de la Panadería:\nLunes - Viernes: 07:30 - 21:30\nSábados: 09:00 - 16:00\nDomingos y festivos: 09:00 - 21:00")


# Cuando un admin va a ir a la Panadería, esto notifica a todos los suscriptores.
@bot.message_handler(commands=['croissant'])
def croissant_notif(message):
    if isAdmin_fromPrivate(message):
      bot.reply_to(message,
                   "Lanzando ofrecimiento de dulce felicidad en forma de Croissant Supremo...")
      notificarAdictos()
    else:
      bot.reply_to(message, "No intentes suplantar al proveedor de Croissants")

# Cuando la compra está hecha, permite al administrador avisar a los adictos que sus croissants están de camino o que es demasiado tarde para pedirlos   
@bot.message_handler(commands=['bought'])
def compra_notif(message):
  if isAdmin_fromPrivate(message):
    notificarAdictosCompra()
    bot.reply_to(message, "Avisando de compra.... \nRecomendado /reset")
  else:
    bot.reply_to(message, "No intentes suplantar al proveedor de Croissants")

# Reinnicia las listas
@bot.message_handler(commands=['reset'])
def compra_notif(message):
  if isAdmin_fromPrivate(message):
    bot.reply_to(message, "Reiniciando listas......")
    sys.exit()
  else:
    bot.reply_to(message, "No intentes suplantar al proveedor de Croissants")

@bot.message_handler(commands=['history'])
def historial(message):
  cursorAdictos = adictos_a_croissants.cursor()
  cont = 1
  texto = []
  texto.append("Ésta es la lista de los mayores adictos a Croissants de todos los tiempos\n")
  for row in cursorAdictos.execute('''SELECT name, numCroissants FROM users ORDER BY numCroissants DESC'''):
    if(cont < 6):
      texto.append("- El número " + str(cont) + " es " + row[0] + ", con " + str(row[1]) + " Croissants")
      cont = cont +1

  texto.append("\n¡Si compras suficientes Croissants podrás tener el honor de poder estar tú mismo en esta lista de afortunados!")
  bot.reply_to(message, '\n'.join(texto))
  
##### RESPUESTA NIVEL 1
@bot.message_handler(func=lambda message: str(message.from_user.id) in adictos_en_tramite.keys() and adictos_en_tramite[str(message.from_user.id)] == 1)
# @bot.message_handler(func=lambda message: adictos_en_tramite[message.from_user.id] == 1)
def respuesta_niv1(message):
  respuesta = message.text
  adictL1 = str(message.from_user.id)
  adictos_no_contestados.remove(adictL1)
  if (respuesta == '¡SÍ!'):
    markup2 = types.ReplyKeyboardMarkup()
    markup2.one_time_keyboard = True
    markup2.row('1')
    markup2.row('2')
    adictos_en_tramite[adictL1] = 2 # El adicto ha progresado al nivel 2
    bot.send_message(adictL1, "¿Cuántos?", reply_markup=markup2)
  else:
    adictos_en_tramite.pop(adictL1)
    bot.send_message(adictL1, ":(")
  
##### RESPUESTA NIVEL 2
@bot.message_handler(func=lambda message: str(message.from_user.id) in adictos_en_tramite.keys() and adictos_en_tramite[str(message.from_user.id)] == 2)
# @bot.message_handler(func=lambda message: adictos_en_tramite[message.from_user.id] == 1)
def respuesta_niv2(message):
  respuesta = message.text
  adictL2 = str(message.from_user.id)
  adictos_en_lista_compra.append(adictL2)
  cursorAdictos = adictos_a_croissants.cursor()
  consumidos = cursorAdictos.execute('''SELECT numCroissants FROM users WHERE id=?''', (message.from_user.id, ))
  consum = consumidos.fetchone()

  if (respuesta == '1'):
    lista_compra.append(str(message.from_user.first_name) + " 1")
    numC = consum[0] + 1

  elif (respuesta == '2'):
    lista_compra.append(str(message.from_user.first_name) + " 2")
    numC = consum[0] + 2

  bot.send_message(6419832, "Lista de la compra:\n" + '\n'.join(lista_compra))
  bot.send_message(adictL2, "¡Apuntado!")
  cursorAdictos.execute('''UPDATE users set numCroissants=? WHERE id=?''', (numC, message.from_user.id))
  adictos_a_croissants.commit()
  adictos_en_tramite.pop(adictL2)

# db.close() ????

bot.skip_pending = True
print("Running....")
bot.polling()