account->accounts

This commit is contained in:
Astra Logical 2026-07-23 14:04:52 -04:00
parent de7c2c6a68
commit 2582e69e3a

View file

@ -47,15 +47,8 @@ log = logging.getLogger("keycloak_bot")
# ------------------------------------------------------------------------------
class PickleCryptoStore(MemoryCryptoStore):
def __init__(self, filepath: str, account_id: str = "bot", pickle_key: str = ""):
self.filepath = filepath
self.account = None
self.sessions = {}
self.inbound_group_sessions = {}
self.outbound_group_sessions = {}
self.devices = {}
self.cross_signing_keys = {}
super().__init__(account_id=account_id, pickle_key=pickle_key)
self.filepath = filepath
self.load()
def load(self):
@ -63,7 +56,7 @@ class PickleCryptoStore(MemoryCryptoStore):
try:
with open(self.filepath, "rb") as f:
data = pickle.load(f)
self.account = data.get("account", None)
self.accounts = data.get("accounts", {})
self.sessions = data.get("sessions", {})
self.inbound_group_sessions = data.get("inbound_group_sessions", {})
self.outbound_group_sessions = data.get("outbound_group_sessions", {})
@ -76,7 +69,7 @@ class PickleCryptoStore(MemoryCryptoStore):
def save(self):
try:
data = {
"account": self.account,
"accounts": self.accounts,
"sessions": self.sessions,
"inbound_group_sessions": self.inbound_group_sessions,
"outbound_group_sessions": self.outbound_group_sessions,