web3-wizzard-lib 1.7.1__py3-none-any.whl → 1.9.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/contract/compound_v3.py

@@ -1,4 +1,5 @@
 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/compound_v3.json")
@@ -8,6 +9,18 @@ class CompoundV3Contract(Contract):
     def __init__(self, contract_address, web3):
         super().__init__(contract_address, web3, abi)
 
+    def borrow_balance_of(self, account):
+        return self.contract.functions.borrowBalanceOf(account.address).call()
+
+    @evm_transaction
+    def supply(self, account, amount, token_address):
+        txn_params = self.build_generic_data(account.address, set_contract_address=False)
+
+        return self.contract.functions.supply(
+            token_address,
+            amount
+        ).build_transaction(txn_params)
+
     def user_collateral(self, account, token_address):
         return self.contract.functions.userCollateral(
             account.address,

web3_wizzard_lib/core/modules/bank/bank.py

@@ -5,6 +5,9 @@ class Bank:
     def borrow(self, account, amount):
         pass
 
+    def get_repay_borrow_amount(self, account):
+        pass
+
     def repay_borrow(self, account, amount):
         pass
 

web3_wizzard_lib/core/modules/bank/cog_bank.py

@@ -0,0 +1,23 @@
+from sybil_engine.data.contracts import get_contracts_for_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):
+        self.contract_address = get_contracts_for_chain(chain)[self.app_name]
+        self.contract = CogErc20Contract(self.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):
+        return self.contract.balance_of(account)

web3_wizzard_lib/core/modules/bank/compound_v3.py

@@ -1,6 +1,7 @@
 from sybil_engine.data.contracts import get_contracts_for_chain
 from sybil_engine.data.networks import get_ids_chain
 from sybil_engine.data.tokens import get_tokens_for_chain
+from sybil_engine.domain.balance.tokens import Erc20Token
 from sybil_engine.utils.utils import ConfigurationException
 
 from web3_wizzard_lib.core.contract.compound_v3 import CompoundV3Contract
@@ -21,15 +22,26 @@ class CompoundV3(Bank):
     def supply(self, account, amount):
         raise ConfigurationException("Only redeem supported for Compound V3")
 
+    def get_repay_borrow_amount(self, account):
+        return self.contract.borrow_balance_of(account)
+
+    def repay_borrow(self, account, amount):
+        erc20_token = Erc20Token(self.contract.chain_instance['chain'], "USDC", self.contract.web3)
+
+        if erc20_token.allowance(account, self.contract.contract_address) < 100:
+            erc20_token.approve(account, self.contract.contract_address)
+
+        return self.contract.supply(account, amount, erc20_token.erc20_contract.contract_address)
+
     def redeem(self, account, amount, token):
-        redeem_action = '0x414354494f4e5f57495448445241575f4e41544956455f544f4b454e00000000'
-        self.contract_bulker.invoke(account, amount, redeem_action)
+        actions = '0x414354494f4e5f57495448445241575f41535345540000000000000000000000'
+        self.contract_bulker.invoke(account, amount, actions)
 
     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,18 +39,23 @@ 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':
             bank_app.borrow(account, amount)
         elif action == 'REPAY':
-            bank_app.repay_borrow(account, amount)
+            amount = bank_app.get_repay_borrow_amount(account)
+
+            if amount > 0:
+                logger.info(f"Repay {amount} {token} in {bank_app.app_name}")
+                bank_app.repay_borrow(account, amount)
+            else:
+                logger.info(f"{token} borrow balance is 0")
         elif action == 'REDEEM':
             amount = bank_app.get_deposit_amount(account, token)
 
@@ -57,7 +63,7 @@ class Banking(Module):
                 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}")
+                logger.info(f"{token} balance is 0")
         else:
             raise ConfigurationException("Unsupported action")
 
@@ -85,7 +91,8 @@ class Banking(Module):
             LayerBank,
             Aave,
             ZeroLend,
-            CompoundV3
+            CompoundV3,
+            Cog
         }
 
     def log(self):