added journal command

bot/commands/journal.py

@@ -1,6 +1,6 @@
 import datetime
-
-from telegram.ext import ConversationHandler, CommandHandler, MessageHandler, filters, CallbackQueryHandler, CallbackContext
+import os
+from telegram.ext import ConversationHandler, CommandHandler, MessageHandler, filters, CallbackQueryHandler
 from telegram import InlineKeyboardButton, InlineKeyboardMarkup
 DATE_CHOICE, DATE_ENTRY, CONTENT = range(3)
 
@@ -28,8 +28,15 @@ class JournalHandler:
 
         self.current_model = None
 
+
     async def start(self, update, context):
         """Send a message when the command /start is issued."""
+        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
+
+        print(f"User: {update.message.from_user.id}")
+        print(f"Chat: {update.message.chat_id}")
 
         options = [
             InlineKeyboardButton("Today", callback_data="today"),
@@ -40,28 +47,36 @@ class JournalHandler:
         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()
+        await 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(
+            self.current_model, new = self.models.JournalEntry.get_or_create(
                 date = date
             )
+            if not new:
+                await query.edit_message_text(text="An entry already exists for this date")
+                return ConversationHandler.END
+            await query.edit_message_text(
+                text="Please enter the content for the entry"
+            )
             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(
+            self.current_model, new = self.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
         except ValueError:
             await update.message.reply_text("Please enter the date in the format DDMMYYYY")
             return DATE_ENTRY
@@ -69,8 +84,26 @@ class JournalHandler:
         await update.message.reply_text("Please enter the content for the entry")
         return CONTENT
 
+
     async def content_text(self, update, context):
+        self.current_model.text = update.message.text
+        self.current_model.author_id = update.message.from_user.id
+        self.current_model.save()
+        return ConversationHandler.END
+
     
-        return 
     async def content_media(self, update, context):
-        return  
+        self.current_model.author_id = update.message.from_user.id
+
+        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()
+        return ConversationHandler.END

bot/models.py

@@ -1,5 +1,15 @@
 from peewee import *
 from pathlib import Path
+import os
+
+ID_MAPPINGS = {
+    "Lia": 0,
+    "Rémy": 364520272,
+}
+ID_MAPPINGS_REV = dict((v, k) for k, v in ID_MAPPINGS.items())
+
+MEDIA_DIR = Path(os.getenv("MEDIA_DIR"))
+MEDIA_DIR.mkdir(parents=True, exist_ok=True)
 
 db = DatabaseProxy()
 
@@ -11,25 +21,38 @@ class BaseModel(Model):
 
 # model for a single journal entry
 class JournalEntry(BaseModel):
-    # the date of the entry
-    date = DateField()
-    # the text of the entry
-    text = TextField()
-    media_path = TextField()
-    author = TextField()
+    date = DateField(unique=True)
+    author = TextField(null=True)
+    text = TextField(null=True)
+    media_path = TextField(null=True)
 
     
     @property
     def media(self):
         return Path(self.media_path).open('rb')
     
-    @media.setter
-    def media(self, media):
-        self.media_path = Path(media).absolute()
+    def save_media(self, media: bytearray, file_name: str):
+        ext = Path(file_name).suffix
+        file_name = f"{self.date.isoformat()}-media{ext}"
+        self.media_path = MEDIA_DIR / file_name
+        self.media_path.write_bytes(media)
         self.save()
 
+    @property
+    def author_id(self):
+        if self.author is None:
+            return None
+        else:
+            return ID_MAPPINGS[self.author]
+
+    @author_id.setter
+    def author_id(self, author_id):
+        self.author = ID_MAPPINGS_REV[author_id]
+        self.save()
+
+
 
 def set_db(db_path):
     db.initialize(SqliteDatabase(db_path))
     db.connect()
-    db.create_tables([JournalEntry], safe=True)
+    db.create_tables([JournalEntry], safe=True)