web3-wizzard-lib 1.0.5__py3-none-any.whl → 1.1.0__py3-none-any.whl

web3_wizzard_lib/core/contract/gmx_reward_router.py

@@ -0,0 +1,22 @@
+from sybil_engine.contract.contract import Contract
+from sybil_engine.contract.transaction_executor import evm_transaction
+from sybil_engine.utils.file_loader import load_abi
+
+abi = load_abi("resources/abi/gmx_reward_router.json")
+
+
+class GmxRewardRouter(Contract):
+    def __init__(self, contract_address, web3):
+        super().__init__(contract_address, web3, abi)
+
+    @evm_transaction
+    def unstake_and_redeem(self, account, shares):
+        txn_params = self.build_generic_data(account.address, False)
+
+        contract_txn = self.contract.functions.redeem(
+            shares,
+            0,
+            account.address,
+        ).build_transaction(txn_params)
+
+        return contract_txn

web3_wizzard_lib/core/contract/new_rage_contract.py

@@ -0,0 +1,28 @@
+from sybil_engine.contract.contract import Contract
+from sybil_engine.contract.transaction_executor import evm_transaction
+from sybil_engine.utils.file_loader import load_abi
+
+abi = load_abi("resources/abi/new_rage_abi.json")
+
+
+class NewRageContract(Contract):
+    def __init__(self, contract_address, web3):
+        super().__init__(contract_address, web3, abi)
+
+    @evm_transaction
+    def redeem(self, account, shares):
+        txn_params = self.build_generic_data(account.address, False)
+
+        contract_txn = self.contract.functions.redeem(
+            shares,
+            account.address,
+            account.address,
+        ).build_transaction(txn_params)
+
+        return contract_txn
+
+    def balance_of(self, account):
+        return self.contract.functions.balanceOf(account.address).call()
+
+    def convert_to_shares(self, amount):
+        return self.contract.functions.convertToShares(amount).call()

web3_wizzard_lib/core/modules/concrete_swap.py

@@ -8,6 +8,7 @@ from sybil_engine.module.module import Module
 from sybil_engine.utils.utils import SwapException
 from sybil_engine.utils.validation_utils import validate_amount_interval, validate_token
 from sybil_engine.utils.web3_utils import init_web3
+from web3 import Web3
 
 from web3_wizzard_lib.core.modules.swap.swap_list import swap_facade
 
@@ -31,10 +32,20 @@ class ConcreteSwap(Module):
         from_token_contract = Erc20Token(chain, from_token, web3)
         to_token_contract = Erc20Token(chain, to_token, web3)
 
+        if Web3.is_address(from_token):
+            from_token_symbol = from_token_contract.symbol()
+        else:
+            from_token_symbol = from_token
+
+        if Web3.is_address(to_token):
+            to_token_symbol = to_token_contract.symbol()
+        else:
+            to_token_symbol = to_token
+
         pair_to_swap = self.create_pair_to_swap(
-            from_token_contract.symbol(),
+            from_token_symbol,
             swap_app,
-            to_token_contract.symbol(),
+            to_token_symbol,
         )
 
         amount_to_swap = amount_to_swap_from_interval(

web3_wizzard_lib/core/modules/gmx_reward.py

@@ -0,0 +1,27 @@
+from loguru import logger
+from sybil_engine.data.contracts import get_contracts_for_chain
+from sybil_engine.data.networks import get_chain_instance
+from sybil_engine.module.module import Module
+from sybil_engine.utils.accumulator import add_accumulator
+from sybil_engine.utils.utils import ConfigurationException
+from sybil_engine.utils.web3_utils import init_web3
+
+from web3_wizzard_lib.core.contract.gmx_reward_router import GmxRewardRouter
+from web3_wizzard_lib.core.contract.new_rage_contract import NewRageContract
+
+
+class GMXRewardRouter(Module):
+    module_name = 'GMX_REWARD_ROUTER'
+    module_config = None
+
+    def execute(self, account, chain='ARBITRUM'):
+        chain_instance = get_chain_instance(chain)
+        web3 = init_web3(chain_instance, None)
+
+        contract_address = get_contracts_for_chain(chain_instance['chain'])['GMX_REWARD_ROUTER']
+        gmx_reward_router = GmxRewardRouter(contract_address, web3)
+
+        gmx_reward_router.unstake_and_redeem(account)
+
+    def log(self):
+        return "GMX REWARD WITHDRAW"

web3_wizzard_lib/core/modules/new_rage_withdraw.py

@@ -0,0 +1,42 @@
+from loguru import logger
+from sybil_engine.data.networks import get_chain_instance
+from sybil_engine.module.module import Module
+from sybil_engine.utils.accumulator import add_accumulator_balance, add_accumulator
+from sybil_engine.utils.utils import ConfigurationException
+from sybil_engine.utils.web3_utils import init_web3
+
+from web3_wizzard_lib.core.contract.new_rage_contract import NewRageContract
+
+
+class NewRageModule(Module):
+    module_name = 'NEW_RAGE_WITHDRAW'
+    module_config = 'new_rage_config'
+
+    def execute(self, token, account, chain='ARBITRUM'):
+        chain_instance = get_chain_instance(chain)
+        web3 = init_web3(chain_instance, None)
+
+        if token == 'USDC':
+            contract_address = "0xf9305009FbA7E381b3337b5fA157936d73c2CF36"
+        elif token == 'GLP':
+            contract_address = "0x8478AB5064EbAC770DdCE77E7D31D969205F041E"
+        else:
+            raise ConfigurationException("No such token supported by Rage withdraw module")
+        rage = NewRageContract(contract_address, web3)
+
+        balance = rage.balance_of(account)
+        shares = rage.convert_to_shares(balance)
+
+        if balance > 0:
+            logger.info(f"Withdraw {balance} {token} which is {shares} shares")
+            rage.redeem(account, shares)
+        else:
+            logger.info(f"Account {account.address} has 0 balance in Rage")
+
+        add_accumulator("Total rage withdraw", balance)
+
+    def log(self):
+        return "RAGE WITHDRAW"
+
+    def parse_params(self, module_params):
+        return module_params['token'],

web3_wizzard_lib/core/modules/swap/odos.py

@@ -9,7 +9,7 @@ from web3_wizzard_lib.core.contract.odos import Odos
 class OdosSwap(Dex):
     dex_name = 'odos'
     swap_contract = 'ODOS'
-    supported_chains = ['ZKSYNC', 'BASE', 'LINEA']
+    supported_chains = ['ZKSYNC', 'BASE', 'LINEA', 'ARBITRUM']
 
     def __init__(self, chain_instance, web3):
         super().__init__(chain_instance, web3)

web3_wizzard_lib/core/modules/swap/woofi.py

@@ -9,7 +9,7 @@ from web3_wizzard_lib.core.contract.woofi_swap import WoofiSwap
 class Woofi(Dex):
     dex_name = 'woofi'
     swap_contract = 'WOOFI'
-    supported_chains = ['ZKSYNC', 'BASE', 'LINEA']
+    supported_chains = ['ZKSYNC', 'BASE', 'LINEA', 'ARBITRUM']
 
     def __init__(self, chain_instance, web3):
         super().__init__(chain_instance, web3)