Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- import telebot
- # import config
- import psycopg2
- from telebot import types
- # from config
- token='176385051:AAF3asP2GWtrY871vUq9FndM8eKrLrxQE38'
- connect = psycopg2.connect(database='d1eam1hffgoggg',
- user='ravnyccawkzsbx',
- host='ec2-54-235-246-67.compute-1.amazonaws.com',
- password='0lo2zjah68Y2pHef7jRrh9KjqO')
- cursor = connect.cursor()
- ticket = 'empty ticket'
- bot = telebot.TeleBot(config.token)
- @bot.message_handler(commands=['start'])
- def show_start_message(message):
- print(message.chat.id)
- cursor.execute("insert into users (id_user, days) values (" + str(message.chat.id)+",5)")
- connect.commit()
- hello_message = "Добро пожаловать! \n " \
- "Базовые команды \n" \
- "/list_all_tickets \n" \
- "/help"
- bot.send_message(message.chat.id, hello_message)
- @bot.message_handler(commands=['help'])
- def show_start_message(message):
- help_message = "/list_all_tickets - показать список существующих квитанций \n" \
- "/list_tickets - просмотр активных квитанций \n" \
- "/days - установка числа - за сколько дней необходимо начать присылать напоминание \n" \
- "/clear - сброс параметров\n" \
- "/help - список всех команд "
- bot.send_message(message.chat.id, help_message)
- @bot.message_handler(commands=['days'])
- def show_start_message(message):
- cursor.execute("select days from users where id_user="+ str(message.chat.id))
- for row in cursor:
- days = row[0]
- days_message = "в течении "+str(days)+" дней будет приходить напоминание об оплате квитанции. "
- markup = types.ReplyKeyboardMarkup(row_width=1)
- markup.row('да')
- markup.row('нет')
- msg = bot.send_message(message.chat.id, days_message+"Хотите изменить это число?", reply_markup=markup)
- bot.register_next_step_handler(msg, response_change_days)
- def response_change_days(message):
- chat_id = message.chat.id
- if (message.text == 'да'):
- print("да")
- msg = bot.send_message(message.chat.id, "Введите число")
- bot.register_next_step_handler(msg, change_days)
- else:
- print("нет")
- def change_days(message):
- cursor.execute("Update users set days="+str(message.text)+" where id_user="+str(message.chat.id))
- connect.commit()
- bot.send_message(message.chat.id, "Число изменено")
- @bot.message_handler(commands=['list_all_tickets'])
- def show_start_message(message):
- cursor.execute("SELECT ticket_name, active_row FROM all_tickets;")
- list_all_tickets_message = ""
- for row in cursor:
- list_all_tickets_message = list_all_tickets_message + row[0].strip() + " " + row[1] + "\n"
- bot.send_message(message.chat.id, list_all_tickets_message)
- @bot.message_handler(commands=['list_tickets'])
- def show_start_message(message):
- cursor.execute("select all_tickets.ticket_name, " \
- "all_tickets.active_row " \
- "from all_tickets " \
- "left outer join user_tickets " \
- "on user_tickets.id_ticket=all_tickets.id_ticket " \
- "where user_tickets.id_user="+str(message.chat.id))
- list_tickets_message = ""
- for row in cursor:
- list_tickets_message = list_tickets_message + row[0].strip() + " " + row[1] + "\n"
- bot.send_message(message.chat.id, "Список активных квитанций: \n"+ list_tickets_message)
- @bot.message_handler(commands=['t_1', 't_2', 't_3', 't_4', 't_5', 't_6', 't_7', 't_8', 't_9', 't_10', 't_11'])
- def add_ticket(message):
- print(str(message.chat.id))
- cursor.execute("select user_tickets.id_user, all_tickets.active_row " \
- "from user_tickets " \
- "left outer join all_tickets " \
- "on all_tickets.id_ticket=user_tickets.id_ticket " \
- "where user_tickets.id_user="+str(message.chat.id))
- for row in cursor:
- print(row[0], str(message.chat.id), row[1], message.text)
- if ((str(row[0]) == str(message.chat.id)) and (str(row[1]).strip() == str(message.text))):
- print("я зашел в if")
- bot.send_message(message.chat.id, "квитанция " + message.text + " у вас есть")
- markup = types.ReplyKeyboardMarkup(row_width=1)
- markup.row('да')
- markup.row('нет')
- msg = bot.send_message(message.chat.id, "Удалить квитанцию?", reply_markup=markup)
- bot.register_next_step_handler(msg, delete_active_ticket)
- return
- markup = types.ReplyKeyboardMarkup()
- markup.row('да')
- markup.row('нет')
- msg = bot.send_message(message.chat.id, "Добавить квитанцию?", reply_markup=markup)
- bot.register_next_step_handler(msg, add_active_ticket)
- ticket = message.text # должен передавать в функцию delete_active_ticket значение /t_1
- def delete_active_ticket(message): pass
- # if (message.text == 'да'):
- # print("да")
- # cursor.execute("delete from user_tickets where id_user="+str(message.chat.id)+" and id_ticket=")
- # connect.commit()
- # else:
- # print("нет")
- def add_active_ticket(message):
- # if (message.text == 'да'):
- # print("да")
- # cursor.execute("select id_ticket from all_tickets where active_row =" +row)
- # for new_row in cursor:
- # id_ticket = new_row[0]
- # cursor.execute("insert into all_tickets "
- # "(id_user, id_ticket) "
- # "values ("+ str(message.chat.id)+","
- # + str(id_ticket).split() +")")
- # else:
- # print("нет")
- # ticket = None
- return True
- if __name__ == '__main__':
- bot.polling(none_stop=True)
- connect.close()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement