Compare commits
No commits in common. "83a79b7b24d7f7ea35d6549b0bb5e924171d918a" and "e897d6c162651d28919c76035579e4b005793173" have entirely different histories.
83a79b7b24
...
e897d6c162
|
@ -95,4 +95,3 @@ ENV/
|
||||||
.ropeproject
|
.ropeproject
|
||||||
|
|
||||||
config.py
|
config.py
|
||||||
chanbot.sqlite
|
|
||||||
|
|
32
bot.py
32
bot.py
|
@ -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()
|
||||||
|
|
|
@ -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);
|
|
Loading…
Reference in New Issue