Compare commits

..

No commits in common. "83a79b7b24d7f7ea35d6549b0bb5e924171d918a" and "e897d6c162651d28919c76035579e4b005793173" have entirely different histories.

3 changed files with 3 additions and 32 deletions

1
.gitignore vendored
View File

@ -95,4 +95,3 @@ ENV/
.ropeproject .ropeproject
config.py config.py
chanbot.sqlite

32
bot.py
View File

@ -1,41 +1,15 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# coding=utf-8 # coding=utf-8
import logging
import config import config
import sqlite3 import sqlite3
from telegram.ext import Updater, MessageHandler, Filters 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): def main(bot, update, 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_id):
cursor = db.cursor()
query = ("INSERT INTO poll(callback, user_id) values(?,?)")
cursor.execute(query, (callback_id, user_id))
db.commit()
def post(bot, update, db):
chat_id = update.message.chat_id chat_id = update.message.chat_id
if chat_id in config.allowed_users: if chat_id in config.allowed_users:
callback_id = get_callback_id(db) pass
add_vote(db, callback_id, 0)
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: else:
bot.send_message(chat_id=chat_id, text="Not Authorized") bot.send_message(chat_id=chat_id, text="Not Authorized")
@ -46,6 +20,6 @@ dispatcher = updater.dispatcher
db = sqlite3.connect("chanbot.sqlite", check_same_thread=False) db = sqlite3.connect("chanbot.sqlite", check_same_thread=False)
cursor = db.cursor() cursor = db.cursor()
dispatcher.add_handler(MessageHandler(Filters.photo, lambda x, y: post(x, dispatcher.add_handler(MessageHandler(Filters.photo, lambda x, y: main(x,
y, db))) y, db)))
updater.start_polling() updater.start_polling()

View File

@ -1,2 +0,0 @@
CREATE TABLE IF NOT EXISTS poll(callback int NOT NULL, user_id int, UNIQUE(callback, user_id));
INSERT INTO poll values(0, 0);