From 7bca30df706ecb06b342ff4eee4886ce408ac8ba Mon Sep 17 00:00:00 2001 From: MathiasDPX Date: Thu, 17 Apr 2025 22:21:53 +0200 Subject: [PATCH] fix: vault owner check --- main.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index b5b0216..b0f3f88 100644 --- a/main.py +++ b/main.py @@ -39,9 +39,13 @@ def websocket(ws): ws.send(json.dumps({"res": "ok", "perFileMax": 2147483647, "userId": 1})) ws.send(json.dumps({"op": "ready", "version": 11})) + hisVault = False for vault in database.list_vaults(data.get("token"))['vaults']: - if vault['id'] != vault_id: - ws.close() + if vault['id'] == vault_id: + hisVault = True + + if not hisVault: + ws.close() vault_id = data.get("id") device_name = data.get("device", "unknown")