fix?

openkriemy.py

@@ -1,4 +1,5 @@
 from asyncio import run, sleep
+from datetime import datetime, timedelta, date, time
 
 from telethon import TelegramClient
 from telethon.events import NewMessage
@@ -14,7 +15,7 @@ from utils import (
 from config import config
 from db import init_db
 from actions import (
-  find_action, 
+  find_action,
   get_random_gif,
   is_admin,
   is_allowed
@@ -22,8 +23,8 @@ from actions import (
 from commands import COMMANDS
 
 bot = TelegramClient(
-  'openkriemy', 
-  config.API_ID, 
+  'openkriemy',
+  config.API_ID,
   config.API_HASH
 ).start(bot_token=config.API_TOKEN)
 
@@ -48,7 +49,7 @@ async def on_message(event):
       await handler.handler(bot, event, command)
 
     return
-     
+
   try:
     action = await find_action(command.name)
   except SyntaxError:
@@ -70,7 +71,7 @@ async def on_message(event):
         if action.kind != Kind.NO_TARGET_MAYBE:
           await event.reply('Это действие нужно применить на кого-то!')
 
-          return 
+          return
     else:
       reply_to = target
       target = target.sender
@@ -82,13 +83,13 @@ async def on_message(event):
    and target.id == event.sender.id:
       await event.reply('Данное действие нельзя применять к самому себе...')
 
-      return 
-    
+      return
+
   try:
     await event.delete()
   except:
     pass
-    
+
   if event.sender is None:
     initiator = await bot.get_entity(event.peer_id.channel_id)
     initiator = initiator.title
@@ -99,7 +100,7 @@ async def on_message(event):
     'initiator': initiator,
     'target': get_link_to_user(target) if target else ''
   })
-  
+
   gif = await get_random_gif(action)
 
   if gif:
@@ -112,34 +113,47 @@ async def on_message(event):
     reply_to=reply_to
   )
 
-# cursed
 async def notify_birthdays():
-  while True:
-    birthdays = await get_all_birthdays()
+  birthdays = await get_all_birthdays()
 
-    for birthday in birthdays:
-      age = calculate_age(birthday.date)
+  for birthday in birthdays:
+    age = calculate_age(birthday.date)
 
-      if age.days_until < 1:
+    if age.days_until < 1:
+      try:
         try:
-          try:
-            entity = await bot.get_entity(birthday.user_id)
-          except ValueError:
-            await bot.get_participants(birthday.peer_id)
+          entity = await bot.get_entity(birthday.user_id)
+        except ValueError:
+          await bot.get_participants(birthday.peer_id)
+
+          entity = await bot.get_entity(birthday.user_id)
+
+        await bot.send_message(
+          birthday.peer_id,
+          f'{get_link_to_user(entity)}, поздравляю с днём рождения!!~~'
+        )
+      except:
+        pass
+
+async def schedule(task, when):
+  await sleep(
+    ((when - datetime.now()) % timedelta(days=1)).total_seconds()
+  )
 
-            entity = await bot.get_entity(birthday.user_id)
+  await task
 
-          await bot.send_message(
-            birthday.peer_id,
-            f'{get_link_to_user(entity)}, поздравляю с днём рождения!!~~'
-          )
-        except:
-          pass
+async def notify_birthdays_loop():
+  interval = datetime.combine(date.today(), time(hour=0, minute=0))
+
+  while True:
+    await sleep(
+      ((interval - datetime.now()) % timedelta(days=1)).total_seconds()
+    )
 
-    await sleep(60 * 60) # 1 hour.
+    await notify_birthdays()
 
 with bot:
   bot.loop.run_until_complete(init_db())
-  bot.loop.create_task(notify_birthdays())
+  bot.loop.create_task(notify_birthdays_loop())
   bot.start()
   bot.run_until_disconnected()