134 lines
4.4 KiB
Python
134 lines
4.4 KiB
Python
#!/usr/bin/env python3
|
|
# coding=utf-8
|
|
|
|
import logging
|
|
import config
|
|
import sqlite3
|
|
from telegram.ext import Updater, MessageHandler, CallbackQueryHandler, 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 message_ids')
|
|
cursor.execute(query)
|
|
return int(cursor.fetchone()[0]) + 1
|
|
|
|
|
|
def add_message_id(db, message_id, callback):
|
|
cursor = db.cursor()
|
|
query = ("INSERT INTO message_ids(callback, message_id) values(?,?)")
|
|
cursor.execute(query, (callback, message_id))
|
|
db.commit()
|
|
|
|
|
|
def add_vote(db, callback_id, user=None):
|
|
user_id = user.id
|
|
name = str(user.first_name)
|
|
if user.last_name:
|
|
name += " " + str(user.last_name)
|
|
|
|
cursor = db.cursor()
|
|
query = ("INSERT INTO poll(callback, user_id, name) values(?,?,?)")
|
|
cursor.execute(query, (callback_id, user_id, name))
|
|
db.commit()
|
|
|
|
|
|
def delete_vote(db, callback_id, user):
|
|
user_id = user.id
|
|
query = "DELETE FROM poll WHERE callback=? AND user_id=?"
|
|
cursor.execute(query, (callback_id, user_id))
|
|
db.commit()
|
|
|
|
|
|
def check_vote(db, callback_id, user):
|
|
user_id = user.id
|
|
cursor = db.cursor()
|
|
query = ("SELECT * FROM poll WHERE callback=? AND user_id=?")
|
|
cursor.execute(query, (callback_id, user_id))
|
|
if cursor.fetchone():
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def update_message(bot, db, callback_id):
|
|
cursor = db.cursor()
|
|
query = "SELECT message_id FROM message_ids WHERE callback = ?"
|
|
cursor.execute(query, (callback_id,))
|
|
message_id = cursor.fetchone()[0]
|
|
query = "SELECT count(user_id) FROM poll WHERE callback = ?"
|
|
cursor.execute(query, (callback_id,))
|
|
count = cursor.fetchone()[0]
|
|
keyboard = InlineKeyboardMarkup([[InlineKeyboardButton(
|
|
text=f"❤️ {count}",
|
|
callback_data=callback_id)]])
|
|
bot.edit_message_reply_markup(chat_id=config.channel_id,
|
|
message_id=message_id,
|
|
reply_markup=keyboard)
|
|
|
|
|
|
def send_stats(bot, update, db):
|
|
cursor = db.cursor()
|
|
query = ("SELECT user_id, name FROM poll NATURAL JOIN message_ids WHERE "
|
|
"message_id = ?")
|
|
cursor.execute(query, (update.message.forward_from_message_id,))
|
|
voters = []
|
|
for row in cursor:
|
|
voters.append(row)
|
|
if len(voters) == 0:
|
|
bot.send_message(chat_id=update.message.chat_id, text="No Votes")
|
|
else:
|
|
text = "Voters\n"
|
|
for i in voters:
|
|
text += f"<a href='tg://user?id={i[0]}'>{i[1]}</a>\n"
|
|
bot.send_message(chat_id=update.message.chat_id, text=text,
|
|
parse_mode="HTML")
|
|
|
|
|
|
def post(bot, update, db):
|
|
chat_id = update.message.chat_id
|
|
user_id = update.message.from_user.id
|
|
if (update.message.forward_from_chat and
|
|
update.message.forward_from_chat.id == config.channel_id and
|
|
user_id in config.allowed_users):
|
|
send_stats(bot, update, db)
|
|
return
|
|
if chat_id in config.allowed_users:
|
|
callback_id = get_callback_id(db)
|
|
photo = update.message.photo[0]
|
|
file_id = photo.file_id
|
|
keyboard = InlineKeyboardMarkup([[InlineKeyboardButton(text="❤️",
|
|
callback_data=callback_id)]])
|
|
reply = bot.send_photo(chat_id=config.channel_id, photo=file_id,
|
|
reply_markup=keyboard)
|
|
add_message_id(db, reply.message_id, callback_id)
|
|
else:
|
|
bot.send_message(chat_id=chat_id, text="Not Authorized")
|
|
|
|
|
|
def click(bot, update, db):
|
|
user = update.callback_query.from_user
|
|
callback_id = update.callback_query.data
|
|
if not check_vote(db, callback_id, user):
|
|
add_vote(db, callback_id, user)
|
|
else:
|
|
delete_vote(db, callback_id, user)
|
|
update_message(bot, db, callback_id)
|
|
bot.answer_callback_query(update.callback_query.id)
|
|
|
|
|
|
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)))
|
|
dispatcher.add_handler(CallbackQueryHandler(lambda x, y: click(x, y, db)))
|
|
updater.start_polling()
|