poly-web3 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

poly_web3/web3_service/base.py

@@ -1,143 +1,181 @@
-# -*- coding = utf-8 -*-
-# @Time: 2025-12-27 15:57:07
-# @Author: PinBar
-# @Site:
-# @File: base.py
-# @Software: PyCharm
-import requests
-from py_builder_relayer_client.client import RelayClient
-from py_clob_client.client import ClobClient
-from web3 import Web3
-
-from poly_web3.const import (
-    RPC_URL,
-    CTF_ADDRESS,
-    CTF_ABI_PAYOUT,
-    ZERO_BYTES32,
-    USDC_POLYGON,
-    CTF_ABI_REDEEM,
-    NEG_RISK_ADAPTER_ADDRESS,
-    RELAYER_URL,
-    POL,
-    AMOY,
-    GET_RELAY_PAYLOAD,
-    NEG_RISK_ADAPTER_ABI_REDEEM,
-)
-from poly_web3.schema import WalletType
-
-
-class BaseWeb3Service:
-    def __init__(
-        self,
-        clob_client: ClobClient = None,
-        relayer_client: RelayClient = None,
-    ):
-        self.relayer_client = relayer_client
-        self.clob_client: ClobClient = clob_client
-        if self.clob_client:
-            self.wallet_type: WalletType = WalletType.get_with_code(
-                self.clob_client.builder.sig_type
-            )
-        else:
-            self.wallet_type = WalletType.PROXY
-        self.w3: Web3 = Web3(Web3.HTTPProvider(RPC_URL))
-        if self.wallet_type == WalletType.PROXY and relayer_client is None:
-            raise Exception("relayer_client must be provided")
-
-    def is_condition_resolved(self, condition_id: str) -> bool:
-        ctf = self.w3.eth.contract(address=CTF_ADDRESS, abi=CTF_ABI_PAYOUT)
-        return ctf.functions.payoutDenominator(condition_id).call() > 0
-
-    def get_winning_indexes(self, condition_id: str) -> list[int]:
-        ctf = self.w3.eth.contract(address=CTF_ADDRESS, abi=CTF_ABI_PAYOUT)
-        if not self.is_condition_resolved(condition_id):
-            return []
-        outcome_count = ctf.functions.getOutcomeSlotCount(condition_id).call()
-        winners: list[int] = []
-        for i in range(outcome_count):
-            if ctf.functions.payoutNumerators(condition_id, i).call() > 0:
-                winners.append(i)
-        return winners
-
-    def get_redeemable_index_and_balance(
-        self, condition_id: str, owner: str
-    ) -> list[tuple]:
-        winners = self.get_winning_indexes(condition_id)
-        if not winners:
-            return []
-        ctf = self.w3.eth.contract(address=CTF_ADDRESS, abi=CTF_ABI_PAYOUT)
-        owner_checksum = Web3.to_checksum_address(owner)
-        redeemable: list[tuple] = []
-        for index in winners:
-            index_set = 1 << index
-            collection_id = ctf.functions.getCollectionId(
-                ZERO_BYTES32, condition_id, index_set
-            ).call()
-            position_id = ctf.functions.getPositionId(
-                USDC_POLYGON, collection_id
-            ).call()
-            balance = ctf.functions.balanceOf(owner_checksum, position_id).call()
-            if balance > 0:
-                redeemable.append((index, balance / 1000000))
-        return redeemable
-
-    def build_ctf_redeem_tx_data(self, condition_id: str) -> str:
-        ctf = self.w3.eth.contract(address=CTF_ADDRESS, abi=CTF_ABI_REDEEM)
-        # 只需要 calldata：encodeABI 即可
-        return ctf.functions.redeemPositions(
-            USDC_POLYGON,
-            ZERO_BYTES32,
-            condition_id,
-            [1, 2],
-        )._encode_transaction_data()
-
-    def build_neg_risk_redeem_tx_data(
-        self, condition_id: str, redeem_amounts: list[int]
-    ) -> str:
-        nr_adapter = self.w3.eth.contract(
-            address=NEG_RISK_ADAPTER_ADDRESS, abi=NEG_RISK_ADAPTER_ABI_REDEEM
-        )
-        return nr_adapter.functions.redeemPositions(
-            condition_id,
-            redeem_amounts,
-        )._encode_transaction_data()
-
-    @classmethod
-    def _get_relay_payload(cls, address: str, wallet_type: WalletType):
-        return requests.get(
-            RELAYER_URL + GET_RELAY_PAYLOAD,
-            params={"address": address, "type": wallet_type},
-        ).json()
-
-    def get_contract_config(self) -> dict:
-        if self.clob_client.chain_id == 137:
-            return POL
-        elif self.clob_client.chain_id == 80002:
-            return AMOY
-        raise Exception("Invalid network")
-
-    def estimate_gas(self, tx):
-        payload = {
-            "jsonrpc": "2.0",
-            "method": "eth_estimateGas",
-            "params": [tx],
-            "id": 1,
-        }
-
-        response = requests.post(RPC_URL, json=payload)
-        result = response.json()
-
-        if "result" in result:
-            # 返回的是16进制 gas 数量
-            gas_hex = result["result"]
-            return str(int(gas_hex, 16))
-        else:
-            raise Exception("Estimate gas error: " + str(result))
-
-    def redeem(
-        self,
-        condition_id: str,
-        neg_risk: bool = False,
-        redeem_amounts: list[int] | None = None,
-    ):  # noqa:
-        raise ImportError()
+# -*- coding = utf-8 -*-
+# @Time: 2025-12-27 15:57:07
+# @Author: PinBar
+# @Site:
+# @File: base.py
+# @Software: PyCharm
+import requests
+from py_builder_relayer_client.client import RelayClient
+from py_clob_client.client import ClobClient
+from web3 import Web3
+
+from poly_web3.const import (
+    RPC_URL,
+    CTF_ADDRESS,
+    CTF_ABI_PAYOUT,
+    ZERO_BYTES32,
+    USDC_POLYGON,
+    CTF_ABI_REDEEM,
+    NEG_RISK_ADAPTER_ADDRESS,
+    RELAYER_URL,
+    POL,
+    AMOY,
+    GET_RELAY_PAYLOAD,
+    NEG_RISK_ADAPTER_ABI_REDEEM,
+)
+from poly_web3.schema import WalletType
+
+
+class BaseWeb3Service:
+    def __init__(
+            self,
+            clob_client: ClobClient = None,
+            relayer_client: RelayClient = None,
+            rpc_url: str | None = None,
+    ):
+        self.relayer_client = relayer_client
+        self.clob_client: ClobClient = clob_client
+        if self.clob_client:
+            self.wallet_type: WalletType = WalletType.get_with_code(
+                self.clob_client.builder.sig_type
+            )
+        else:
+            self.wallet_type = WalletType.PROXY
+        self.rpc_url = rpc_url or RPC_URL
+        self.w3: Web3 = Web3(Web3.HTTPProvider(self.rpc_url))
+        if self.wallet_type == WalletType.PROXY and relayer_client is None:
+            raise Exception("relayer_client must be provided")
+
+    def _resolve_user_address(self):
+        funder = getattr(getattr(self.clob_client, "builder", None), "funder", None)
+        if funder:
+            return funder
+        return self.clob_client.get_address()
+
+    @classmethod
+    def fetch_positions(cls, user_address: str) -> list[dict]:
+        """
+        Fetches current positions for a user from the official Polymarket API.
+
+        :param user_address: User wallet address (0x-prefixed, 40 hex chars)
+        :return: List of position dictionaries from the API
+        """
+        url = "https://data-api.polymarket.com/positions"
+        params = {
+            "user": user_address,
+            "sizeThreshold": 1,
+            "limit": 100,
+            "redeemable": True,
+            "sortBy": "RESOLVING",
+            "sortDirection": "DESC",
+        }
+        try:
+            response = requests.get(url, params=params)
+            response.raise_for_status()
+            positions = response.json()
+            return [i for i in positions if i.get("percentPnl") > 0]
+        except Exception as e:
+            print(f"Failed to fetch positions from API: {e}")
+            return []
+
+    def is_condition_resolved(self, condition_id: str) -> bool:
+        ctf = self.w3.eth.contract(address=CTF_ADDRESS, abi=CTF_ABI_PAYOUT)
+        return ctf.functions.payoutDenominator(condition_id).call() > 0
+
+    def get_winning_indexes(self, condition_id: str) -> list[int]:
+        ctf = self.w3.eth.contract(address=CTF_ADDRESS, abi=CTF_ABI_PAYOUT)
+        if not self.is_condition_resolved(condition_id):
+            return []
+        outcome_count = ctf.functions.getOutcomeSlotCount(condition_id).call()
+        winners: list[int] = []
+        for i in range(outcome_count):
+            if ctf.functions.payoutNumerators(condition_id, i).call() > 0:
+                winners.append(i)
+        return winners
+
+    def get_redeemable_index_and_balance(
+            self, condition_id: str
+    ) -> list[tuple]:
+        owner = self._resolve_user_address()
+        winners = self.get_winning_indexes(condition_id)
+        if not winners:
+            return []
+        ctf = self.w3.eth.contract(address=CTF_ADDRESS, abi=CTF_ABI_PAYOUT)
+        owner_checksum = Web3.to_checksum_address(owner)
+        redeemable: list[tuple] = []
+        for index in winners:
+            index_set = 1 << index
+            collection_id = ctf.functions.getCollectionId(
+                ZERO_BYTES32, condition_id, index_set
+            ).call()
+            position_id = ctf.functions.getPositionId(
+                USDC_POLYGON, collection_id
+            ).call()
+            balance = ctf.functions.balanceOf(owner_checksum, position_id).call()
+            if balance > 0:
+                redeemable.append((index, balance / 1000000))
+        return redeemable
+
+    def build_ctf_redeem_tx_data(self, condition_id: str) -> str:
+        ctf = self.w3.eth.contract(address=CTF_ADDRESS, abi=CTF_ABI_REDEEM)
+        # 只需要 calldata：encodeABI 即可
+        return ctf.functions.redeemPositions(
+            USDC_POLYGON,
+            ZERO_BYTES32,
+            condition_id,
+            [1, 2],
+        )._encode_transaction_data()
+
+    def build_neg_risk_redeem_tx_data(
+            self, condition_id: str, redeem_amounts: list[int]
+    ) -> str:
+        nr_adapter = self.w3.eth.contract(
+            address=NEG_RISK_ADAPTER_ADDRESS, abi=NEG_RISK_ADAPTER_ABI_REDEEM
+        )
+        return nr_adapter.functions.redeemPositions(
+            condition_id,
+            redeem_amounts,
+        )._encode_transaction_data()
+
+    @classmethod
+    def _get_relay_payload(cls, address: str, wallet_type: WalletType):
+        return requests.get(
+            RELAYER_URL + GET_RELAY_PAYLOAD,
+            params={"address": address, "type": wallet_type},
+        ).json()
+
+    def get_contract_config(self) -> dict:
+        if self.clob_client.chain_id == 137:
+            return POL
+        elif self.clob_client.chain_id == 80002:
+            return AMOY
+        raise Exception("Invalid network")
+
+    def estimate_gas(self, tx):
+        payload = {
+            "jsonrpc": "2.0",
+            "method": "eth_estimateGas",
+            "params": [tx],
+            "id": 1,
+        }
+
+        response = requests.post(self.rpc_url, json=payload)
+        result = response.json()
+
+        if "result" in result:
+            # 返回的是16进制 gas 数量
+            gas_hex = result["result"]
+            return str(int(gas_hex, 16))
+        else:
+            raise Exception("Estimate gas error: " + str(result))
+
+    def redeem(
+            self,
+            condition_id: str,
+            neg_risk: bool = False,
+            redeem_amounts: list[int] | None = None,
+    ):  # noqa:
+        raise ImportError()
+
+    def redeem_all(self) -> list[dict] | None:
+        raise ImportError()

poly_web3/web3_service/eoa_service.py

@@ -1,17 +1,17 @@
-# -*- coding = utf-8 -*-
-# @Time: 2025-12-27 16:01:09
-# @Author: PinBar
-# @Site:
-# @File: eoa_service.py
-# @Software: PyCharm
-from poly_web3.web3_service.base import BaseWeb3Service
-
-
-class EOAWeb3Service(BaseWeb3Service):
-    def redeem(
-        self,
-        condition_id: str,
-        neg_risk: bool = False,
-        redeem_amounts: list[int] | None = None,
-    ):
-        raise ImportError("EOA wallet redeem not supported")
+# -*- coding = utf-8 -*-
+# @Time: 2025-12-27 16:01:09
+# @Author: PinBar
+# @Site:
+# @File: eoa_service.py
+# @Software: PyCharm
+from poly_web3.web3_service.base import BaseWeb3Service
+
+
+class EOAWeb3Service(BaseWeb3Service):
+    def redeem(
+        self,
+        condition_id: str,
+        neg_risk: bool = False,
+        redeem_amounts: list[int] | None = None,
+    ):
+        raise ImportError("EOA wallet redeem not supported")

poly_web3/web3_service/proxy_service.py

@@ -1,135 +1,163 @@
-# -*- coding = utf-8 -*-
-# @Time: 2025-12-27 16:00:42
-# @Author: PinBar
-# @Site:
-# @File: proxy_service.py
-# @Software: PyCharm
-import requests
-
-from web3 import Web3
-from eth_utils import to_bytes
-
-from poly_web3.const import (
-    proxy_wallet_factory_abi,
-    CTF_ADDRESS,
-    RELAYER_URL,
-    PROXY_INIT_CODE_HASH,
-    SUBMIT_TRANSACTION,
-    STATE_MINED,
-    STATE_CONFIRMED,
-    STATE_FAILED,
-    NEG_RISK_ADAPTER_ADDRESS,
-)
-from poly_web3.web3_service.base import BaseWeb3Service
-from poly_web3.signature.build import derive_proxy_wallet, create_struct_hash
-from poly_web3.signature.hash_message import hash_message
-from poly_web3.signature import secp256k1
-
-
-class ProxyWeb3Service(BaseWeb3Service):
-    def build_proxy_transaction_request(self, args: dict) -> dict:
-        proxy_contract_config = self.get_contract_config()["ProxyContracts"]
-        to = proxy_contract_config["ProxyFactory"]
-        proxy = derive_proxy_wallet(args["from"], to, PROXY_INIT_CODE_HASH)
-        relayer_fee = "0"
-        relay_hub = proxy_contract_config["RelayHub"]
-        gas_limit_str = self.estimate_gas(
-            tx={"from": args["from"], "to": to, "data": args["data"]}
-        )
-        sig_params = {
-            "gasPrice": args.get("gasPrice"),
-            "gasLimit": gas_limit_str,
-            "relayerFee": relayer_fee,
-            "relayHub": relay_hub,
-            "relay": args.get("relay"),
-        }
-        tx_hash = create_struct_hash(
-            args["from"],
-            to,
-            args["data"],
-            relayer_fee,
-            args.get("gasPrice"),
-            gas_limit_str,
-            args["nonce"],
-            relay_hub,
-            args.get("relay"),
-        )
-        message = {"raw": list(to_bytes(hexstr=tx_hash))}
-
-        r, s, recovery = secp256k1.sign(
-            hash_message(message)[2:], self.clob_client.signer.private_key
-        )
-        signature = {
-            "r": secp256k1.int_to_hex(r, 32),
-            "s": secp256k1.int_to_hex(s, 32),
-            "v": 28 if recovery else 27,
-            "yParity": recovery,
-        }
-        final_sig = secp256k1.serialize_signature(**signature, to="hex")
-        req = {
-            "from": args["from"],
-            "to": to,
-            "proxyWallet": proxy,
-            "data": args["data"],
-            "nonce": args["nonce"],
-            "signature": final_sig,
-            "signatureParams": sig_params,
-            "type": self.wallet_type.value,
-            "metadata": "redeem",
-        }
-        return req
-
-    def encode_proxy_transaction_data(self, txns):
-        # Prepare the arguments for the 'proxy' function
-        calls_data = [
-            (txn["typeCode"], txn["to"], txn["value"], txn["data"]) for txn in txns
-        ]
-
-        # Create the contract object
-        contract = self.w3.eth.contract(abi=proxy_wallet_factory_abi)
-
-        # Encode function data
-        function_data = contract.encodeABI(fn_name="proxy", args=[calls_data])
-
-        return function_data
-
-    def redeem(
-        self,
-        condition_id: str,
-        neg_risk: bool = False,
-        redeem_amounts: list[int] | None = None,
-    ):
-        if neg_risk:
-            if redeem_amounts is None or len(redeem_amounts) != 2:
-                raise Exception("negRisk redeem requires redeem_amounts with length 2")
-            tx_data = self.build_neg_risk_redeem_tx_data(condition_id, redeem_amounts)
-            tx_to = NEG_RISK_ADAPTER_ADDRESS
-        else:
-            tx_data = self.build_ctf_redeem_tx_data(condition_id)
-            tx_to = CTF_ADDRESS
-        tx = {"to": tx_to, "data": tx_data, "value": 0, "typeCode": 1}
-        if self.clob_client is None:
-            raise Exception("signer not found")
-        _from = Web3.to_checksum_address(self.clob_client.get_address())
-        rp = self._get_relay_payload(_from, self.wallet_type)
-        args = {
-            "from": _from,
-            "gasPrice": "0",
-            "data": self.encode_proxy_transaction_data([tx]),
-            "relay": rp["address"],
-            "nonce": rp["nonce"],
-        }
-        req = self.build_proxy_transaction_request(args)
-        headers = self.relayer_client._generate_builder_headers(
-            "POST", SUBMIT_TRANSACTION, req
-        )
-        response = requests.post(
-            RELAYER_URL + SUBMIT_TRANSACTION, json=req, headers=headers
-        ).json()
-        redeem_res = self.relayer_client.poll_until_state(
-            transaction_id=response["transactionID"],
-            states=[STATE_MINED, STATE_CONFIRMED],
-            fail_state=STATE_FAILED,
-            max_polls=100,
-        )
-        return redeem_res
+# -*- coding = utf-8 -*-
+# @Time: 2025-12-27 16:00:42
+# @Author: PinBar
+# @Site:
+# @File: proxy_service.py
+# @Software: PyCharm
+import requests
+
+from web3 import Web3
+from eth_utils import to_bytes
+
+from poly_web3.const import (
+    proxy_wallet_factory_abi,
+    CTF_ADDRESS,
+    RELAYER_URL,
+    PROXY_INIT_CODE_HASH,
+    SUBMIT_TRANSACTION,
+    STATE_MINED,
+    STATE_CONFIRMED,
+    STATE_FAILED,
+    NEG_RISK_ADAPTER_ADDRESS,
+)
+from poly_web3.web3_service.base import BaseWeb3Service
+from poly_web3.signature.build import derive_proxy_wallet, create_struct_hash
+from poly_web3.signature.hash_message import hash_message
+from poly_web3.signature import secp256k1
+
+
+class ProxyWeb3Service(BaseWeb3Service):
+    def build_proxy_transaction_request(self, args: dict) -> dict:
+        proxy_contract_config = self.get_contract_config()["ProxyContracts"]
+        to = proxy_contract_config["ProxyFactory"]
+        proxy = derive_proxy_wallet(args["from"], to, PROXY_INIT_CODE_HASH)
+        relayer_fee = "0"
+        relay_hub = proxy_contract_config["RelayHub"]
+        gas_limit_str = self.estimate_gas(
+            tx={"from": args["from"], "to": to, "data": args["data"]}
+        )
+        sig_params = {
+            "gasPrice": args.get("gasPrice"),
+            "gasLimit": gas_limit_str,
+            "relayerFee": relayer_fee,
+            "relayHub": relay_hub,
+            "relay": args.get("relay"),
+        }
+        tx_hash = create_struct_hash(
+            args["from"],
+            to,
+            args["data"],
+            relayer_fee,
+            args.get("gasPrice"),
+            gas_limit_str,
+            args["nonce"],
+            relay_hub,
+            args.get("relay"),
+        )
+        message = {"raw": list(to_bytes(hexstr=tx_hash))}
+
+        r, s, recovery = secp256k1.sign(
+            hash_message(message)[2:], self.clob_client.signer.private_key
+        )
+        signature = {
+            "r": secp256k1.int_to_hex(r, 32),
+            "s": secp256k1.int_to_hex(s, 32),
+            "v": 28 if recovery else 27,
+            "yParity": recovery,
+        }
+        final_sig = secp256k1.serialize_signature(**signature, to="hex")
+        req = {
+            "from": args["from"],
+            "to": to,
+            "proxyWallet": proxy,
+            "data": args["data"],
+            "nonce": args["nonce"],
+            "signature": final_sig,
+            "signatureParams": sig_params,
+            "type": self.wallet_type.value,
+            "metadata": "redeem",
+        }
+        return req
+
+    def encode_proxy_transaction_data(self, txns):
+        # Prepare the arguments for the 'proxy' function
+        calls_data = [
+            (txn["typeCode"], txn["to"], txn["value"], txn["data"]) for txn in txns
+        ]
+
+        # Create the contract object
+        contract = self.w3.eth.contract(abi=proxy_wallet_factory_abi)
+
+        # Encode function data
+        function_data = contract.encodeABI(fn_name="proxy", args=[calls_data])
+
+        return function_data
+
+    def redeem(
+            self,
+            condition_id: str,
+            neg_risk: bool = False,
+            redeem_amounts: list[int] | None = None,
+    ):
+        if neg_risk:
+            if redeem_amounts is None or len(redeem_amounts) != 2:
+                raise Exception("negRisk redeem requires redeem_amounts with length 2")
+            tx_data = self.build_neg_risk_redeem_tx_data(condition_id, redeem_amounts)
+            tx_to = NEG_RISK_ADAPTER_ADDRESS
+        else:
+            tx_data = self.build_ctf_redeem_tx_data(condition_id)
+            tx_to = CTF_ADDRESS
+        tx = {"to": tx_to, "data": tx_data, "value": 0, "typeCode": 1}
+        if self.clob_client is None:
+            raise Exception("signer not found")
+        _from = Web3.to_checksum_address(self.clob_client.get_address())
+        rp = self._get_relay_payload(_from, self.wallet_type)
+        args = {
+            "from": _from,
+            "gasPrice": "0",
+            "data": self.encode_proxy_transaction_data([tx]),
+            "relay": rp["address"],
+            "nonce": rp["nonce"],
+        }
+        req = self.build_proxy_transaction_request(args)
+        headers = self.relayer_client._generate_builder_headers(
+            "POST", SUBMIT_TRANSACTION, req
+        )
+        response = requests.post(
+            RELAYER_URL + SUBMIT_TRANSACTION, json=req, headers=headers
+        ).json()
+        redeem_res = self.relayer_client.poll_until_state(
+            transaction_id=response["transactionID"],
+            states=[STATE_MINED, STATE_CONFIRMED],
+            fail_state=STATE_FAILED,
+            max_polls=100,
+        )
+        return redeem_res
+
+    def redeem_all(self) -> list[dict]:
+        positions = self.fetch_positions(user_address=self._resolve_user_address())
+        if not positions:
+            return []
+        redeem_list = []
+        for pos in positions:
+            condition_id = pos.get("conditionId")
+            try:
+                can_redeem = self.get_redeemable_index_and_balance(condition_id)
+                if not can_redeem:
+                    continue
+                if pos.get("negativeRisk"):
+                    amounts = [0, 0]
+                    amounts[pos.get("outcomeIndex")] = pos.get("size")
+                    int_amounts = [int(amount * 1e6) for amount in amounts]
+                    redeem_res = self.redeem(condition_id=condition_id, redeem_amounts=int_amounts, neg_risk=True)
+                else:
+                    redeem_res = self.redeem(condition_id=condition_id)
+            except Exception as e:
+                print(f"redeem error, {condition_id=}, error={e}")
+            else:
+                redeem_list.append(redeem_res)
+                buy_price = pos.get("avgPrice")
+                size = pos.get("size")
+                volume = 1 / buy_price * (buy_price * size)
+                print(f"{pos.get('slug')} redeem success, volume={volume:.4f} usdc")
+        return redeem_list