Fixed afk

Harshushende28 · Apr 15, 2021 · b7f8cd5 · b7f8cd5
1 parent 13af61d
commit b7f8cd5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/plugins/afk.py b/plugins/afk.py
@@ -20,7 +20,9 @@
 
 from pyUltroid.functions.pmpermit_db import *
 from telethon import events
-from telethon.tl import functions, types
+from telethon.tl.functions.account import GetPrivacyRequest
+from telethon.tl.types import InputPrivacyKeyStatusTimestamp, PrivacyValueAllowAll
+
 
 from . import *
 
@@ -182,9 +184,9 @@ async def _(event):
         pic = None
     if not USER_AFK:
         last_seen_status = await ultroid_bot(
-            functions.account.GetPrivacyRequest(types.InputPrivacyKeyStatusTimestamp()),
+            GetPrivacyRequest(InputPrivacyKeyStatusTimestamp()),
         )
-        if isinstance(last_seen_status.rules, types.PrivacyValueAllowAll):
+        if isinstance(last_seen_status.rules, PrivacyValueAllowAll):
             afk_time = datetime.datetime.now()
         USER_AFK = f"yes: {reason} {pic}"
         if reason: