andreacavalli
/
pyrogram-rockserver-storage
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix username bug

This commit is contained in:
Andrea Cavalli 2024-03-30 22:22:27 +01:00
parent 1c39d50386
commit b32035f549
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pyrogram_rockserver_storage/__init__.py
View File

@ -258,6 +258,9 @@ class RockServerStorage(Storage):
encoded_value = await fetchone(self._client, self._peer_col, keys)
value_tuple = decode_peer_info(peer_id, encoded_value)
if value_tuple is None:
raise KeyError(f"Username not found: {username}")
if int(time.time() - value_tuple['last_update_on']) > self.USERNAME_TTL:
raise KeyError(f"Username expired: {username}")