bot.py

from calendar import month
import re
import telebot
from telebot import types
import time
import datetime
import re
from db import add_sub_data, list_subs, user_exist


TOKEN = "5329743826:AAHOy3_DDCekqlN2i7sGPpVIVKT_OYQRp18"

bot = telebot.TeleBot(token = TOKEN)
current_time = datetime.datetime.now()
print("😄 KeepSub - напоминания о подписках")
print(f"{current_time} | Бот запущен")

ok_keyboard = types.ReplyKeyboardMarkup(
            resize_keyboard=True,
            one_time_keyboard=True,
            row_width=1
        )
ok_keyboard.add("Главное меню")

@bot.message_handler(commands = ["start"])
def start_bot(message):
    first_name = message.from_user.first_name
    username = f"@{message.from_user.username}"
    if message.from_user.username is None:
        username = f"@{message.from_user.id}"
    print(f"{username} запустил бота...")
    if user_exist(message.chat.id) == False:
        start_message = f"""
Привет, {first_name}!

Да, я в курсе, что ты часто забываешь отключать ненужные подписки на какме-либо сервисы, и в самый неподходящий момент с твоей карты списывается кругленькая сумма. Я могу помочь тебе в этом деле.

Все, что от тебя потребуется, так это написать мне название сервсиса, дату оплаты, стоимость и период подписки. За два дня до окончания подписки я напомню тебе о ней, а дальше сам решай, отключать ее или пользоваться ей дальше.

Давай добавим первый сервис?
"""
        start_keyboard = types.ReplyKeyboardMarkup(
            resize_keyboard=True,
            one_time_keyboard=True,
            row_width=1
        )
        start_keyboard.add("Добавить новую подписку")
        bot.send_message(message.chat.id, start_message, reply_markup=start_keyboard)
        bot.register_next_step_handler_by_chat_id(message.chat.id, add_sub_name)
    else:
        old_user_greet = """
Привет! Пояивлась новая подписка? 😅"""
        bot.send_message(message.chat.id, old_user_greet)
        main_menu(message)

def add_sub_name(message):
    quest_about_name = """
Пришли мне название сервиса, у которого ты приобрел подписку?

Например, VK Музыка, Яндекс.Плюс и т. п."""
    bot.send_message(message.chat.id, quest_about_name)
    bot.register_next_step_handler_by_chat_id(message.chat.id, add_sub_date)


def add_sub_date(message):
    global name_sub_service
    name_sub_service = message.text
    quest_about_date = f"""
Отлично, я запомнил — {name_sub_service}. Теперь дальше.

Укажи дату оплаты подписки в формате [дд.мм.гггг]

Например, 24.02.2022
"""
    bot.send_message(message.chat.id, quest_about_date)
    bot.register_next_step_handler_by_chat_id(message.chat.id, add_sub_summ)


def add_sub_summ(message):
    global date_sub
    date_sub = message.text
    quest_about_summ = f"""
Понял, — {date_sub}. Переходим к следующему шагу.

Укажи стоимость подписки в формате [стоимость]
Например, 299"""
    bot.send_message(message.chat.id, quest_about_summ)
    bot.register_next_step_handler_by_chat_id(message.chat.id, add_sub_period)


def add_sub_period(message):
    global summ_sub
    summ_sub = message.text
    quest_about_period = f"""
Хорошо, стоимость записал — {summ_sub} рублей (-я). Остался последний вопрос.

На сколько месяцев ты приобрел подписку?

Например, 1
"""
    bot.send_message(message.chat.id, quest_about_period)
    bot.register_next_step_handler_by_chat_id(message.chat.id, add_sub_total)


def add_sub_total(message):
    global validation_keyboard
    validation_keyboard = types.ReplyKeyboardMarkup(
        resize_keyboard=True,
        one_time_keyboard=True,
        row_width=1
    )
    validation_keyboard.add("Да, все верно", "Нет, хочу внести исправления")
    global period_sub
    period_sub = message.text
    message_total = f"""
Супер, срок запомнил — {period_sub} месяц(-а, -ев). Подведем итог.

Название сервиса — {name_sub_service};
Дата приобретения подписки — {date_sub};
Стоимость подписки — {summ_sub} рублей (-я);
Срок подпсики — {period_sub} месяц(-а, -ев);

Все верно?"""
    bot.send_message(message.chat.id, message_total, reply_markup=validation_keyboard)
    bot.register_next_step_handler_by_chat_id(message.chat.id, check_data)
    return name_sub_service, date_sub, summ_sub, period_sub


def check_data(message):
    if message.text.lower() == "да, все верно":
        message_ok = f"""
Вот и славно. Я все запомнил.

👌 Напомню тебе о подписке на {name_sub_service} за два дня до ее окончания"""
        bot.send_message(message.chat.id, message_ok, reply_markup=ok_keyboard)
        bot.register_next_step_handler_by_chat_id(message.chat.id, main_menu)
        x = re.split("\.", date_sub)
        d, m, y = x[0], x[1], x[2]
        y, m, d = int(d), int(m), int(y)
        a = datetime.datetime(d, m, y)
        b = datetime.timedelta(days = 30*int(period_sub))
        c = datetime.timedelta(days = (30*int(period_sub)-2))
        finish_sub = str(a+b)
        notify_date_sub = str(a+c)
        add_sub_data(message.chat.id, name_sub_service=name_sub_service, date_sub=date_sub, summ_sub=summ_sub, period_sub=period_sub, finish_sub=finish_sub, notify_date_sub=notify_date_sub)
    elif message.text.lower() == "нет, хочу внести исправления":
        edit_keyboard = types.ReplyKeyboardMarkup(
            resize_keyboard=True,
            one_time_keyboard=True,
            row_width=1
        )
        edit_keyboard.add("Изменить название сервиса", "Изменить дату приобретения", "Изменить стоимость подписки", "Изменить срок подписки")
        message_edit = """
Так, сейчас все поправим. В чем ошибка?"""
        bot.send_message(message.chat.id, message_edit, reply_markup=edit_keyboard)
        bot.register_next_step_handler_by_chat_id(message.chat.id, enter_edits)


def main_menu(message):
    global main_menu_keyboard
    main_menu_keyboard = types.ReplyKeyboardMarkup(
        resize_keyboard=True,
        one_time_keyboard=True,
        row_width=1
    )
    main_menu_keyboard.add("Мои подписки", "Добавить новую подписку")
    global main_menu_message
    main_menu_message = """
Сейчас мы в главном меню.

Тут ты можешь посмотреть все свои добавленные подписки или добавить новую"""
    bot.send_message(message.chat.id, main_menu_message, reply_markup=main_menu_keyboard)
    bot.register_next_step_handler_by_chat_id(message.chat.id, what_next)


def enter_edits(message):
    const_text = "Просто напиши мне этот параметр снова (только на этот раз давай бех ошибок 😄)"
    if message.text.lower() == "изменить название сервиса":
        edit_name_service_sub_message = f"""
Окей, изменим название сервиса. {const_text}"""
        bot.send_message(message.chat.id, edit_name_service_sub_message)
        bot.register_next_step_handler_by_chat_id(message.chat.id, edit_name_service)
    elif message.text.lower() == "изменить дату приобретения":
        edit_date_sub_message = f"""
Окей, изменим дату приобретения. {const_text}"""
        bot.send_message(message.chat.id, edit_date_sub_message)
        bot.register_next_step_handler_by_chat_id(message.chat.id, edit_date_sub)
    elif message.text.lower() == "изменить стоимость подписки":
        edit_summ_sub_message = f"""
Окей, изменим стоимость подписки. {const_text}"""
        bot.send_message(message.chat.id, edit_summ_sub_message)
        bot.register_next_step_handler_by_chat_id(message.chat.id, edit_summ_sub)
    elif message.text.lower() == "изменить срок подписки":
        edit_period_sub_message = f"""
Окей, изменим срок подписки. {const_text}"""
        bot.send_message(message.chat.id, edit_period_sub_message)
        bot.register_next_step_handler_by_chat_id(message.chat.id, edit_period_sub)


def edit_name_service(message):
    name_sub_service = message.text
    changed_sub_info = f"""
Вот так лучше. Получилось так:

Название сервиса — {name_sub_service};
Дата приобретения подписки — {date_sub};
Стоимость подписки — {summ_sub} рублей (-я);
Срок подпсики — {period_sub} месяц(-а, -ев);

Все верно?"""
    bot.send_message(message.chat.id, changed_sub_info, reply_markup=validation_keyboard)
    bot.register_next_step_handler_by_chat_id(message.chat.id, check_data)


def edit_date_sub(message):
    date_sub = message.text
    changed_sub_info = f"""
Вот так лучше. Получилось так:

Название сервиса — {name_sub_service};
Дата приобретения подписки — {date_sub};
Стоимость подписки — {summ_sub} рублей (-я);
Срок подпсики — {period_sub} месяц(-а, -ев);

Все верно?"""
    bot.send_message(message.chat.id, changed_sub_info, reply_markup=validation_keyboard)
    bot.register_next_step_handler_by_chat_id(message.chat.id, check_data)


def edit_summ_sub(message):
    summ_sub = message.text
    changed_sub_info = f"""
Вот так лучше. Получилось так:

Название сервиса — {name_sub_service};
Дата приобретения подписки — {date_sub};
Стоимость подписки — {summ_sub} рублей (-я) (-я);
Срок подпсики — {period_sub} месяц(-а, -ев);

Все верно?"""
    bot.send_message(message.chat.id, changed_sub_info, reply_markup=validation_keyboard)
    bot.register_next_step_handler_by_chat_id(message.chat.id, check_data)


def edit_period_sub(message):
    period_sub = message.text
    changed_sub_info = f"""
Вот так лучше. Получилось так:

Название сервиса — {name_sub_service};
Дата приобретения подписки — {date_sub};
Стоимость подписки — {summ_sub} рублей (-я);
Срок подпсики — {period_sub} месяц(-а, -ев);

Все верно?"""
    bot.send_message(message.chat.id, changed_sub_info, reply_markup=validation_keyboard)
    bot.register_next_step_handler_by_chat_id(message.chat.id, check_data)


def what_next(message):
    if message.text.lower() == "мои подписки":
        your_subs_keyboard = types.ReplyKeyboardMarkup(
            resize_keyboard=True,
            one_time_keyboard=True,
            row_width=1
        )
        your_subs_keyboard.add("Удалить напоминание о подписке", "Главное меню")
        bot.register_next_step_handler_by_chat_id(message.chat.id, your_subs_menu)
        subs = list_subs(message.chat.id)
        each_subs_info = "\n\n".join([f"Сервис: {name_sub_service}\nДата приобретения: {date_sub}г.\nСтоимость подписки: {summ_sub}рублей (-я)\nСрок подписки: {period_sub} месяц(-а, -ев)" for name_sub_service, date_sub, summ_sub, period_sub in subs])
        your_subs_info = f"""
Вот вся информация о твоих подписках:

{each_subs_info}"""
        bot.send_message(message.chat.id, your_subs_info, reply_markup=ok_keyboard)
    elif message.text.lower() == "добавить новую подписку":
        add_sub_info = """
Все, что от тебя потребуется, так это написать мне название сервсиса, дату оплаты, стоимость и период подписки. За два дня до окончания подписки я напомню тебе о ней, а дальше сам решай, отключать ее или пользоваться ей дальше.
"""
        bot.send_message(message.chat.id, add_sub_info)
        bot.register_next_step_handler_by_chat_id(message.chat.id, add_sub_name)


def your_subs_menu(message):
    if message.text.lower() == "удалить напомниание о подписке":
        subs_list = []
        your_subs_list_keyboard = types.ReplyKeyboardMarkup(
            resize_keyboard=True,
            one_time_keyboard=True,
            row_width=1
        )
        for subs_button in subs_list:
            your_subs_list_keyboard.add(subs_button)
        choice_subs = """
Выбери ненужную подписку из списка:"""
        bot.send_message(message.chat.id, choice_subs,reply_markup=your_subs_list_keyboard)
    elif message.text.lower() == "главное меню":
        bot.send_message(message.chat.id, main_menu_message, reply_markup=main_menu_keyboard)


if __name__ == "__main__":
    bot.polling(none_stop=True, interval=0)