mirror of
https://github.com/ada-dmitry/PyProg_ada.git
synced 2026-09-24 01:10:40 +00:00
54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
import telebot
|
|
import psycopg2
|
|
import config
|
|
from telebot import types
|
|
import random
|
|
import FuncForBot
|
|
|
|
connection = psycopg2.connect(
|
|
host=config.hostname, dbname=config.databname, user=config.username, password=config.passw)
|
|
cursor = connection.cursor()
|
|
try:
|
|
array = FuncForBot.sel(cursor, 'parser')
|
|
except psycopg2.errors.UndefinedTable:
|
|
connection.rollback()
|
|
import tableCreator
|
|
array = FuncForBot.sel(cursor, 'parser')
|
|
finally:
|
|
FuncForBot.upd(connection, cursor, 'parser')
|
|
cursor.close()
|
|
connection.close()
|
|
|
|
bot = telebot.TeleBot(config.api)
|
|
|
|
|
|
@bot.message_handler(commands=["start"])
|
|
def start(m, res=False):
|
|
markup = types.ReplyKeyboardMarkup(resize_keyboard=True)
|
|
item1 = types.KeyboardButton("Рандомное вино!")
|
|
item2 = types.KeyboardButton("Повтори!")
|
|
markup.add(item1)
|
|
markup.add(item2)
|
|
bot.send_message(
|
|
m.chat.id, '"Рандомное вино!" - для получения карточки товара\n Введите число(1-20) для вывода определенной позиции\n "Повтори!" - для повторения сообщений.', reply_markup=markup)
|
|
|
|
|
|
@bot.message_handler(content_types=["text"])
|
|
def handle_text(message):
|
|
m = message.text.strip()
|
|
if (m == 'Рандомное вино!'):
|
|
i = random.choice(array)
|
|
ans = f'Название: {i[1]}\nЦена: {i[2]}\nЦена без скидки: {i[3]}\nОценка: {i[4]}\n'
|
|
bot.send_message(message.chat.id, ans)
|
|
bot.send_photo(message.chat.id, open(i[5], 'rb'))
|
|
elif (m.isnumeric() == True):
|
|
i = array[int(m)]
|
|
ans = f'Название: {i[1]}\nЦена: {i[2]}\nЦена без скидки: {i[3]}\nОценка: {i[4]}\n'
|
|
bot.send_message(message.chat.id, ans)
|
|
bot.send_photo(message.chat.id, open(i[5], 'rb'))
|
|
elif (m == 'Повтори!'):
|
|
bot.send_message(message.chat.id, 'Ввод: ' + message.text)
|
|
|
|
|
|
bot.polling(none_stop=True, interval=0)
|