Explorar o código

Add missing method to implement login-matrix commands

Tulir Asokan %!s(int64=3) %!d(string=hai) anos
pai
achega
7fe1696975
Modificáronse 1 ficheiros con 5 adicións e 0 borrados
  1. 5 0
      mautrix_instagram/user.py

+ 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()