Explorar el Código

fix: _ban_restricted_user_until_date (#257)

SI_NN_ER_LS hace 2 años
padre
commit
2f88fd6ea9
Se han modificado 1 ficheros con 3 adiciones y 3 borrados
  1. 3 3
      wbb/modules/greetings.py

+ 3 - 3
wbb/modules/greetings.py

@@ -24,7 +24,7 @@ SOFTWARE.
 
 
 import asyncio
 import asyncio
 import os
 import os
-from datetime import datetime
+from datetime import datetime, timedelta
 from random import shuffle
 from random import shuffle
 
 
 from pyrogram import filters
 from pyrogram import filters
@@ -239,7 +239,7 @@ async def handle_new_member(member, chat):
 async def welcome(_, user: ChatMemberUpdated):
 async def welcome(_, user: ChatMemberUpdated):
     if not (
     if not (
         user.new_chat_member
         user.new_chat_member
-        and user.new_chat_member.status not in {CMS.BANNED}
+        and user.new_chat_member.status in {CMS.RESTRICTED}
         and not user.old_chat_member
         and not user.old_chat_member
     ):
     ):
         return
         return
@@ -403,7 +403,7 @@ async def _ban_restricted_user_until_date(
     try:
     try:
         member = await group_chat.get_member(user_id)
         member = await group_chat.get_member(user_id)
         if member.status == ChatMemberStatus.RESTRICTED:
         if member.status == ChatMemberStatus.RESTRICTED:
-            until_date = int(datetime.utcnow().timestamp() + duration)
+            until_date = (datetime.now() + timedelta(seconds=duration))
             await group_chat.ban_member(user_id, until_date=until_date)
             await group_chat.ban_member(user_id, until_date=until_date)
     except UserNotParticipant:
     except UserNotParticipant:
         pass
         pass