eth-prototype 1.5.0b1__tar.gz → 1.5.0b3__tar.gz

{eth_prototype-1.5.0b1 → eth_prototype-1.5.0b3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eth-prototype
-Version: 1.5.0b1
+Version: 1.5.0b3
 Summary: Prototype Ethereum Smart Contracts in Python
 Home-page: https://github.com/gnarvaja/eth-prototype
 Author: Guillermo M. Narvaja

{eth_prototype-1.5.0b1 → eth_prototype-1.5.0b3}/src/eth_prototype.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eth-prototype
-Version: 1.5.0b1
+Version: 1.5.0b3
 Summary: Prototype Ethereum Smart Contracts in Python
 Home-page: https://github.com/gnarvaja/eth-prototype
 Author: Guillermo M. Narvaja

{eth_prototype-1.5.0b1 → eth_prototype-1.5.0b3}/src/ethproto/aa_bundler.py

@@ -13,6 +13,7 @@ from eth_account.messages import encode_defunct
 from eth_typing import HexAddress
 from eth_utils import add_0x_prefix, function_signature_to_4byte_selector
 from hexbytes import HexBytes
+from requests import HTTPError
 from web3 import Web3
 from web3.constants import ADDRESS_ZERO
 from web3.types import StateOverride, TxParams
@@ -197,7 +198,7 @@ class UserOperation:
             "maxFeePerGas": "0x%x" % self.max_fee_per_gas,
             "signature": add_0x_prefix(self.signature.hex()),
             "paymaster": self.paymaster,
-            "paymasterData": self.paymaster_data,
+            "paymasterData": self.paymaster_data.to_0x_hex(),
             "paymasterVerificationGasLimit": "0x%x" % self.paymaster_verification_gas_limit,
             "paymasterPostOpGasLimit": "0x%x" % self.paymaster_post_op_gas_limit,
         }
@@ -441,7 +442,7 @@ class Bundler:
     def estimate_user_operation_gas(self, user_operation: UserOperation) -> UserOpEstimation:
         resp = self.bundler.provider.make_request(
             "eth_estimateUserOperationGas",
-            [user_operation.as_reduced_dict(), self.entrypoint, self.overrides],
+            [user_operation.as_dict(), self.entrypoint, self.overrides],
         )
         if "error" in resp:
             raise BundlerRevertError(resp["error"]["message"], user_operation, resp)
@@ -461,23 +462,32 @@ class Bundler:
         )
 
     def alchemy_estimation(self, user_operation: UserOperation) -> AlchemyGasAndPaymasterAndData:
-        resp = self.bundler.provider.make_request(
-            "alchemy_requestGasAndPaymasterAndData",
-            [
-                {
-                    "policyId": self.alchemy_gas_policy_id,
-                    "entryPoint": self.entrypoint,
-                    "dummySignature": DUMMY_SIGNATURE,
-                    "userOperation": user_operation.as_reduced_dict(),
-                    "overrides": {
-                        "maxFeePerGas": {"multiplier": self.base_gas_price_factor},
-                        "maxPriorityFeePerGas": {"multiplier": self.priority_gas_price_factor},
-                        "callGasLimit": {"multiplier": self.gas_limit_factor},
-                        "verificationGasLimit": {"multiplier": self.verification_gas_factor},
-                    },
-                }
-            ],
-        )
+        try:
+            resp = self.bundler.provider.make_request(
+                "alchemy_requestGasAndPaymasterAndData",
+                [
+                    {
+                        "policyId": self.alchemy_gas_policy_id,
+                        "entryPoint": self.entrypoint,
+                        "dummySignature": DUMMY_SIGNATURE,
+                        "userOperation": user_operation.as_reduced_dict(),
+                        "overrides": {
+                            "maxFeePerGas": {"multiplier": self.base_gas_price_factor},
+                            "maxPriorityFeePerGas": {"multiplier": self.priority_gas_price_factor},
+                            "callGasLimit": {"multiplier": self.gas_limit_factor},
+                            "verificationGasLimit": {"multiplier": self.verification_gas_factor},
+                        },
+                        # Alchemy seems to be ignoring this, even though it's documented
+                        "stateOverrideSet": self.overrides,
+                    }
+                ],
+            )
+        except HTTPError as e:
+            raise BundlerRevertError(
+                f"HTTP error while requesting gas and paymaster data: {str(e)}",
+                userop=user_operation,
+                response=e.response.json(),
+            ) from e
 
         if "error" in resp:
             raise BundlerRevertError(resp["error"]["message"], userop=user_operation, response=resp)
@@ -516,6 +526,35 @@ class Bundler:
             paymaster_and_data=paymaster_and_data,
         )
 
+    def pimlico_gas_price(self):
+        resp = self.bundler.provider.make_request("pimlico_getUserOperationGasPrice", [])
+        if "error" in resp:
+            raise BundlerRevertError(resp["error"]["message"], response=resp)
+        # {
+        #   "jsonrpc": "2.0",
+        #   "id": 1,
+        #   "result": {
+        #           "slow": {
+        #           "maxFeePerGas": "0x829b42b5",
+        #           "maxPriorityFeePerGas": "0x829b42b5"
+        #       },
+        #           "standard": {
+        #           "maxFeePerGas": "0x88d36a75",
+        #           "maxPriorityFeePerGas": "0x88d36a75"
+        #       },
+        #           "fast": {
+        #           "maxFeePerGas": "0x8f0b9234",
+        #           "maxPriorityFeePerGas": "0x8f0b9234"
+        #       }
+        #   }
+        # }
+        priority_fee = int(resp["result"]["standard"]["maxPriorityFeePerGas"], 16)
+        total_fee = int(resp["result"]["standard"]["maxFeePerGas"], 16)
+        base_fee = total_fee - priority_fee
+        max_priority_fee_per_gas = int(priority_fee * self.priority_gas_price_factor)
+        max_fee_per_gas = min(max_priority_fee_per_gas + base_fee, self.max_fee_per_gas)
+        return GasPrice(max_priority_fee_per_gas=max_priority_fee_per_gas, max_fee_per_gas=max_fee_per_gas)
+
     def generic_gas_price(self):
         base_fee = self.get_base_fee()
         priority_fee = self.w3.eth.max_priority_fee
@@ -539,6 +578,15 @@ class Bundler:
                 estimation_and_paymaster.paymaster_and_data
             )
 
+        elif self.bundler_type == "pimlico":
+            estimation = self.estimate_user_operation_gas(user_operation)
+
+            user_operation = user_operation.add_estimation(estimation)
+
+            gas_price = self.pimlico_gas_price()
+
+            user_operation = user_operation.add_gas_price(gas_price)
+
         elif self.bundler_type == "generic":
             estimation = self.estimate_user_operation_gas(user_operation)