Эх сурвалжийг харах

Add missing method to implement login-matrix commands

Tulir Asokan 3 жил өмнө
parent
commit
7fe1696975

+ 5 - 0
mautrix_instagram/user.py

@@ -109,6 +109,11 @@ class User(DBUser, BaseUser):
     async def is_logged_in(self) -> bool:
         return bool(self.client) and self._is_logged_in
 
+    async def get_puppet(self) -> Optional[pu.Puppet]:
+        if not self.igpk:
+            return None
+        return await pu.Puppet.get_by_pk(self.igpk)
+
     async def try_connect(self) -> None:
         try:
             await self.connect()