Fix validator registration query (#14)

stakewise · Dec 29, 2021 · ca8bb93 · ca8bb93
1 parent ed2b01a
commit ca8bb93
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/operator_cli/eth2.py b/operator_cli/eth2.py
@@ -155,7 +155,7 @@ def generate_unused_validator_keys(
             )
             registrations = result["validatorRegistrations"]
             for registration in registrations:
-                del pub_key_to_priv_key[registration["id"]]
+                del pub_key_to_priv_key[registration["publicKey"]]
 
             bar.update(len(pub_key_to_priv_key) - curr_progress)
 

diff --git a/operator_cli/queries.py b/operator_cli/queries.py
@@ -44,11 +44,11 @@ def get_stakewise_gql_client(chain: str) -> Client:
 
 REGISTRATIONS_QUERY = gql(
     """
-    query getValidatorRegistrations($public_keys: [ID!]) {
+    query getValidatorRegistrations($public_keys: [Bytes!]) {
       validatorRegistrations(
-        where: { id_in: $public_keys }
+        where: { publicKey_in: $public_keys }
       ) {
-        id
+        publicKey
       }
     }
 """