2023-04-23 16:54:58 +00:00
|
|
|
|
import redis
|
2023-09-11 15:52:35 +00:00
|
|
|
|
from profile import Profile as ProfileObj
|
|
|
|
|
from bot.config import REDIS_URL
|
2023-04-28 12:24:14 +00:00
|
|
|
|
import json
|
|
|
|
|
|
2023-04-23 16:54:58 +00:00
|
|
|
|
|
2023-04-24 06:14:35 +00:00
|
|
|
|
# сохраняет сессии, айди кнопок в чатах для удаления и пересылаемые сообщения между перезагрузками
|
2023-04-23 16:54:58 +00:00
|
|
|
|
storage = redis.from_url(REDIS_URL)
|
|
|
|
|
|
|
|
|
|
# хранение необходимой информации о пользователях
|
|
|
|
|
Profile = ProfileObj(storage)
|
2023-04-28 12:24:14 +00:00
|
|
|
|
|
|
|
|
|
# достаёт из хранилища jsonы по маске и количеству
|
|
|
|
|
def scan(match='usr-*', count=100):
|
|
|
|
|
cursor = 0
|
|
|
|
|
keys = []
|
|
|
|
|
r = storage
|
|
|
|
|
while True:
|
|
|
|
|
# Scan for keys starting with <match> in batches of <count>
|
|
|
|
|
cursor, batch_keys = r.scan(cursor=cursor, match=match, count=count)
|
|
|
|
|
keys += batch_keys
|
|
|
|
|
# If the cursor is 0, then we've reached the end of the keys
|
|
|
|
|
if cursor == 0:
|
|
|
|
|
break
|
|
|
|
|
# Get the values of all the keys
|
|
|
|
|
values = r.mget(keys)
|
|
|
|
|
# Parse the JSON data from each value
|
|
|
|
|
items = []
|
|
|
|
|
for value in values:
|
|
|
|
|
value_str = value.decode('utf-8')
|
|
|
|
|
i = json.loads(value_str)
|
|
|
|
|
items.append(i)
|
|
|
|
|
print(f'scan found {len(items)} items')
|
|
|
|
|
|
|
|
|
|
return keys, items
|