polynode 0.7.0__py3-none-any.whl → 0.7.1__py3-none-any.whl

polynode/trading/constants.py

@@ -39,5 +39,8 @@ POLY_USD = "0xc011a7e12a19f7b1f670d46f03b03f3342e82dfb"
 COLLATERAL_ONRAMP = "0x93070a847efef7f70739046a929d47a521f5b8ee"
 COLLATERAL_OFFRAMP = "0x2957922eb93258b93368531d39facca3b4dc5854"
 
+# Fee Escrow
+FEE_ESCROW_ADDRESS = "0xa11D28433B79D0A88F3119b16A090075752258EA"
+
 # Metadata cache TTL (5 minutes)
 META_TTL_SECONDS = 300

polynode/trading/escrow.py

@@ -0,0 +1,166 @@
+"""Fee Escrow — EIP-712 signing, fee calculation, and nonce fetching
+for the on-chain FeeEscrow contract.
+
+The escrow is completely optional. When fee_bps=0, none of this code runs.
+"""
+
+from __future__ import annotations
+
+import math
+import os
+import secrets
+from dataclasses import dataclass
+from typing import Any
+
+import httpx
+
+from .constants import CHAIN_ID, FEE_ESCROW_ADDRESS
+from .types import Eip712Payload
+
+# ── EIP-712 Types ──
+
+FEE_AUTH_DOMAIN = {
+    "name": "PolyNodeFeeEscrow",
+    "version": "1",
+    "chainId": CHAIN_ID,
+    "verifyingContract": FEE_ESCROW_ADDRESS,
+}
+
+FEE_AUTH_TYPES = {
+    "FeeAuth": [
+        {"name": "orderId", "type": "bytes32"},
+        {"name": "payer", "type": "address"},
+        {"name": "signer", "type": "address"},
+        {"name": "feeAmount", "type": "uint256"},
+        {"name": "deadline", "type": "uint256"},
+        {"name": "nonce", "type": "uint256"},
+    ],
+}
+
+ZERO_ADDRESS = "0x0000000000000000000000000000000000000000"
+
+
+# ── Fee Calculation ──
+
+def calculate_fee(price: float, size: float, fee_bps: int) -> int:
+    """Calculate fee amount in raw USDC (6 decimals).
+
+    Args:
+        price: Order price (e.g. 0.55)
+        size: Order size in tokens (e.g. 100)
+        fee_bps: Fee in basis points (e.g. 50 = 0.5%)
+
+    Returns:
+        Fee amount as int in raw USDC units
+    """
+    notional_raw = math.floor(price * size * 1e6)
+    fee = math.floor(notional_raw * fee_bps / 10000)
+    return max(fee, 0)
+
+
+# ── Order ID Generation ──
+
+def generate_escrow_order_id() -> str:
+    """Generate a random bytes32 escrow order ID (0x-prefixed hex)."""
+    return "0x" + secrets.token_hex(32)
+
+
+# ── Nonce Fetching ──
+
+async def fetch_escrow_nonce(rpc_url: str, signer_address: str) -> int:
+    """Fetch the current nonce for a signer from the FeeEscrow contract via eth_call."""
+    # getNonce(address) selector = 0x2d0335ab
+    addr = signer_address.lower().replace("0x", "").zfill(64)
+    data = "0x2d0335ab" + addr
+
+    async with httpx.AsyncClient() as client:
+        resp = await client.post(
+            rpc_url,
+            json={
+                "jsonrpc": "2.0",
+                "id": 1,
+                "method": "eth_call",
+                "params": [{"to": FEE_ESCROW_ADDRESS, "data": data}, "latest"],
+            },
+        )
+        result = resp.json()
+
+    if "error" in result:
+        raise RuntimeError(f"Failed to fetch escrow nonce: {result['error']}")
+    return int(result.get("result", "0x0"), 16)
+
+
+# ── EIP-712 Signing ──
+
+@dataclass
+class FeeAuthRequest:
+    escrow_order_id: str
+    payer: str
+    signer: str
+    fee_amount: str
+    deadline: int
+    nonce: int
+    signature: str
+    affiliate: str
+    affiliate_share_bps: int
+
+
+async def sign_fee_auth(
+    sign_typed_data: Any,
+    *,
+    escrow_order_id: str,
+    payer: str,
+    signer_address: str,
+    fee_amount: int,
+    deadline: int,
+    nonce: int,
+    affiliate: str | None = None,
+    affiliate_share_bps: int | None = None,
+) -> FeeAuthRequest:
+    """Sign a FeeAuth EIP-712 message.
+
+    Args:
+        sign_typed_data: Callable that signs an Eip712Payload and returns a hex signature.
+        escrow_order_id: Random bytes32 hex string.
+        payer: Safe wallet address (where USDC is pulled from).
+        signer_address: EOA address (signs the message).
+        fee_amount: Fee in raw USDC (6 decimals).
+        deadline: Unix timestamp — authorization expires after this.
+        nonce: Signer's current nonce from the escrow contract.
+        affiliate: Partner wallet address.
+        affiliate_share_bps: Partner's share of fee in bps.
+
+    Returns:
+        FeeAuthRequest ready for the cosigner.
+    """
+    message = {
+        "orderId": escrow_order_id,
+        "payer": payer,
+        "signer": signer_address,
+        "feeAmount": str(fee_amount),
+        "deadline": str(deadline),
+        "nonce": str(nonce),
+    }
+
+    payload = Eip712Payload(
+        domain=FEE_AUTH_DOMAIN,
+        types=FEE_AUTH_TYPES,
+        primary_type="FeeAuth",
+        message=message,
+    )
+
+    signature = await sign_typed_data(payload)
+    if not signature.startswith("0x"):
+        signature = f"0x{signature}"
+
+    return FeeAuthRequest(
+        escrow_order_id=escrow_order_id,
+        payer=payer,
+        signer=signer_address,
+        fee_amount=str(fee_amount),
+        deadline=deadline,
+        nonce=nonce,
+        signature=signature,
+        affiliate=affiliate or ZERO_ADDRESS,
+        affiliate_share_bps=affiliate_share_bps or 0,
+    )

polynode/trading/onboarding.py

@@ -28,6 +28,7 @@ from .constants import (
     SPENDERS,
     USDC,
     V2_SPENDERS,
+    FEE_ESCROW_ADDRESS,
 )
 from .types import ApprovalStatus, BalanceInfo, ExchangeVersion, SignatureType
 
@@ -185,6 +186,11 @@ async def check_approvals(
         approved = ctf_contract.functions.isApprovedForAll(funder, spender_addr).call()
         ctf_status[name] = approved
 
+    # Fee Escrow USDC approval (always USDC.e regardless of exchange version)
+    usdc_contract = w3.eth.contract(address=Web3.to_checksum_address(USDC), abi=erc20_abi)
+    escrow_allowance = usdc_contract.functions.allowance(funder, Web3.to_checksum_address(FEE_ESCROW_ADDRESS)).call()
+    usdc_status["fee_escrow"] = escrow_allowance >= threshold
+
     all_approved = all(usdc_status.values()) and all(ctf_status.values())
 
     return ApprovalStatus(
@@ -264,6 +270,17 @@ async def set_approvals(
         tx_hashes.append(tx_hash.hex())
         nonce += 1
 
+    # Fee Escrow USDC approval (always USDC.e regardless of exchange version)
+    escrow_contract = w3.eth.contract(address=Web3.to_checksum_address(USDC), abi=erc20_abi)
+    tx = escrow_contract.functions.approve(Web3.to_checksum_address(FEE_ESCROW_ADDRESS), max_uint).build_transaction({
+        "from": account.address,
+        "nonce": nonce,
+        "chainId": CHAIN_ID,
+    })
+    signed = account.sign_transaction(tx)
+    tx_hash = w3.eth.send_raw_transaction(signed.raw_transaction)
+    tx_hashes.append(tx_hash.hex())
+
     return tx_hashes
 
 

polynode/trading/sqlite_backend.py

@@ -13,11 +13,14 @@ from .types import HistoryParams, MarketMeta, OrderHistoryRow, SignatureType, St
 class TradingSqliteBackend:
     """Local SQLite storage for trading module."""
 
+    SCHEMA_VERSION = 2
+
     def __init__(self, db_path: str) -> None:
         self._db = sqlite3.connect(db_path)
         self._db.execute("PRAGMA journal_mode=WAL")
         self._db.execute("PRAGMA synchronous=NORMAL")
         self._init_schema()
+        self._migrate()
 
     def _init_schema(self) -> None:
         self._db.executescript("""
@@ -54,13 +57,27 @@ class TradingSqliteBackend:
                 status TEXT NOT NULL DEFAULT 'submitting',
                 error_msg TEXT,
                 response_json TEXT,
-                created_at REAL NOT NULL
+                created_at REAL NOT NULL,
+                fee_amount_raw TEXT,
+                escrow_order_id TEXT,
+                fee_escrow_tx_hash TEXT
             );
 
             CREATE INDEX IF NOT EXISTS idx_order_history_wallet ON order_history(wallet_address);
             CREATE INDEX IF NOT EXISTS idx_order_history_token ON order_history(token_id);
         """)
 
+    def _migrate(self) -> None:
+        """Run schema migrations for existing databases."""
+        # Check if fee escrow columns exist (v2 migration)
+        cols = {row[1] for row in self._db.execute("PRAGMA table_info(order_history)").fetchall()}
+        if "fee_amount_raw" not in cols:
+            self._db.executescript("""
+                ALTER TABLE order_history ADD COLUMN fee_amount_raw TEXT;
+                ALTER TABLE order_history ADD COLUMN escrow_order_id TEXT;
+                ALTER TABLE order_history ADD COLUMN fee_escrow_tx_hash TEXT;
+            """)
+
     def get_credentials(self, wallet_address: str) -> StoredCredentials | None:
         row = self._db.execute(
             "SELECT * FROM clob_credentials WHERE wallet_address = ? COLLATE NOCASE",
@@ -148,13 +165,16 @@ class TradingSqliteBackend:
         cur = self._db.execute(
             """INSERT INTO order_history
                (wallet_address, order_id, token_id, side, price, size, order_type,
-                status, error_msg, response_json, created_at)
-               VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
+                status, error_msg, response_json, created_at,
+                fee_amount_raw, escrow_order_id, fee_escrow_tx_hash)
+               VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)""",
             (
                 order["wallet_address"], order.get("order_id"), order["token_id"],
                 order["side"], order["price"], order["size"], order.get("order_type", "GTC"),
                 order.get("status", "submitting"), order.get("error_msg"),
                 order.get("response_json"), order.get("created_at", time.time()),
+                order.get("fee_amount_raw"), order.get("escrow_order_id"),
+                order.get("fee_escrow_tx_hash"),
             ),
         )
         self._db.commit()
@@ -163,13 +183,17 @@ class TradingSqliteBackend:
     def update_order_status(
         self, row_id: int, status: str, order_id: str | None = None,
         error_msg: str | None = None, response_json: str | None = None,
+        fee_amount_raw: str | None = None, escrow_order_id: str | None = None,
+        fee_escrow_tx_hash: str | None = None,
     ) -> None:
         self._db.execute(
             """UPDATE order_history
                SET status = ?, order_id = COALESCE(?, order_id),
-                   error_msg = COALESCE(?, error_msg), response_json = COALESCE(?, response_json)
+                   error_msg = COALESCE(?, error_msg), response_json = COALESCE(?, response_json),
+                   fee_amount_raw = COALESCE(?, fee_amount_raw), escrow_order_id = COALESCE(?, escrow_order_id),
+                   fee_escrow_tx_hash = COALESCE(?, fee_escrow_tx_hash)
                WHERE id = ?""",
-            (status, order_id, error_msg, response_json, row_id),
+            (status, order_id, error_msg, response_json, fee_amount_raw, escrow_order_id, fee_escrow_tx_hash, row_id),
         )
         self._db.commit()
 
@@ -200,6 +224,9 @@ class TradingSqliteBackend:
                 id=r[0], wallet_address=r[1], order_id=r[2], token_id=r[3],
                 side=r[4], price=r[5], size=r[6], order_type=r[7],
                 status=r[8], error_msg=r[9], response_json=r[10], created_at=r[11],
+                fee_amount_raw=r[12] if len(r) > 12 else None,
+                escrow_order_id=r[13] if len(r) > 13 else None,
+                fee_escrow_tx_hash=r[14] if len(r) > 14 else None,
             )
             for r in rows
         ]

polynode/trading/trader.py

@@ -24,6 +24,7 @@ from .constants import (
 )
 from .cosigner import build_l2_headers, send_via_cosigner
 from .eip712 import create_signed_order, create_signed_order_v2
+from .escrow import calculate_fee, generate_escrow_order_id, fetch_escrow_nonce, sign_fee_auth
 from .onboarding import (
     check_approvals as check_approvals_onchain,
     check_balance as check_balance_onchain,
@@ -55,6 +56,7 @@ from .types import (
     TraderConfig,
     WalletExport,
     WalletInfo,
+    FeeConfig,
 )
 
 
@@ -71,6 +73,7 @@ class PolyNodeTrader:
         self._rpc_url = c.rpc_url
         self._builder_credentials = c.builder_credentials
         self._exchange_version = c.exchange_version
+        self._fee_config = c.fee_config
 
         self._db: TradingSqliteBackend | None = None
         self._active_signer: NormalizedSigner | None = None
@@ -579,9 +582,47 @@ class PolyNodeTrader:
             "status": "submitting",
         })
 
+        # Build fee auth if escrow is enabled
+        fee_config = params.fee_config or self._fee_config
+        fee_auth_dict = None
+        fee_amount_raw = 0
+        if fee_config and fee_config.fee_bps > 0:
+            fee_amount_raw = calculate_fee(params.price, params.size, fee_config.fee_bps)
+            if fee_amount_raw > 0:
+                funder = creds.funder_address or creds.wallet_address
+                nonce = await fetch_escrow_nonce(self._rpc_url, signer.address)
+                deadline = int(time.time()) + 300
+                escrow_oid = generate_escrow_order_id()
+                fee_auth = await sign_fee_auth(
+                    signer.sign_typed_data,
+                    escrow_order_id=escrow_oid,
+                    payer=funder,
+                    signer_address=signer.address,
+                    fee_amount=fee_amount_raw,
+                    deadline=deadline,
+                    nonce=nonce,
+                    affiliate=fee_config.affiliate,
+                    affiliate_share_bps=fee_config.affiliate_share_bps,
+                )
+                fee_auth_dict = {
+                    "escrow_order_id": fee_auth.escrow_order_id,
+                    "payer": fee_auth.payer,
+                    "signer": fee_auth.signer,
+                    "fee_amount": fee_auth.fee_amount,
+                    "deadline": fee_auth.deadline,
+                    "nonce": fee_auth.nonce,
+                    "signature": fee_auth.signature,
+                    "affiliate": fee_auth.affiliate,
+                    "affiliate_share_bps": fee_auth.affiliate_share_bps,
+                }
+
+        request_data: dict[str, Any] = {"method": "POST", "path": "/order", "body": body_str, "headers": headers}
+        if fee_auth_dict:
+            request_data["fee_auth"] = fee_auth_dict
+
         result = await send_via_cosigner(
             self._cosigner_url, self._polynode_key, self._fallback_direct,
-            {"method": "POST", "path": "/order", "body": body_str, "headers": headers},
+            request_data,
             builder_credentials=self._builder_credentials,
             clob_host=clob_host,
         )
@@ -603,6 +644,8 @@ class PolyNodeTrader:
             error=error,
             making_amount=result.get("makingAmount"),
             taking_amount=result.get("takingAmount"),
+            fee_escrow_tx_hash=result.get("feeEscrowTxHash"),
+            fee_amount=str(fee_amount_raw / 1e6) if fee_amount_raw > 0 else None,
         )
 
     async def cancel_order(self, order_id: str) -> CancelResult:

polynode/trading/types.py

@@ -38,6 +38,14 @@ class GeneratedWallet:
     address: str
 
 
+@dataclass
+class FeeConfig:
+    """Optional fee configuration for per-order fee collection via on-chain escrow."""
+    fee_bps: int  # fee in basis points (e.g. 50 = 0.5%)
+    affiliate: str | None = None  # partner wallet address
+    affiliate_share_bps: int | None = None  # partner's share of fee in bps
+
+
 @dataclass
 class BuilderCredentials:
     """Polymarket builder credentials for order attribution."""
@@ -56,6 +64,7 @@ class TraderConfig:
     rpc_url: str = "https://polygon-bor-rpc.publicnode.com"
     builder_credentials: BuilderCredentials | None = None
     exchange_version: ExchangeVersion = ExchangeVersion.V1
+    fee_config: FeeConfig | None = None
 
 
 @dataclass
@@ -127,6 +136,7 @@ class OrderParams:
     type: OrderType = "GTC"
     expiration: int | None = None
     post_only: bool = False
+    fee_config: FeeConfig | None = None
 
 
 @dataclass
@@ -137,6 +147,8 @@ class OrderResult:
     error: str | None = None
     making_amount: str | None = None
     taking_amount: str | None = None
+    fee_escrow_tx_hash: str | None = None
+    fee_amount: str | None = None
 
 
 @dataclass
@@ -224,6 +236,9 @@ class OrderHistoryRow:
     error_msg: str | None
     response_json: str | None
     created_at: float
+    fee_amount_raw: str | None = None
+    escrow_order_id: str | None = None
+    fee_escrow_tx_hash: str | None = None
 
 
 @dataclass

{polynode-0.7.0.dist-info → polynode-0.7.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polynode
-Version: 0.7.0
+Version: 0.7.1
 Summary: Python SDK for the PolyNode real-time prediction market data platform
 Project-URL: Homepage, https://polynode.dev
 Project-URL: Documentation, https://docs.polynode.dev

{polynode-0.7.0.dist-info → polynode-0.7.1.dist-info}/RECORD

@@ -13,16 +13,17 @@ polynode/ws.py,sha256=VZdn2dkUJH2-CYBu_L45YnvCU0kQejxHcy2Ji2pRu60,9653
 polynode/cache/__init__.py,sha256=A9lXcIE3ISKKo1r0o3o-CkmQpBhLdKwdQfCR0eW5i9o,519
 polynode/trading/__init__.py,sha256=0mtvY4w2k1x000YOX7_ua9JZj4CY27Cz7p0u6L3Hfrw,906
 polynode/trading/clob_api.py,sha256=ygaOoMIg4PGtMzNBT70wk4XrSbVX0EyCCU6PmXK3xrs,5308
-polynode/trading/constants.py,sha256=CjHcfaOTzPFvlorY4XeyGtMS8XNKH4jjoxqKtpN6NmY,1782
+polynode/trading/constants.py,sha256=_DDwutJkOfeyPkClrTcyJCOYMoqDymZzDpjYWeHbfG0,1862
 polynode/trading/cosigner.py,sha256=Fww9OgaKdHQAHd8Kom1AJBtmMGC8FbNZxdFaFALNBY8,3058
 polynode/trading/eip712.py,sha256=pZf0Pf4DDIPpK4qbz4DWfcdOeU8PlCQhJPpKSQ97pC4,9553
-polynode/trading/onboarding.py,sha256=ouP9bzioFHCt4fnluAXLWL8O8U6d_1ePeu5JmBXeDyc,13768
+polynode/trading/escrow.py,sha256=mDNAUQUzFOvzo6HpmqJTzjMEx7o3ySQHoVbvNiqIwmA,4549
+polynode/trading/onboarding.py,sha256=-Ub5tsJA4vqPJmkNLx2OE_szUY08UmXYi_9ATThajkA,14670
 polynode/trading/position_management.py,sha256=z2pYls4ErhyRE6thQdOs45BF4llKvKNNruQgcMEFT_U,3827
 polynode/trading/privy.py,sha256=iiUQZsBqjj29C7oUiFFktoT2muZY06bcMzABKstFKqE,4523
 polynode/trading/signer.py,sha256=piifRquwnwFwrKLL24jsRQYMh_cdkz8wNz99Vx8_KTY,3413
-polynode/trading/sqlite_backend.py,sha256=9ho7DnQVE_nJqmPH5yPSAYcVjVmTS1XmD3NebR-UAwA,7986
-polynode/trading/trader.py,sha256=VVNQ_DEkum5u3IC2M5IfLml7s06A7fCqjGXvSX3ayCg,29497
-polynode/trading/types.py,sha256=WGSH25c1LIZax4KNPavHz0G6HEPLUK9ak9VnEYd8fxM,4944
+polynode/trading/sqlite_backend.py,sha256=zfo4WevuunWV-7i6Z3gQi_e7Tonw-7m1TsEd88O5I4w,9486
+polynode/trading/trader.py,sha256=U2dghRO5Ra1-2UqCBzOkI3GYvGPvAHKJAc2lFF-UJXQ,31541
+polynode/trading/types.py,sha256=JxhjsRRhh8WA127VNNPLf7QJKn5e_PZucaG7WAfdOsM,5528
 polynode/types/__init__.py,sha256=0v_CByqG-zT1xucbT2So_77YDmsPj9u956at94nyrLM,280
 polynode/types/enums.py,sha256=2cx9l10A-LUeDNBJZn7l0XBTka2YfxRQcXW0s7s3_4k,1082
 polynode/types/events.py,sha256=MJGZqxBFKfT1y44dZ96bHbwvoPpGvtutBwYE6OTT6nU,6880
@@ -30,6 +31,6 @@ polynode/types/orderbook.py,sha256=MwD6P3poo3F70bONUZ5i3ajk1sgw39hHTOw6D0fP5_8,1
 polynode/types/rest.py,sha256=ftJKQZH0wBrR_l199rDMqnVS8cyt8YnCvo9cKIsTcCw,7012
 polynode/types/short_form.py,sha256=xfGklDi587u6K2aXaRWiwbxqkipW8BiEPeNIDT45v2Y,799
 polynode/types/ws.py,sha256=EzE9Vzrb6cILemfssmnqM8fbHt4USn3latpUHKatwk4,994
-polynode-0.7.0.dist-info/METADATA,sha256=qIF1aMlx-nQAUySzKpoqj3y9Mx66vRNnG4Cuuwvbt0Q,3302
-polynode-0.7.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-polynode-0.7.0.dist-info/RECORD,,
+polynode-0.7.1.dist-info/METADATA,sha256=-RaAIV8DkzVWW3HIGy-5bkdSkThsihEk909bBWjMsx8,3302
+polynode-0.7.1.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+polynode-0.7.1.dist-info/RECORD,,