From 78428e8ece682f9b8b0335011cc9ea0eb6302687 Mon Sep 17 00:00:00 2001 From: Amir Rachum Date: Sat, 28 Sep 2024 16:30:41 +0300 Subject: [PATCH] Fix book timekilling feature --- src/spanreed/plugins/timekiller.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/spanreed/plugins/timekiller.py b/src/spanreed/plugins/timekiller.py index 4633aee..eb7a119 100644 --- a/src/spanreed/plugins/timekiller.py +++ b/src/spanreed/plugins/timekiller.py @@ -18,6 +18,8 @@ class TimekillerPlugin(Plugin): + LAST_ASKED_BOOKS_KEY = "currently-reading-books-last-asked" + @classmethod def name(cls) -> str: return "Timekiller" @@ -58,10 +60,12 @@ async def get_available_time_killers( "Scan Processing": self.prompt_for_scan_processing, } + last_asked: datetime.datetime = datetime.datetime.now() - datetime.timedelta( + days=4 + ) last_asked_str: str | None = await self.get_user_data( - user, "currently-reading-books-last-asked" + user, self.LAST_ASKED_BOOKS_KEY ) - last_asked: datetime.datetime = datetime.datetime.now() if last_asked_str is not None: last_asked = datetime.datetime.fromisoformat( last_asked_str @@ -324,6 +328,7 @@ async def prompt_for_currently_reading_books( "\n\n### Thoughts\n" + await bot.request_user_input("Go ahead then:"), ) + await self.set_user_data(user, self.LAST_ASKED_BOOKS_KEY, str(datetime.datetime.now())) async def prompt_for_scan_processing( self, _user: User, bot: TelegramBotApi, obsidian: ObsidianApi