initial framework

bot/commands/journal.py

@@ -0,0 +1,76 @@
+import datetime
+
+from telegram.ext import ConversationHandler, CommandHandler, MessageHandler, filters, CallbackQueryHandler, CallbackContext
+from telegram import InlineKeyboardButton, InlineKeyboardMarkup
+DATE_CHOICE, DATE_ENTRY, CONTENT = range(3)
+
+
+class JournalHandler:
+    def __init__(self, entry_string, models):
+        self.models = models
+
+        self.handler = ConversationHandler(
+            entry_points=[CommandHandler(entry_string, self.start)],
+            states={
+                DATE_CHOICE: [
+                    CallbackQueryHandler(self.date_choice),
+                    ],
+                DATE_ENTRY: [
+                    MessageHandler(filters.TEXT, self.date_entry),
+                    ],
+                CONTENT: [
+                    MessageHandler(filters.TEXT, self.content_text),
+                    MessageHandler(filters.ATTACHMENT, self.content_media),
+                    ],
+            },
+            fallbacks=[],
+        )
+
+        self.current_model = None
+
+    async def start(self, update, context):
+        """Send a message when the command /start is issued."""
+
+        options = [
+            InlineKeyboardButton("Today", callback_data="today"),
+            InlineKeyboardButton("Yesterday", callback_data="yesterday"),
+            InlineKeyboardButton("Custom date", callback_data="custom"),
+        ]
+        keyboard = InlineKeyboardMarkup([options])
+        await update.message.reply_text("Please choose an option for the entry:", reply_markup=keyboard)
+        return DATE_CHOICE
+
+    async def date_choice(self, update, context):
+        query = update.callback_query
+        query.answer()
+        if query.data == "today" or query.data == "yesterday":
+            date = datetime.datetime.now().date() if query.data == "today" else datetime.datetime.now().date() - datetime.timedelta(days=1)
+            self.current_model = self.models.JournalEntry(
+                date = date
+            )
+            return CONTENT
+        else:
+            await query.edit_message_text(text="Please enter the date in the format DDMMYYYY")
+            return DATE_ENTRY
+
+    async def date_entry(self, update, context):
+        # create an inline keyboard with the option today and yesterday and custom
+        # date
+        date = update.message.text
+        try:
+            date = datetime.datetime.strptime(date, "%d%m%Y").date()
+            self.current_model = self.models.JournalEntry(
+                date = date
+            )
+        except ValueError:
+            await update.message.reply_text("Please enter the date in the format DDMMYYYY")
+            return DATE_ENTRY
+
+        await update.message.reply_text("Please enter the content for the entry")
+        return CONTENT
+
+    async def content_text(self, update, context):
+    
+        return 
+    async def content_media(self, update, context):
+        return