remoll/journal-bot
1
0
Fork 0
Code Issues 1 Pull Requests 4 Actions Packages Projects Releases Wiki Activity

actually wal is reuqired?
Some checks reported errors
continuous-integration/drone/push Build was killed
Details

Browse Source
This commit is contained in:
Remy Moll
2023-04-24 20:19:44 +02:00
parent 0dfb5f3134
commit d9051ee6c0
3 changed files with 30 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
bot/commands/journal.py
View File

@@ -70,38 +70,42 @@ class JournalHandler(BaseHandler):
date = update.message.text
try:
date = datetime.datetime.strptime(date, "%d%m%Y").date()
with self.models.db:
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
await update.message.reply_text("Please enter the content for the entry")
return CONTENT
self.logger.info(f"Date: {date}")
with self.models.db:
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
else:
await update.message.reply_text("Please enter the content for the entry")
return CONTENT
async def content_save(self, update, context):
self.current_model.author_id = update.message.from_user.id
with self.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()
if update.message.text:
self.current_model.text = update.message.text
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)
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()
self.current_model.text = update.message.caption
self.current_model.save()
await update.message.reply_text(f"Saved entry for {self.current_model.date.strftime('%A, %-d. %B %Y')}")
return ConversationHandler.END