gistfile1.txt

"""
References:
    https://core.telegram.org/bots/api
    https://python-telegram-bot.readthedocs.io/en/latest
"""
import datetime
import time
#import telegram 
from threading import Thread, Timer
from settings import TOKEN, msg_flood, msg_interval, con, cursor
from telegram.ext import Updater, MessageHandler, Filters, CommandHandler
import random

flood_counter = []
data = []

def on_new_animation(bot, update):
    print("[!][on_new_animation]")
    try:
        item = "{}:{}:{}".format(update.message.chat_id, update.message.from_user.id, update.message.message_id)
        print("[!][New Animation] {}".format(item))
    
        data.append(item)
        flood_counter.append(item)
    
        check_flood(bot, update)
        Timer(msg_interval, timeout, [bot, update]).start()
    except Exception as e:
        print(str(e))

def deleteMsgs(bot, update, msgIDs):
    print("[!][deleteMsgs]")
    try:
        chat_id = str(update.message.chat_id)
        user_id = str(update.message.from_user.id)
        for msg in msgIDs[msg_flood:]:
            Thread(target = deleteMsg, args = (bot, update, int(msg), ), ).start()
            item = "{}:{}:{}".format(update.message.chat_id, update.message.from_user.id, msg)
            data.remove(item)
        xinga(bot, update)
    except Exception as e:
        print(str(e))

def xinga(bot, update):
    print("[!][xinga]")
    try:
        query = "SELECT text FROM xingamento ORDER BY RAND() LIMIT 1"
        cursor.execute(query)
        row = cursor.fetchone()
        quote = "__{}__".format(row[0])

        msg = "[!] Cala a boca {}, {}!".format(update.message.from_user.first_name, quote)
        bot.send_message(update.message.chat_id, msg, parse_mode = 'Markdown')
    except Exception as e:
        print(str(e))

def deleteMsg(bot, update, msgId):
    print("[!][deleteMsg]")
    try:
        chat_id = update.message.chat_id
        print("[!][deleteMsg] Deleting  = > {}".format(msgId))
    
        bot.delete_message(chat_id = chat_id, message_id = msgId)
        item = "{}:{}:{}".format(update.message.chat_id, update.message.from_user.id, msgId)
        data.remove(item)
    except Exception as e:
        print(str(e))

def check_flood(bot, update):
    print("[!][check_flood]")
    try:
        chat_id = str(update.message.chat_id)
        user_id = str(update.message.from_user.id)
        counter = 0
        #msgIds = []

        print("[!] Checking flood")

        for i, item in enumerate(flood_counter):
            if (chat_id + ":" + user_id) in item:
                #msgIds.append((flood_counter[i].split(":")[2]))
                counter + =  1
        if counter > msg_flood:
            print("[!] Flood in action: {} ({})".format(update.message.user_id, counter))
            #Thread(target = deleteMsg, args = (bot, update, update.message.message_id), ).start()
            deleteMsg(bot, update, update.message.message_id)
            #deleteMsgs(bot, update, msgIds)
    except Exception as e:
        print(str(e))

def timeout(bot, update):
    print("[!][timeout]")
    try:
        print("[!] timeout - flood_counter: {}".format(flood_counter))
        print("[!] timeout - data: {}".format(data))
        item = "{}:{}:{}".format(update.message.chat_id, update.message.from_user.id, update.message.message_id)
        flood_counter.remove(item)
        #data.remove(item)
    except Exception as e:
        print(str(e))

def logging(bot, update):
    print("[!][logging]")
    print("##############################")
    print(str(update))
    print(" ")
    try:
        photo = ""
        query  = ("INSERT INTO log (message_id, date, chat_id, text, photo, from_id, first_name, last_name, username) VALUES ('{}', '{}', '{}', '{}', '{}', '{}', '{}', '{}', '{}')").format(update.message.message_id, str(update.message.date), update.message.chat.id, str(update.message.text), photo, update.message.from_user.id, update.message.from_user.first_name, update.message.from_user.last_name, update.message.from_user.username )
#    else:
#        photo = update.message.photo.file_id
#    print(photo)
#        print(query)
        cursor.execute(query)
        con.commit()
    except Exception as e:
        print(str(e))

def put_quote(bot, update):
    print("[!][put_quote]")
    try:
        message_id = update.message.reply_to_message.message_id
        print(message_id)
        query = ("INSERT INTO quote (message_id, chat_id) VALUES ({}, {})").format(message_id, update.message.chat_id)
        cursor.execute(query)
        con.commit()
        print("OK")
    except Exception as e:
        print(str(e))

def get_quote(bot, update):
    print("[!][get_quote]")
    try:
        if update.message.text ==  "/lauters":
            query = "SELECT text FROM lauters ORDER BY RAND() LIMIT 1"
        else:
            quote_id = (update.message.text).split(" ")
            if len(quote_id) > 1 :
                query = "SELECT message_ID FROM quote WHERE chat_ID = {} AND id = {} ORDER BY RAND() LIMIT 1".format(update.message.chat_id, int(quote_id[1]))
            else:
                query = "SELECT message_ID FROM quote WHERE chat_ID = {} ORDER BY RAND() LIMIT 1".format(update.message.chat_id)
    
            cursor.execute(query)
            message_id = cursor.fetchone()[0]
            query = "SELECT text FROM log WHERE message_id = {}".format(message_id)
#        print(query)
        cursor.execute(query)
        row = cursor.fetchone()
        quote = "__{}__".format(row[0])
        print(quote) 
    except Exception as e:
        print(str(e))
    print(row)
    bot.send_message(update.message.chat_id, quote, parse_mode = 'Markdown')

try:
    updater = Updater(token = TOKEN)
    dp = updater.dispatcher
    
    log_handler = MessageHandler(Filters.all, logging)
    flood_handler = MessageHandler(Filters.animation | Filters.photo | Filters.video, on_new_animation)
    save_handler = CommandHandler("save", put_quote)
    quote_handler = CommandHandler("quote", get_quote)
    lauters_handler = CommandHandler("lauters", get_quote)
    
    dp.add_handler(save_handler)
    dp.add_handler(quote_handler)
    dp.add_handler(lauters_handler)
    dp.add_handler(flood_handler)
    dp.add_handler(log_handler)
    
    updater.start_polling()
    updater.idle()
except Exception as e:
    print(str(e))