Channel-Poll-Bot/bot.py

#!/usr/bin/env python3
# coding=utf-8

import logging
import config
import sqlite3
from telegram.ext import Updater, MessageHandler, Filters
from telegram import InlineKeyboardButton, InlineKeyboardMarkup

logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - \
                    %(message)s', level=logging.INFO)


def get_callback_id(db):
    cursor = db.cursor()
    query = ('SELECT max(callback) FROM poll')
    cursor.execute(query)
    return int(cursor.fetchone()[0]) + 1


def add_vote(db, callback_id, user=None):
    if user is not None:
        user_id = user.id
        name = str(user.first_name)
        if user.last_name:
            name += " " + str(user.last_name)
    else:
        user_id = 0
        name = ""

    cursor = db.cursor()
    query = ("INSERT INTO poll(callback, user_id, name) values(?,?,?)")
    cursor.execute(query, (callback_id, user_id, name))
    db.commit()


def post(bot, update, db):
    chat_id = update.message.chat_id
    if chat_id in config.allowed_users:
        callback_id = get_callback_id(db)
        add_vote(db, callback_id)
        photo = update.message.photo[0]
        file_id = photo.file_id
        keyboard = InlineKeyboardMarkup([[InlineKeyboardButton(text="❤️",
                                        callback_data=callback_id)]])
        bot.send_photo(chat_id=config.channel_id, photo=file_id,
                       reply_markup=keyboard)
    else:
        bot.send_message(chat_id=chat_id, text="Not Authorized")


updater = Updater(token=config.api_key)
dispatcher = updater.dispatcher

db = sqlite3.connect("chanbot.sqlite", check_same_thread=False)
cursor = db.cursor()

dispatcher.add_handler(MessageHandler(Filters.photo, lambda x, y: post(x,
                                      y, db)))
updater.start_polling()