Mxid to pickle

This commit is contained in:
Astra Logical 2026-07-23 13:35:33 -04:00
parent 322e418a81
commit a95a3e274f

View file

@ -45,8 +45,8 @@ log = logging.getLogger("keycloak_bot")
# Custom Pickle-Backed Crypto Store for Persistence
# ------------------------------------------------------------------------------
class PickleCryptoStore(MemoryCryptoStore):
def __init__(self, filepath: str):
super().__init__()
def __init__(self, filepath: str, account_id: str = "bot", pickle_key: str = ""):
super().__init__(account_id=account_id, pickle_key=pickle_key)
self.filepath = filepath
self.load()
@ -406,7 +406,7 @@ async def main():
# Initialize stores using Pickle and Custom Memory State Stores
state_store = CustomMemoryStateStore(client.mxid)
crypto_store = PickleCryptoStore(CRYPTO_STORE_FILE)
crypto_store = PickleCryptoStore(CRYPTO_STORE_FILE, account_id=client.mxid)
client.state_store = state_store
client.crypto = OlmMachine(client, crypto_store, state_store)