diff --git a/matrix_keycloak_bot.py b/matrix_keycloak_bot.py index 6527e90..b809bc3 100644 --- a/matrix_keycloak_bot.py +++ b/matrix_keycloak_bot.py @@ -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,