import datetime import os from telegram.ext import ConversationHandler, CommandHandler, MessageHandler, filters, CallbackQueryHandler from telegram import InlineKeyboardButton, InlineKeyboardMarkup from telegram.constants import ParseMode import models ENTRY_OPTIONS, CONTENT_ENTRY, DAY_RATING = range(3) BUTTON_COUNT = 5 from .basehandler import BaseHandler class JournalHandler(BaseHandler): def __init__(self, entry_string): self.entry_string = entry_string self.handler = ConversationHandler( entry_points=[CommandHandler(entry_string, self.entry_point)], states={ ENTRY_OPTIONS: [ CallbackQueryHandler(self.date_button, pattern=r"^\d{8}$"), # a serialized date CallbackQueryHandler(self.date_custom, pattern=r"^\d{1,3}$"), # a ~ small delta, symbolizing a new range to show CallbackQueryHandler(self.option_delete, pattern="delete"), MessageHandler(filters.ALL, self.date_entry), ], CONTENT_ENTRY: [ MessageHandler(filters.ALL, self.content_save), ], DAY_RATING: [ CallbackQueryHandler(self.day_rating_save), ], }, fallbacks=[], ) self.current_model = None async def entry_point(self, update, context): await super().entry_point(update, context) if os.getenv("DOCKERIZED", "false") == "true" and os.getenv("CHAT_ID") != str(update.message.chat_id): await update.message.reply_text("You are not authorized to use this bot") return ConversationHandler.END dates = [(datetime.datetime.now() - datetime.timedelta(days = i)).date() for i in range(BUTTON_COUNT + 2)][::-1] # since there are two buttons additional buttons, we need to have two more days names = get_names(dates) callbacks = [d.strftime("%d%m%Y") for d in dates] options = [ [InlineKeyboardButton(n, callback_data=c)] for n,c in zip(names[::-1], callbacks[::-1]) ] + [ [ InlineKeyboardButton("<<", callback_data=BUTTON_COUNT + 2) ], [ InlineKeyboardButton("Delete", callback_data="delete") ] ] keyboard = InlineKeyboardMarkup(options) await update.message.reply_text("Please choose a date \(or type it in the format _DDMMYYYY_\)", reply_markup=keyboard, parse_mode=ParseMode.MARKDOWN_V2) return ENTRY_OPTIONS async def date_button(self, update, context): query = update.callback_query await query.answer() date = datetime.datetime.strptime(query.data, "%d%m%Y").date() with models.db: self.current_model, new = models.JournalEntry.get_or_create( date = date ) if new: count = models.JournalEntry.select().count() await query.edit_message_text( text=f"Journal entry no. {count}. What happened on {self.current_model.date_pretty}?" ) else: await query.edit_message_text(text="An entry already exists for this date") return ConversationHandler.END return CONTENT_ENTRY async def date_custom(self, update, context): query = update.callback_query await query.answer() delta = int(query.data) dates = [(datetime.datetime.now() - datetime.timedelta(days = i + delta)).date() for i in range(BUTTON_COUNT)][::-1] names = get_names(dates) callbacks = [d.strftime("%d%m%Y") for d in dates] options = [ [ InlineKeyboardButton(">>", callback_data=delta - BUTTON_COUNT) ] ] + [ [InlineKeyboardButton(n, callback_data=c)] for n,c in zip(names[::-1], callbacks[::-1]) ] + [ [ InlineKeyboardButton("<<", callback_data=delta + BUTTON_COUNT) ], [ InlineKeyboardButton("Delete", callback_data="delete") ] ] keyboard = InlineKeyboardMarkup(options) await query.edit_message_text("Please choose a date \(or type it in the format _DDMMYYYY_\)", parse_mode=ParseMode.MARKDOWN_V2, reply_markup=keyboard) return ENTRY_OPTIONS async def date_entry(self, update, context): date = update.message.text try: date = datetime.datetime.strptime(date, "%d%m%Y").date() except ValueError: await update.message.reply_text("Please enter the date in the format _DDMMYYYY_", parse_mode=ParseMode.MARKDOWN_V2) return ENTRY_OPTIONS if context.chat_data.get("delete", False): # if not set, delete was not chosen with models.db: self.current_model = models.JournalEntry.get_or_none( date = date ) if self.current_model: await self.delete_entry(update, context) else: await update.message.reply_text("No entry found for this date") context.chat_data["delete"] = False return ConversationHandler.END else: with models.db: self.current_model, new = models.JournalEntry.get_or_create( date = date ) if not new: await update.message.reply_text("An entry already exists for this date") return ConversationHandler.END else: count = models.JournalEntry.select().count() await update.message.reply_text( text=f"Journal entry no. {count}. What happened on {self.current_model.date_pretty}?" ) return CONTENT_ENTRY async def content_save(self, update, context): with models.db: self.current_model.author_id = update.message.from_user.id if update.message.text: self.current_model.text = update.message.text else: if update.message.photo: file = await update.message.effective_attachment[-1].get_file() else: file = await update.message.effective_attachment.get_file() file_bytes = await file.download_as_bytearray() file_path = file.file_path self.current_model.save_media(file_bytes, file_path) self.current_model.text = update.message.caption self.current_model.save() options = [ [InlineKeyboardButton(models.RATING_MAPPING[idx], callback_data=idx) for idx in [1,2,3,4,5]] ] await update.message.reply_text(f"Saved entry ✅. How was the day?", reply_markup=InlineKeyboardMarkup(options)) return DAY_RATING async def day_rating_save(self, update, context): query = update.callback_query await query.answer() rating = int(query.data) with models.db: self.current_model.rating = rating self.current_model.save() await query.edit_message_text(text="Rating saved ✅") return ConversationHandler.END async def option_delete(self, update, context): query = update.callback_query await query.answer() await query.edit_message_text(text="Please enter the date in the format _DDMMYYYY_", parse_mode=ParseMode.MARKDOWN_V2) context.chat_data["delete"] = True return ENTRY_OPTIONS async def delete_entry(self, update, context): with models.db: self.current_model.delete_instance() context.chat_data["delete"] = False await update.message.reply_text(text="Entry deleted ✅") ### HELPERS def get_names(dates: list): names = [] for d in dates: suffix = "" if models.JournalEntry.get_or_none(date = d): suffix = " ✅" if d == datetime.datetime.now().date(): names.append("Today" + suffix) elif d == datetime.datetime.now().date() - datetime.timedelta(days = 1): names.append("Yesterday" + suffix) else: names.append(d.strftime("%d.%m.") + suffix) return names