pmxt 1.3.4__py3-none-any.whl → 1.4.0__py3-none-any.whl

pmxt/__init__.py

@@ -33,7 +33,7 @@ from .models import (
     CreateOrderParams,
 )
 
-__version__ = "1.3.4"
+__version__ = "1.4.0"
 __all__ = [
     # Exchanges
     "Polymarket",

pmxt/_server/server/bundled.js

@@ -24359,12 +24359,61 @@ var require_lib3 = __commonJS({
   }
 });
 
+// dist/utils/math.js
+var require_math = __commonJS({
+  "dist/utils/math.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.getExecutionPrice = getExecutionPrice;
+    exports2.getExecutionPriceDetailed = getExecutionPriceDetailed;
+    function getExecutionPrice(orderBook, side, amount) {
+      const result = getExecutionPriceDetailed(orderBook, side, amount);
+      return result.fullyFilled ? result.price : 0;
+    }
+    function getExecutionPriceDetailed(orderBook, side, amount) {
+      if (amount <= 0) {
+        throw new Error("Amount must be greater than 0");
+      }
+      let levels = (side === "buy" ? orderBook.asks : orderBook.bids).filter((l) => l.size > 0);
+      levels.sort((a, b) => side === "buy" ? a.price - b.price : b.price - a.price);
+      if (levels.length === 0) {
+        return {
+          price: 0,
+          filledAmount: 0,
+          fullyFilled: false
+        };
+      }
+      let remainingAmount = amount;
+      let totalCost = 0;
+      let filledAmount = 0;
+      const EPSILON = 1e-8;
+      for (const level of levels) {
+        if (remainingAmount <= EPSILON) {
+          break;
+        }
+        const fillSize = Math.min(remainingAmount, level.size);
+        totalCost += fillSize * level.price;
+        filledAmount += fillSize;
+        remainingAmount -= fillSize;
+      }
+      const fullyFilled = remainingAmount <= EPSILON;
+      const executionPrice = filledAmount > EPSILON ? totalCost / filledAmount : 0;
+      return {
+        price: executionPrice,
+        filledAmount,
+        fullyFilled
+      };
+    }
+  }
+});
+
 // dist/BaseExchange.js
 var require_BaseExchange = __commonJS({
   "dist/BaseExchange.js"(exports2) {
     "use strict";
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.PredictionMarketExchange = void 0;
+    var math_1 = require_math();
     var PredictionMarketExchange = class {
       constructor(credentials) {
         this.credentials = credentials;
@@ -24438,6 +24487,12 @@ var require_BaseExchange = __commonJS({
       async fetchBalance() {
         throw new Error("Method fetchBalance not implemented.");
       }
+      getExecutionPrice(orderBook, side, amount) {
+        return (0, math_1.getExecutionPrice)(orderBook, side, amount);
+      }
+      getExecutionPriceDetailed(orderBook, side, amount) {
+        return (0, math_1.getExecutionPriceDetailed)(orderBook, side, amount);
+      }
       // ----------------------------------------------------------------------------
       // WebSocket Streaming Methods
       // ----------------------------------------------------------------------------

pmxt/client.py

@@ -7,9 +7,10 @@ OpenAPI client, matching the JavaScript API exactly.
 
 import os
 import sys
-from typing import List, Optional, Dict, Any
+from typing import List, Optional, Dict, Any, Literal
 from datetime import datetime
 from abc import ABC, abstractmethod
+import json
 
 # Add generated client to path
 _GENERATED_PATH = os.path.join(os.path.dirname(__file__), "..", "generated")
@@ -35,6 +36,7 @@ from .models import (
     MarketFilterParams,
     HistoryFilterParams,
     CreateOrderParams,
+    ExecutionPriceResult,
 )
 from .server_manager import ServerManager
 
@@ -169,6 +171,15 @@ def _convert_balance(raw: Dict[str, Any]) -> Balance:
     )
 
 
+def _convert_execution_result(raw: Dict[str, Any]) -> ExecutionPriceResult:
+    """Convert raw API response to ExecutionPriceResult."""
+    return ExecutionPriceResult(
+        price=raw.get("price", 0),
+        filled_amount=raw.get("filledAmount", 0),
+        fully_filled=raw.get("fullyFilled", False),
+    )
+
+
 class Exchange(ABC):
     """
     Base class for prediction market exchanges.
@@ -831,6 +842,77 @@ class Exchange(ABC):
         except ApiException as e:
             raise Exception(f"Failed to fetch balance: {e}")
 
+    def get_execution_price(
+        self,
+        order_book: OrderBook,
+        side: Literal["buy", "sell"],
+        amount: float
+    ) -> float:
+        """
+        Calculate the average execution price for a given amount.
+        
+        Args:
+            order_book: The current order book
+            side: "buy" or "sell"
+            amount: The amount to execute
+            
+        Returns:
+            The volume-weighted average price, or 0 if insufficient liquidity
+        """
+        result = self.get_execution_price_detailed(order_book, side, amount)
+        return result.price if result.fully_filled else 0
+
+    def get_execution_price_detailed(
+        self,
+        order_book: OrderBook,
+        side: Literal["buy", "sell"],
+        amount: float
+    ) -> ExecutionPriceResult:
+        """
+        Calculate detailed execution price information.
+        
+        Args:
+            order_book: The current order book
+            side: "buy" or "sell"
+            amount: The amount to execute
+            
+        Returns:
+            Detailed execution result
+        """
+        try:
+            # Convert order_book to dict for API call
+            bids = [{"price": b.price, "size": b.size} for b in order_book.bids]
+            asks = [{"price": a.price, "size": a.size} for a in order_book.asks]
+            ob_dict = {"bids": bids, "asks": asks, "timestamp": order_book.timestamp}
+
+            body = {
+                "args": [ob_dict, side, amount]
+            }
+            
+            creds = self._get_credentials_dict()
+            if creds:
+                body["credentials"] = creds
+                
+            url = f"{self._api_client.configuration.host}/api/{self.exchange_name}/getExecutionPriceDetailed"
+            
+            headers = {"Content-Type": "application/json", "Accept": "application/json"}
+            headers.update(self._api_client.default_headers)
+            
+            response = self._api_client.call_api(
+                method="POST",
+                url=url,
+                body=body,
+                header_params=headers
+            )
+            
+            response.read()
+            data_json = json.loads(response.data)
+            
+            data = self._handle_response(data_json)
+            return _convert_execution_result(data)
+        except Exception as e:
+            raise Exception(f"Failed to get execution price: {e}")
+
 
 class Polymarket(Exchange):
     """

pmxt/models.py

@@ -217,6 +217,20 @@ class OrderBook:
     """Unix timestamp (milliseconds)"""
 
 
+@dataclass
+class ExecutionPriceResult:
+    """Result of an execution price calculation."""
+    
+    price: float
+    """The volume-weighted average price"""
+    
+    filled_amount: float
+    """The actual amount that can be filled"""
+    
+    fully_filled: bool
+    """Whether the full requested amount can be filled"""
+
+
 @dataclass
 class Trade:
     """A historical trade."""

{pmxt-1.3.4.dist-info → pmxt-1.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pmxt
-Version: 1.3.4
+Version: 1.4.0
 Summary: Unified prediction market data API - The ccxt for prediction markets
 Author: PMXT Contributors
 License: MIT
@@ -160,6 +160,8 @@ for pos in positions:
 - `fetch_ohlcv(outcome_id, params)` - Get historical price candles
 - `fetch_order_book(outcome_id)` - Get current order book
 - `fetch_trades(outcome_id, params)` - Get trade history
+- `get_execution_price(order_book, side, amount)` - Get execution price
+- `get_execution_price_detailed(order_book, side, amount)` - Get detailed execution info
 
 ### Trading Methods (require authentication)
 

{pmxt-1.3.4.dist-info → pmxt-1.4.0.dist-info}/RECORD

@@ -1,20 +1,20 @@
-pmxt/__init__.py,sha256=QUyTYIHTeamwGXyGf7_d-7UM91n-trP78TgmiooWQjA,1150
-pmxt/client.py,sha256=YytS9Y1CD8A2NihUxedpmojH830eWE32v4-hbMsA4a8,30175
-pmxt/models.py,sha256=6ku911tJoT1ItglYWHmg_OfvDmCPEvFXWDL562wCBIs,8259
+pmxt/__init__.py,sha256=MAOqnWgwuaG-IN2JEPoGdUrp2DHARbZAFjHd-SmyEvs,1150
+pmxt/client.py,sha256=1qa8ObZtf-tdzhDXQXvWWdeLvGbyxsiVHAgQvl8Z4Uk,32955
+pmxt/models.py,sha256=-jiQ9mmv_qnF6mzj3DrvNgEA77tE_Pl0RCblM1VbV7o,8581
 pmxt/server_manager.py,sha256=-G97dYEdKl7F3HK9bAOKYl-SGWP6HsvzZIx2QxiZm24,11494
 pmxt/_server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pmxt/_server/bin/pmxt-ensure-server,sha256=kXIond0UbxS52FAVQD7kHmSBaL_s6cbIyapLRr4KZJw,4544
-pmxt/_server/server/bundled.js,sha256=KxMsOJdiw5D2ynSJGFUCeojDlA-w7Eu6ldlgDoeApp4,4196247
-pmxt_internal/__init__.py,sha256=rTbculzYsP3ZAsVNJckX_iFXMNdLU3eETpPLcR-Y6Is,6762
-pmxt_internal/api_client.py,sha256=7tnxPGfTzVx9xEneYQMVhZFCix6NMoR2NKHa1gkbeQk,27889
+pmxt/_server/server/bundled.js,sha256=ETpsJ6ZmRtO1gQmBZYwwHWTHJIEI8ooeeMhOwoeDJ84,4198184
+pmxt_internal/__init__.py,sha256=Kbxeq4WFfhIjcDkz72ZeXhWFpecUcOXCwUgqUzs6qsw,7578
+pmxt_internal/api_client.py,sha256=2XZZdrKh90kAA4P3bJ54IdzRfAFMawSUqwnp2pRZNYA,27889
 pmxt_internal/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
-pmxt_internal/configuration.py,sha256=sqCSSQUykt5E1s7EEI1gr-NYzCzKBtiei3cJXUVqnDs,18320
+pmxt_internal/configuration.py,sha256=evyyjQzPpYofXsIUdNKTMlMbpmq38l2ISokDuXJ0qJ0,18320
 pmxt_internal/exceptions.py,sha256=txF8A7vlan57JS69kFPs-IZF-Qhp7IZobBTJVa4fOaM,6644
 pmxt_internal/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pmxt_internal/rest.py,sha256=FMj4yaV6XLr842u_ScWHSzQsTFdk0jaUeuWLJoRbogQ,9760
 pmxt_internal/api/__init__.py,sha256=ppJSCipQ5IAk2z6UZkFaGSsEmnoAnSSHb8sjb_DYUkY,101
-pmxt_internal/api/default_api.py,sha256=4ySNtjkT9zJBfaHzX4fmXN5IgE-zIsr-uJs6WMO_N28,183099
-pmxt_internal/models/__init__.py,sha256=VPMpFsTZYckNs5JLcdO_8UwcZn4xEK3xQ58LUO0rxv4,3667
+pmxt_internal/api/default_api.py,sha256=6nMYcL__0wtDJAyDAHadN8zk1Sw-qE3nhcpvqUf3ymw,206490
+pmxt_internal/models/__init__.py,sha256=nviUepbTwMI-ro_lY6JbUgsQ1MskT7AKN3mb5WKJR74,4141
 pmxt_internal/models/balance.py,sha256=Dj5kFiLrsXOZyyXTC18bPjWrgw7qdWnTgTSCmk_l6xk,2962
 pmxt_internal/models/base_request.py,sha256=ZNipF7ycXFkQJ6j3QmB1TzA0UO3fB54AMPlAgIA3KOA,2987
 pmxt_internal/models/base_response.py,sha256=g-NG4Swxl3cg4-YOCPl65dUeHzOnA9S7ubTj8HOYZs0,2975
@@ -25,6 +25,7 @@ pmxt_internal/models/create_order_request.py,sha256=prj6TnIFbhB8eDq-6e-MK2DtfDzA
 pmxt_internal/models/error_detail.py,sha256=590jlnQmIgqcjUQ5ef07_nWMn7fnyjuAvkUjmSoAkMs,2605
 pmxt_internal/models/error_response.py,sha256=c5DgOpTGqx9Qoz7hYKhRPAM3UfoX4GBWkyhfxerMXEI,2979
 pmxt_internal/models/exchange_credentials.py,sha256=BtZCkGnnGQ24KPolTRtoA_jtXp-5_1Y1FmQLe1L5WCo,3308
+pmxt_internal/models/execution_price_result.py,sha256=gsYJWD4GXVC9-_YmtLleegxmS4p7Q9bLNaJNEz6Ljsk,3012
 pmxt_internal/models/fetch_balance200_response.py,sha256=VN5yrsVSKnI_WprK1U4wbl60jGEiSnu4VAOGkSM5K4o,3539
 pmxt_internal/models/fetch_markets200_response.py,sha256=olOMs8IBKPL2Foqw3r66jeGcRlJpOpGcGq0jNccUPuA,3564
 pmxt_internal/models/fetch_markets_request.py,sha256=2kV2N3jVPy7qKMpVMt-QiTP_Ea0Unrkq2R-qBnB5xKk,3627
@@ -39,6 +40,10 @@ pmxt_internal/models/fetch_positions200_response.py,sha256=mzEq-rEYJaZHmjqjcaW9w
 pmxt_internal/models/fetch_positions_request.py,sha256=F823DuKhltLMw04QrtSi1ZNVhkSISKz72evpWaEIaE8,3208
 pmxt_internal/models/fetch_trades200_response.py,sha256=fdsEd4luXJpzk6MuM192PaZ62fOBA3He3VreZhCd4-w,3527
 pmxt_internal/models/fetch_trades_request.py,sha256=GQZGh2dy-Y3n8DnR8x5b9bLH_nLmCmIJqjcTTNzcoSA,3690
+pmxt_internal/models/get_execution_price200_response.py,sha256=uf5eCUHZK5ZVzgJ2UrXMvOJnNvVhNq5Zwb6_MAWa_O4,3172
+pmxt_internal/models/get_execution_price_detailed200_response.py,sha256=PqBksuFhxToYvl-O4eM9WhSn0h5x0lwK4c7cttbAoL8,3463
+pmxt_internal/models/get_execution_price_request.py,sha256=sVPuP6QKDeAMOueYlJT9MQTg9JnQzDHI-C4B4Db4Cvc,3756
+pmxt_internal/models/get_execution_price_request_args_inner.py,sha256=CoiBFk1iGMIfWNGEDCLBGAyT4fFhjANttQftNOKaDIo,6225
 pmxt_internal/models/get_markets_by_slug_request.py,sha256=PtjXAw6hixHuYzCvSGH1aka2rZrsHmT8ofXYSuuOh9k,3173
 pmxt_internal/models/health_check200_response.py,sha256=yR_OkIlTPztO0zFmpyWllSwyaEbI48RpumfmVVCJAyc,2758
 pmxt_internal/models/history_filter_params.py,sha256=0i9oQLwJsRWRyzPZuW467y_0ccKpc2_I6T2lgZCkHd4,3239
@@ -59,7 +64,7 @@ pmxt_internal/models/unified_market.py,sha256=DoYhiH4HycYGlq858PEeB-CIA7haT6rxmJ
 pmxt_internal/models/watch_order_book_request.py,sha256=kavGUI-SLz2-Kam_jcJ_h0GDe0-9UkxqCmVsAi6Uios,3726
 pmxt_internal/models/watch_order_book_request_args_inner.py,sha256=ZHrjmFDGxRG5MXbuz4mUp9KFfo3XS7zuXWTyMNgi4xI,5464
 pmxt_internal/models/watch_trades_request.py,sha256=brrg8JbEe-aeg7mIe_Y2HzRPogp-IfRhkXChrxzqoLU,3722
-pmxt-1.3.4.dist-info/METADATA,sha256=Xp9QzK2hMi2HUFkBrflPTR88Do9Q_GLNd4QPLwHaNxE,6288
-pmxt-1.3.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-pmxt-1.3.4.dist-info/top_level.txt,sha256=J_jrcouJ-x-5lpcXMxeW0GOSi1HsBVR5_PdSfvigVrw,19
-pmxt-1.3.4.dist-info/RECORD,,
+pmxt-1.4.0.dist-info/METADATA,sha256=6fTb2tiWCkPoyKkYSs16e6yzhUdp-s-NaE7NPY6VFhc,6449
+pmxt-1.4.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+pmxt-1.4.0.dist-info/top_level.txt,sha256=J_jrcouJ-x-5lpcXMxeW0GOSi1HsBVR5_PdSfvigVrw,19
+pmxt-1.4.0.dist-info/RECORD,,

pmxt_internal/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "1.3.4"
+__version__ = "1.4.0"
 
 # Define package exports
 __all__ = [
@@ -38,6 +38,7 @@ __all__ = [
     "ErrorDetail",
     "ErrorResponse",
     "ExchangeCredentials",
+    "ExecutionPriceResult",
     "FetchBalance200Response",
     "FetchMarkets200Response",
     "FetchMarketsRequest",
@@ -52,6 +53,10 @@ __all__ = [
     "FetchPositionsRequest",
     "FetchTrades200Response",
     "FetchTradesRequest",
+    "GetExecutionPrice200Response",
+    "GetExecutionPriceDetailed200Response",
+    "GetExecutionPriceRequest",
+    "GetExecutionPriceRequestArgsInner",
     "GetMarketsBySlugRequest",
     "HealthCheck200Response",
     "HistoryFilterParams",
@@ -99,6 +104,7 @@ from pmxt_internal.models.create_order_request import CreateOrderRequest as Crea
 from pmxt_internal.models.error_detail import ErrorDetail as ErrorDetail
 from pmxt_internal.models.error_response import ErrorResponse as ErrorResponse
 from pmxt_internal.models.exchange_credentials import ExchangeCredentials as ExchangeCredentials
+from pmxt_internal.models.execution_price_result import ExecutionPriceResult as ExecutionPriceResult
 from pmxt_internal.models.fetch_balance200_response import FetchBalance200Response as FetchBalance200Response
 from pmxt_internal.models.fetch_markets200_response import FetchMarkets200Response as FetchMarkets200Response
 from pmxt_internal.models.fetch_markets_request import FetchMarketsRequest as FetchMarketsRequest
@@ -113,6 +119,10 @@ from pmxt_internal.models.fetch_positions200_response import FetchPositions200Re
 from pmxt_internal.models.fetch_positions_request import FetchPositionsRequest as FetchPositionsRequest
 from pmxt_internal.models.fetch_trades200_response import FetchTrades200Response as FetchTrades200Response
 from pmxt_internal.models.fetch_trades_request import FetchTradesRequest as FetchTradesRequest
+from pmxt_internal.models.get_execution_price200_response import GetExecutionPrice200Response as GetExecutionPrice200Response
+from pmxt_internal.models.get_execution_price_detailed200_response import GetExecutionPriceDetailed200Response as GetExecutionPriceDetailed200Response
+from pmxt_internal.models.get_execution_price_request import GetExecutionPriceRequest as GetExecutionPriceRequest
+from pmxt_internal.models.get_execution_price_request_args_inner import GetExecutionPriceRequestArgsInner as GetExecutionPriceRequestArgsInner
 from pmxt_internal.models.get_markets_by_slug_request import GetMarketsBySlugRequest as GetMarketsBySlugRequest
 from pmxt_internal.models.health_check200_response import HealthCheck200Response as HealthCheck200Response
 from pmxt_internal.models.history_filter_params import HistoryFilterParams as HistoryFilterParams