add main so no error.

BalancerMaxis · Oct 9, 2023 · 1e5fc40 · 1e5fc40
1 parent a3c4cfc
commit 1e5fc40
Showing 1 changed file with 19 additions and 14 deletions.
diff --git a/action-scripts/brownie/scripts/pokeInjectors.py b/action-scripts/brownie/scripts/pokeInjectors.py
@@ -2,18 +2,23 @@
 from brownie import network, accounts, Contract
 import os
 
-## LOAD wallet
-mnemonic =  os.environ["KEYWORDS"]
-account = accounts.from_mnemonic(mnemonic)
-print(f"Keeper Address: {account.address}")
-## ZKEVM
-network.disconnect()
-network.connect("zkevm-main")
-book = AddrBook("zkevm")
-injectors =  book.extras.maxiKeepers.gaugeRewardsInjectors.values()
-for injectorAddress in injectors:
-    injector = Contract(injectorAddress)
-    (ready, performdata) = injector.checkUpkeep(b"")
-    if ready:
-        injector.performUpkeep(performdata, {"from": account})
+def main()
+    ## LOAD wallet
+    mnemonic =  os.environ["KEYWORDS"]
+    account = accounts.from_mnemonic(mnemonic)
+    print(f"Keeper Address: {account.address}")
+    ## ZKEVM
+    network.disconnect()
+    network.connect("zkevm-main")
+    book = AddrBook("zkevm")
+    injectors =  book.extras.maxiKeepers.gaugeRewardsInjectors.values()
+    for injectorAddress in injectors:
+        injector = Contract(injectorAddress)
+        (ready, performdata) = injector.checkUpkeep(b"")
+        if ready:
+            injector.performUpkeep(performdata, {"from": account})
+        else:
+            print(f"{injector.address} not ready")
 
+if __name__ == "__main__":
+    main()