DanielVF · September 7, 2023 12:34
diff --git a/amo_balancing.ipynb b/amo_balancing.ipynb
diff --git a/amo_tools_eth_results.csv b/amo_tools_eth_results.csv
diff --git a/amo_tools_oeth_results.csv b/amo_tools_oeth_results.csv
diff --git a/brownie_fork.py b/brownie_fork.py
 import json
 import random
 import hexbytes
 from world import *
 import pandas as pd

 deploy = json.load(open("contracts/deployments/localhost/ConvexEthMetaStrategy.json"))
 strat = Contract.from_abi("new_oeth_strat", OETH_CONVEX_OETH_ETH_STRAT, deploy["abi"], owner=STRATEGIST, persist=False)
 pool = Contract.from_explorer("0x94b17476a93b3262d87b9a326965d1e91f9c13e7")
 zapper = Contract.from_explorer(OETH_ZAPPER)





 # Imbalance Pool
 # Pick too much OETH
 filename = "amo_tools_oeth_results.csv"
 zapper.deposit({'from': BINANCE, 'value': 10000e18})
 oeth.approve(pool, 1e70, {'from': BINANCE})
 pool.add_liquidity([0, 10000e18], 0, {'from': BINANCE})
 # Or too much ETH
 filename = "amo_tools_eth_results.csv"
 pool.add_liquidity([20000e18,0], 0, {'value': 20000e18, 'from': BINANCE})






 ## The goal is to test the new tools under a varity of randomized circumstances
 ZERO_HEX = hexbytes.HexBytes('0x0000000000000000000000000000000000000000000000000000000000000000')
 TRANSFER_HEX = hexbytes.HexBytes('0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef')
 STRAT_HEX = hexbytes.HexBytes('0x0000000000000000000000001827f9ea98e0bf96550b2fc20f7233277fcd7e63')


 eth_balance = pool.balances(0)
 oeth_balance = pool.balances(1)
 virtual_price = pool.get_virtual_price()

 NUM_RUNS = 400
 rows = []
 for i in range(0,NUM_RUNS):
    with TemporaryFork():
        correction_factor = random.random() * 1.10 - 0.05
        print(i, correction_factor)

        diff = oeth_balance - eth_balance
        negative_change_amount = int(diff * correction_factor)


        row = {}

        row['before_pool_eth'] = pool.balances(0)
        row['before_pool_oeth'] = pool.balances(1)
        row['before_vault_oeth'] = weth.balanceOf(oeth_vault_core)
        row['before_strat_balance'] = strat.checkBalance(weth)
        o_token_change = 0

        if negative_change_amount > 0:
            try:
                tx = strat.removeAndBurnOTokens(negative_change_amount * 1e18 / virtual_price)
                for x in tx.logs:
                        if (
                            len(x.topics) == 3
                            and x.address == oeth.address
                            and x.topics[0] == TRANSFER_HEX
                            and x.topics[1] == STRAT_HEX
                            and x.topics[2] == ZERO_HEX
                            ):
                            o_token_change = int(x.data, 16) * -1
                            break
            except:
                pass
        else:
            if random.randint(0,1) == 1:
                try:
                    tx = strat.mintAndAddOTokens(0-negative_change_amount)
                    for x in tx.logs:
                        if (
                            len(x.topics) == 3
                            and x.address == oeth.address
                            and x.topics[0] == TRANSFER_HEX
                            and x.topics[1] == ZERO_HEX
                            and x.topics[2] == STRAT_HEX
                            ):
                            o_token_change = int(x.data, 16)
                            break

                except:
                    pass
            else:
                try:
                    strat.removeOnlyAssets((0-negative_change_amount) * 1e18 / virtual_price)
                except:
                    pass

        row['after_pool_eth'] = pool.balances(0)
        row['after_pool_oeth'] = pool.balances(1)
        row['after_vault_oeth'] = weth.balanceOf(oeth_vault_core)
        row['after_strat_balance'] = strat.checkBalance(weth)
        row['o_token_change'] = o_token_change

        rows.append(row)


 print(pd.DataFrame.from_records(rows).to_csv(filename))
	import json
	import random
	import hexbytes
	from world import *
	import pandas as pd

	deploy = json.load(open("contracts/deployments/localhost/ConvexEthMetaStrategy.json"))
	strat = Contract.from_abi("new_oeth_strat", OETH_CONVEX_OETH_ETH_STRAT, deploy["abi"], owner=STRATEGIST, persist=False)
	pool = Contract.from_explorer("0x94b17476a93b3262d87b9a326965d1e91f9c13e7")
	zapper = Contract.from_explorer(OETH_ZAPPER)





	# Imbalance Pool
	# Pick too much OETH
	filename = "amo_tools_oeth_results.csv"
	zapper.deposit({'from': BINANCE, 'value': 10000e18})
	oeth.approve(pool, 1e70, {'from': BINANCE})
	pool.add_liquidity([0, 10000e18], 0, {'from': BINANCE})
	# Or too much ETH
	filename = "amo_tools_eth_results.csv"
	pool.add_liquidity([20000e18,0], 0, {'value': 20000e18, 'from': BINANCE})






	## The goal is to test the new tools under a varity of randomized circumstances
	ZERO_HEX = hexbytes.HexBytes('0x0000000000000000000000000000000000000000000000000000000000000000')
	TRANSFER_HEX = hexbytes.HexBytes('0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef')
	STRAT_HEX = hexbytes.HexBytes('0x0000000000000000000000001827f9ea98e0bf96550b2fc20f7233277fcd7e63')


	eth_balance = pool.balances(0)
	oeth_balance = pool.balances(1)
	virtual_price = pool.get_virtual_price()

	NUM_RUNS = 400
	rows = []
	for i in range(0,NUM_RUNS):
	with TemporaryFork():
	correction_factor = random.random() * 1.10 - 0.05
	print(i, correction_factor)

	diff = oeth_balance - eth_balance
	negative_change_amount = int(diff * correction_factor)


	row = {}

	row['before_pool_eth'] = pool.balances(0)
	row['before_pool_oeth'] = pool.balances(1)
	row['before_vault_oeth'] = weth.balanceOf(oeth_vault_core)
	row['before_strat_balance'] = strat.checkBalance(weth)
	o_token_change = 0

	if negative_change_amount > 0:
	try:
	tx = strat.removeAndBurnOTokens(negative_change_amount * 1e18 / virtual_price)
	for x in tx.logs:
	if (
	len(x.topics) == 3
	and x.address == oeth.address
	and x.topics[0] == TRANSFER_HEX
	and x.topics[1] == STRAT_HEX
	and x.topics[2] == ZERO_HEX
	):
	o_token_change = int(x.data, 16) * -1
	break
	except:
	pass
	else:
	if random.randint(0,1) == 1:
	try:
	tx = strat.mintAndAddOTokens(0-negative_change_amount)
	for x in tx.logs:
	if (
	len(x.topics) == 3
	and x.address == oeth.address
	and x.topics[0] == TRANSFER_HEX
	and x.topics[1] == ZERO_HEX
	and x.topics[2] == STRAT_HEX
	):
	o_token_change = int(x.data, 16)
	break

	except:
	pass
	else:
	try:
	strat.removeOnlyAssets((0-negative_change_amount) * 1e18 / virtual_price)
	except:
	pass

	row['after_pool_eth'] = pool.balances(0)
	row['after_pool_oeth'] = pool.balances(1)
	row['after_vault_oeth'] = weth.balanceOf(oeth_vault_core)
	row['after_strat_balance'] = strat.checkBalance(weth)
	row['o_token_change'] = o_token_change

	rows.append(row)


	print(pd.DataFrame.from_records(rows).to_csv(filename))