mirror.py connection fix

conf/mirrorprice/dexorder-mirrorprice.toml

@@ -1,7 +1,7 @@
 metadata='metadata.json'  # the Dockerfile will move metadata-finaldata.json into positon
 accounts = [
-    # dev account #6
+    # dev account #4
-    '0x92db14e403b83dfe3df233f83dfa3a0d7096f21ca9b0d6d6b8d88b2b4ec1564e', # 0x976EA74026E726554dB657fA54763abd0C3a0aa9
+    '0x7c852118294e51e653712a81e05800f419141751be58f605c371e15141b007a6', # 0x90F79bf6EB2c4f870365E785982E1f101E93b906
 ]
 rpc_url = '${rpc_urls.arbsep_alchemy}'
 mirror_source_rpc_url='${rpc_urls.arbsep_alchemy}'

src/dexorder/base/account.py

@@ -39,7 +39,7 @@ class Account (LocalAccount):
         MUST call account.release() after the transaction has completed, to return this Account to the available pool.
         """
         Account._init_pool()
-        log.debug(f'available accounts: {Account._pool.qsize()}')
+        # log.debug(f'available accounts: {Account._pool.qsize()}')
         try:
             async with asyncio.timeout(1):
                 result = await Account._pool.get()

src/dexorder/bin/mirror.py

@@ -144,35 +144,28 @@ async def main():
     tokens = set(i[1] for i in pool_infos).union(i[2] for i in pool_infos)
 
     log.debug(f'Mirroring tokens')
-    txs = []
     for t in tokens:
         # noinspection PyBroadException
         try:
             info = await get_token_info(t)
             # anvil had trouble estimating the gas, so we hardcode it.
             tx = await mirrorenv.transact.mirrorToken(info, gas=1_000_000)
-            txs.append(tx.wait())
+            await tx.wait()
         except Exception:
             log.exception(f'Failed to mirror token {t}')
             exit(1)
-    results = await asyncio.gather(*txs)
-    if any(result['status'] != 1 for result in results):
-        log.error('Mirroring a token reverted.')
-        exit(1)
     log.info(f'Tokens deployed')
 
     log.debug(f'Mirroring pools {", ".join(pools)}')
-    txs = []
     for pool, info in zip(pools, pool_infos):
         # noinspection PyBroadException
         try:
             # anvil had trouble estimating the gas, so we hardcode it.
             tx = await mirrorenv.transact.mirrorPool(info, gas=5_500_000)
+            await tx.wait()
         except Exception:
             log.exception(f'Failed to mirror pool {pool}')
             exit(1)
-        txs.append(tx.wait())
-    await asyncio.gather(*txs)
     log.info('Pools deployed')
 
     mirror_pool_list = []

src/dexorder/blockchain/connection.py

@@ -9,7 +9,6 @@ from aiohttp import ClientResponseError, ClientSession, ClientTimeout, TCPConnec
 from eth_typing import URI
 from hexbytes import HexBytes
 from web3 import WebsocketProviderV2, AsyncWeb3, AsyncHTTPProvider
-from web3.exceptions import Web3Exception
 from web3.middleware.signing import async_construct_sign_and_send_raw_middleware
 from web3.types import RPCEndpoint, RPCResponse
 
@@ -17,7 +16,6 @@ from .. import current_w3, Blockchain, config, Account, NARG
 from ..base.chain import current_chain
 from ..contract import get_contract_data
 
-
 log = logging.getLogger(__name__)
 
 
@@ -64,7 +62,8 @@ async def create_w3(rpc_url: Union[str,list[str]]=None, account=NARG, autosign=T
         http_provider = RetryHTTPProvider(url)
         await http_provider.cache_async_session(session)
         w3 = AsyncWeb3(http_provider)
-        w3.middleware_onion.add(archive_intercept_middleware, 'block_number_intercept_middleware')
+        if archive_urls:
+            w3.middleware_onion.add(archive_intercept_middleware, 'block_number_intercept_middleware')
         w3.middleware_onion.remove('attrdict')
         w3.middleware_onion.add(clean_input_async, 'clean_input')
         w3.eth.Contract = _make_contract(w3.eth)
@@ -86,7 +85,7 @@ async def create_w3(rpc_url: Union[str,list[str]]=None, account=NARG, autosign=T
     chain_ids = await asyncio.gather(*(w3.eth.chain_id for w3 in itertools.chain(w3_instances, archive_instances)))
     if len(set(chain_ids)) != 1:
         raise RuntimeError("All RPC URLs must belong to the same blockchain")
-        
+
     # noinspection PyTypeChecker
     return RoundRobinWebProxy(w3_instances, archive_instances) if len(w3_instances) > 1 or archive_instances else w3_instances[0]
 

src/dexorder/contract/contract_proxy.py

@@ -8,7 +8,6 @@ from web3.types import TxReceipt, TxData
 
 from dexorder import current_w3, Account
 from dexorder.blocks import current_block
-from dexorder.blockstate.fork import current_fork
 from dexorder.util import hexstr
 
 log = logging.getLogger(__name__)
@@ -19,10 +18,11 @@ class ContractTransaction:
         # This is the standard RPC transaction dictionary
         self.tx = tx
 
-        # These three fields are populated only after signing
+        # These fields are populated only after signing
         self.id_bytes: Optional[bytes] = None
         self.id: Optional[str] = None
         self.data: Optional[bytes] = None
+        self.account: Optional[Account] = None
 
         # This field is populated only after the transaction has been mined
         self.receipt: Optional[TxReceipt] = None  # todo could be multiple receipts for different branches!
@@ -33,6 +33,7 @@ class ContractTransaction:
     async def wait(self) -> TxReceipt:
         if self.receipt is None:
             self.receipt = await current_w3.get().eth.wait_for_transaction_receipt(self.id)
+            self.account.release()
         return self.receipt
 
     async def sign(self, account: Account):
@@ -42,6 +43,7 @@ class ContractTransaction:
         self.data = signed['rawTransaction']
         self.id_bytes = signed['hash']
         self.id = hexstr(self.id_bytes)
+        self.account = account
 
 
 class DeployTransaction (ContractTransaction):
@@ -74,13 +76,13 @@ def call_wrapper(addr, name, func):
 
 def transact_wrapper(addr, name, func):
     async def f(*args, **kwargs):
+        tx = await func(*args).build_transaction(kwargs)
+        ct = ContractTransaction(tx)
+        account = await Account.acquire()
+        if account is None:
+            raise ValueError(f'No account to sign transaction {addr}.{name}()')
+        await ct.sign(account)
         try:
-            tx = await func(*args).build_transaction(kwargs)
-            ct = ContractTransaction(tx)
-            account = Account.get()
-            if account is None:
-                raise ValueError(f'No account to sign transaction {addr}.{name}()')
-            await ct.sign(account)
             tx_id = await current_w3.get().eth.send_raw_transaction(ct.data)
             assert tx_id == ct.id_bytes
             return ct

src/dexorder/order/triggers.py

@@ -121,15 +121,15 @@ async def end_trigger_updates():
     PriceLineTrigger.end_updates(current_clock.get().timestamp)
     while _dirty:
         tk = _dirty.pop()
-        log.debug(f'check dirty tranche {tk}')
+        # log.debug(f'check dirty tranche {tk}')
         if _trigger_state.get(tk,0) == 0:
             # all clear for execution. add to active list with any necessary proofs
             active_tranches[tk] = PriceProof(0)
-            log.debug(f'active tranche {tk}')
+            # log.debug(f'active tranche {tk}')
         else:
             # blocked by one or more triggers being False (nonzero mask)
             reason = ', '.join(t.name for t in TrancheTrigger.all[tk].blocking_triggers)
-            log.debug(f'tranche {tk} blocked by {reason}')
+            # log.debug(f'tranche {tk} blocked by {reason}')
             # check expiry constraint
             try:
                 TrancheTrigger.all[tk].check_expire()
@@ -418,7 +418,7 @@ class PriceLineTrigger (Trigger):
         if self.inverted:
             price = 1/price
         self.last_price = price
-        log.debug(f'price trigger {price}')
+        # log.debug(f'price trigger {price}')
         if self not in PriceLineTrigger.triggers_set:
             self.add_computation(price)
         else:
@@ -449,8 +449,8 @@ class PriceLineTrigger (Trigger):
         line_value = m * time + b
         price_diff = sign * (y - line_value)
         activated = price_diff > 0
-        for price, line, s, a, diff in zip(y, line_value, sign, activated, price_diff):
+        # for price, line, s, a, diff in zip(y, line_value, sign, activated, price_diff):
-            log.debug(f'price: {line} {"<" if s == 1 else ">"} {price} {a} ({diff:+})')
+        #     log.debug(f'price: {line} {"<" if s == 1 else ">"} {price} {a} ({diff:+})')
         for t, activated in zip(PriceLineTrigger.triggers, activated):
             t.handle_result(activated)
         PriceLineTrigger.clear_data()