ability to get single account by name

vladkens · Jan 5, 2024 · 490ffe2 · 490ffe2
1 parent ae2b15d
commit 490ffe2
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/twscrape/accounts_pool.py b/twscrape/accounts_pool.py
@@ -121,6 +121,13 @@ async def get_all(self):
         rs = await fetchall(self._db_file, qs)
         return [Account.from_rs(x) for x in rs]
 
+    async def get_account(self, username: str):
+        qs = "SELECT * FROM accounts WHERE username = :username"
+        rs = await fetchone(self._db_file, qs, {"username": username})
+        if not rs:
+            return None
+        return Account.from_rs(rs)
+
     async def save(self, account: Account):
         data = account.to_rs()
         cols = list(data.keys())