web3-wizzard-lib 1.7.0__py3-none-any.whl → 1.8.0__py3-none-any.whl

web3_wizzard_lib/core/contract/cog_erc20.py

@@ -0,0 +1,47 @@
+from loguru import logger
+from web3 import Web3
+
+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/cog_erc20.json")
+
+MAX_ALLOWANCE = 115792089237316195423570985008687907853269984665640564039457584007913129639935
+
+
+class CogErc20Contract(Contract):
+    def __init__(self, contract_address, web3):
+        super().__init__(contract_address, web3, abi)
+
+    @evm_transaction
+    def approve(self, account, contract_on_approve):
+        logger.info(f"Approving token")
+
+        txn_params = self.build_generic_data(account.address, set_contract_address=False)
+
+        return self.contract.functions.approve(
+            Web3.to_checksum_address(contract_on_approve),
+            MAX_ALLOWANCE
+        ).build_transaction(txn_params)
+
+    @evm_transaction
+    def redeem(self, account, amount):
+        txn_params = self.build_generic_data(account.address, set_contract_address=False)
+
+        return self.contract.functions.redeem(amount).build_transaction(txn_params)
+
+    def balance_of(self, account):
+        return self.contract.functions.balanceOf(account.address).call()
+
+    def max_redeem(self, account):
+        return self.contract.functions.maxRedeem(account.address).call()
+
+    def allowance(self, account, allowance_contract):
+        return self.contract.functions.allowance(account.address, Web3.to_checksum_address(allowance_contract)).call()
+
+    def decimals(self):
+        return self.contract.functions.decimals().call()
+
+    def symbol(self):
+        return self.contract.functions.symbol().call()

web3_wizzard_lib/core/modules/bank/cog_bank.py

@@ -0,0 +1,26 @@
+from sybil_engine.data.contracts import get_contracts_for_chain
+from sybil_engine.data.networks import get_ids_chain
+from sybil_engine.utils.utils import ConfigurationException
+
+from web3_wizzard_lib.core.contract.cog_erc20 import CogErc20Contract
+from web3_wizzard_lib.core.modules.bank.bank import Bank
+
+
+class Cog(Bank):
+    app_name = 'COG_USDC'
+    supported_chains = ['SCROLL']
+
+    def __init__(self, chain, web3):
+        contract_address = get_contracts_for_chain(chain)[self.app_name]
+        self.contract = CogErc20Contract(contract_address, web3)
+
+    def supply(self, account, amount):
+        raise ConfigurationException("Only redeem supported for Compound V3")
+
+    def redeem(self, account, amount, token):
+        self.contract.redeem(account, amount)
+
+    def get_deposit_amount(self, account, token):
+        cog_usdc_token = get_contracts_for_chain(get_ids_chain()[self.contract.web3.eth.chain_id])['COG_USDC']
+
+        return CogErc20Contract(cog_usdc_token, self.contract.web3).balance_of(account)

web3_wizzard_lib/core/modules/bank/compound_v3.py

@@ -22,14 +22,13 @@ class CompoundV3(Bank):
         raise ConfigurationException("Only redeem supported for Compound V3")
 
     def redeem(self, account, amount, token):
-        redeem_action = '0x414354494f4e5f57495448445241575f4e41544956455f544f4b454e00000000'
-        self.contract_bulker.invoke(account, amount, redeem_action)
+        self.contract_bulker.invoke(account, amount)
 
     def get_deposit_amount(self, account, token):
         token_address = get_tokens_for_chain(get_ids_chain()[self.contract.web3.eth.chain_id])[token]
-        compound_v3 = get_contracts_for_chain(get_ids_chain()[self.contract.web3.eth.chain_id])[self.app_name]
+        compound_v3_address = get_contracts_for_chain(get_ids_chain()[self.contract.web3.eth.chain_id])[self.app_name]
 
-        return CompoundV3Contract(compound_v3, self.contract.web3).user_collateral(
+        return CompoundV3Contract(compound_v3_address, self.contract.web3).user_collateral(
             account,
             token_address
         )

web3_wizzard_lib/core/modules/bank_module.py

@@ -9,6 +9,7 @@ from sybil_engine.utils.utils import ConfigurationException
 from sybil_engine.utils.web3_utils import init_web3
 
 from web3_wizzard_lib.core.modules.bank.basilisk import Basilisk
+from web3_wizzard_lib.core.modules.bank.cog_bank import Cog
 from web3_wizzard_lib.core.modules.bank.compound_v3 import CompoundV3
 from web3_wizzard_lib.core.modules.bank.eralend import Eralend
 from web3_wizzard_lib.core.modules.bank.layerbank import LayerBank
@@ -38,12 +39,11 @@ class Banking(Module):
 
         bank_app = self.get_bank_app(bank_app_name, chain, web3)
 
-        logger.info(f"{bank_app.app_name}")
+        logger.info(f"{bank_app.app_name} | {action}")
 
         self.perform_action(bank_app, action, amount, token, account)
 
     def perform_action(self, bank_app, action, amount, token, account):
-        logger.info(action)
         if action == 'SUPPLY':
             bank_app.supply(account, amount)
         elif action == 'BORROW':
@@ -52,9 +52,12 @@ class Banking(Module):
             bank_app.repay_borrow(account, amount)
         elif action == 'REDEEM':
             amount = bank_app.get_deposit_amount(account, token)
-            logger.info(f"Redeem {amount} of {token} from {bank_app.app_name}")
 
-            bank_app.redeem(account, amount, token)
+            if amount > 0:
+                logger.info(f"Redeem {amount} of {token} from {bank_app.app_name}")
+                bank_app.redeem(account, amount, token)
+            else:
+                logger.info(f"{account.address} does not have {token}")
         else:
             raise ConfigurationException("Unsupported action")
 
@@ -82,7 +85,8 @@ class Banking(Module):
             LayerBank,
             Aave,
             ZeroLend,
-            CompoundV3
+            CompoundV3,
+            Cog
         }
 
     def log(self):