pmxt 1.0.3__py3-none-any.whl → 1.1.0__py3-none-any.whl

pmxt/client.py

@@ -198,8 +198,18 @@ class Exchange(ABC):
         # Configure the API client with the actual base URL
         config = Configuration(host=base_url)
         self._api_client = ApiClient(configuration=config)
+        
+        # Add access token from lock file
+        server_info = self._server_manager.get_server_info()
+        if server_info and 'accessToken' in server_info:
+            self._api_client.default_headers['x-pmxt-access-token'] = server_info['accessToken']
+            
         self._api = DefaultApi(api_client=self._api_client)
     
+    def close(self):
+        """No-op for now, kept for API compatibility with TS."""
+        pass
+    
     def _handle_response(self, response: Dict[str, Any]) -> Any:
         """Handle API response and extract data."""
         if not response.get("success"):
@@ -436,6 +446,106 @@ class Exchange(ABC):
         except ApiException as e:
             raise Exception(f"Failed to fetch trades: {e}")
     
+    # WebSocket Streaming Methods
+    
+    def watch_order_book(self, outcome_id: str, limit: Optional[int] = None) -> OrderBook:
+        """
+        Watch real-time order book updates via WebSocket.
+        
+        Returns a promise that resolves with the next order book update.
+        Call repeatedly in a loop to stream updates (CCXT Pro pattern).
+        
+        Args:
+            outcome_id: Outcome ID to watch
+            limit: Optional depth limit for order book
+            
+        Returns:
+            Next order book update
+            
+        Example:
+            >>> # Stream order book updates
+            >>> while True:
+            ...     order_book = exchange.watch_order_book(outcome_id)
+            ...     print(f"Best bid: {order_book.bids[0].price}")
+            ...     print(f"Best ask: {order_book.asks[0].price}")
+        """
+        try:
+            args = [outcome_id]
+            if limit is not None:
+                args.append(limit)
+            
+            body_dict = {"args": args}
+            
+            # Add credentials if available
+            creds = self._get_credentials_dict()
+            if creds:
+                body_dict["credentials"] = creds
+            
+            request_body = internal_models.WatchOrderBookRequest.from_dict(body_dict)
+            
+            response = self._api.watch_order_book(
+                exchange=self.exchange_name,
+                watch_order_book_request=request_body,
+            )
+            
+            data = self._handle_response(response.to_dict())
+            return _convert_order_book(data)
+        except ApiException as e:
+            raise Exception(f"Failed to watch order book: {e}")
+    
+    def watch_trades(
+        self,
+        outcome_id: str,
+        since: Optional[int] = None,
+        limit: Optional[int] = None
+    ) -> List[Trade]:
+        """
+        Watch real-time trade updates via WebSocket.
+        
+        Returns a promise that resolves with the next trade(s).
+        Call repeatedly in a loop to stream updates (CCXT Pro pattern).
+        
+        Args:
+            outcome_id: Outcome ID to watch
+            since: Optional timestamp to filter trades from
+            limit: Optional limit for number of trades
+            
+        Returns:
+            Next trade update(s)
+            
+        Example:
+            >>> # Stream trade updates
+            >>> while True:
+            ...     trades = exchange.watch_trades(outcome_id)
+            ...     for trade in trades:
+            ...         print(f"Trade: {trade.price} @ {trade.amount}")
+        """
+        try:
+            args = [outcome_id]
+            if since is not None:
+                args.append(since)
+            if limit is not None:
+                args.append(limit)
+            
+            body_dict = {"args": args}
+            
+            # Add credentials if available
+            creds = self._get_credentials_dict()
+            if creds:
+                body_dict["credentials"] = creds
+            
+            request_body = internal_models.WatchTradesRequest.from_dict(body_dict)
+            
+            response = self._api.watch_trades(
+                exchange=self.exchange_name,
+                watch_trades_request=request_body,
+            )
+            
+            data = self._handle_response(response.to_dict())
+            return [_convert_trade(t) for t in data]
+        except ApiException as e:
+            raise Exception(f"Failed to watch trades: {e}")
+    
     # Trading Methods (require authentication)
     
     def create_order(self, params: CreateOrderParams) -> Order:

pmxt/server_manager.py

@@ -69,15 +69,62 @@ class ServerManager:
         Raises:
             Exception: If server fails to start or become healthy
         """
-        # Step 1: Check if server is already running
+        # Step 1: Check if force restart is requested (DEV MODE)
+        if os.getenv('PMXT_ALWAYS_RESTART') == '1':
+            self._kill_old_server()
+
+        # Step 2: Check if server is already running and matches version
         if self.is_server_alive():
-            return
+            if self._is_version_mismatch():
+                # print("PMXT: Version mismatch detected. Restarting server...")
+                self._kill_old_server()
+            else:
+                return
         
-        # Step 2: Start server via launcher
+        # Step 3: Start server via launcher
         self._start_server_via_launcher()
         
-        # Step 3: Wait for health check
+        # Step 4: Wait for health check
         self._wait_for_health()
+
+    def _is_version_mismatch(self) -> bool:
+        """Check if running server version matches expected version."""
+        server_info = self.get_server_info()
+        if not server_info or 'version' not in server_info:
+            return True # Old server without version
+        
+        # Get expected version
+        try:
+            # 1. Check production path (bundled)
+            pkg_path = Path(__file__).parent / '_server' / 'package.json'
+            
+            # 2. Check dev path (monorepo)
+            if not pkg_path.exists():
+                # Traverse up to find core/package.json
+                pkg_path = Path(__file__).parent.parent.parent.parent / 'core' / 'package.json'
+
+            if pkg_path.exists():
+                data = json.loads(pkg_path.read_text())
+                expected_version = data.get('version')
+                if expected_version and not server_info['version'].startswith(expected_version):
+                     return True
+        except:
+            pass
+            
+        return False
+
+    def _kill_old_server(self) -> None:
+        """Kill the currently running server."""
+        server_info = self.get_server_info()
+        if server_info and 'pid' in server_info:
+            import signal
+            try:
+                os.kill(server_info['pid'], signal.SIGTERM)
+                # Brief wait for cleanup
+                time.sleep(0.5)
+            except:
+                pass
+        self._remove_stale_lock()
     
     def is_server_alive(self) -> bool:
         """

{pmxt-1.0.3.dist-info → pmxt-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pmxt
-Version: 1.0.3
+Version: 1.1.0
 Summary: Unified prediction market data API - The ccxt for prediction markets
 Author: PMXT Contributors
 License: MIT
@@ -206,10 +206,10 @@ For deep-dive methods like `fetch_ohlcv()`, `fetch_order_book()`, and `fetch_tra
 
 ```python
 markets = poly.search_markets("Trump")
-outcome_id = markets[0].outcomes[0].id  # ✅ Correct
+outcome_id = markets[0].outcomes[0].id  # Correct
 
-candles = poly.fetch_ohlcv(outcome_id, ...)  # ✅ Works
-candles = poly.fetch_ohlcv(markets[0].id, ...)  # ❌ Wrong!
+candles = poly.fetch_ohlcv(outcome_id, ...)  # Works
+candles = poly.fetch_ohlcv(markets[0].id, ...)  # Wrong!
 ```
 
 ### Prices are 0.0 to 1.0

{pmxt-1.0.3.dist-info → pmxt-1.1.0.dist-info}/RECORD

@@ -1,20 +1,20 @@
-pmxt/__init__.py,sha256=vh7zkQGfvusno6-CgWx2iDFSKjQS2Yk_aCPWHOuaCEA,1150
-pmxt/client.py,sha256=pXAPl_IPhnza2xwlObSlLia9aWpBHA97Nsxq9aP3yvU,23485
+pmxt/__init__.py,sha256=2XbKCO7ujQzhPPnyxAh1Qtc3aQuvilehplhDgoo1t3w,1150
+pmxt/client.py,sha256=2qlCJsxkQiHYrYvZYX4p-APvYq0rF_yicNfUPgcE3ZU,27368
 pmxt/models.py,sha256=Mu0hLVjQU3PM5m68poK61VQcJtPQ8ZdLdLU7pq1lqjQ,6004
-pmxt/server_manager.py,sha256=GVTZAFBs5LC8dXGpeH_kt2ZlKpYk2dbYncCbt8gRcX0,8799
+pmxt/server_manager.py,sha256=wmHcGBM3oMNLSajCZ0xnANNSmqBYt9vWxQzeYmfo0-Y,10635
 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=Dvq8eVbqk8UrGK71b6jBK-kXI0-ixnspMe5zScxWCrA,3395166
-pmxt_internal/__init__.py,sha256=UfA5SFa5dJ73wYol0X2vem-rfhir4DOd6_nSYjkCaqg,5973
-pmxt_internal/api_client.py,sha256=yhAWPtRlSAZZo6u4ShePgRTTbL5j4DXXDyXtCB_ayfM,27889
+pmxt/_server/server/bundled.js,sha256=TXtX8YZlSnrIoN_ZmxMA25WqQw6j-eZXMi4YNr32p58,4175627
+pmxt_internal/__init__.py,sha256=gbAxJ-pDBsZHbr7bgWhX66LV_CEkm--f9BYRB9X483o,6400
+pmxt_internal/api_client.py,sha256=viGpxECLDA9KW58HKl-hxCM9a-0nuYNMWnWHXPAirrk,27889
 pmxt_internal/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
-pmxt_internal/configuration.py,sha256=ezNaGgJUCKgOLcF0-UjwrhwthQiKNTNULw4hUDiZ8Us,18320
+pmxt_internal/configuration.py,sha256=pngy-anwjS9enhuiaYdNcz0AP1xJ6cYHEoTYUJjXsQg,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=V_-PxsEbTL4atCQR1r7GbeFvhtsVdbf3TSbTRM--Pv8,147482
-pmxt_internal/models/__init__.py,sha256=KpXW3FEk-LTETIR0djQy7GX_NxGIElnIg_SzfHLmqAA,3196
+pmxt_internal/api/default_api.py,sha256=h1vGidUmbkXhBHAE1THOWsGjiJc0RnIQTy1fRVSye7M,171451
+pmxt_internal/models/__init__.py,sha256=vjHF9UaS2iEhSGavMQL9PtS5NFaFUMTEKKzg-qJwKeE,3449
 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
@@ -53,7 +53,10 @@ pmxt_internal/models/search_markets_request.py,sha256=BARoy2GXgV7RQNIGck6UaOyQqf
 pmxt_internal/models/search_markets_request_args_inner.py,sha256=PkusFd_OxhUsItsBpluPJA11zg0sXXjbOK-lPmemvLs,5561
 pmxt_internal/models/trade.py,sha256=U6Fc18rbwILs9FmX8CSDYYL8dF6763l8QzeMQNRxQdo,3328
 pmxt_internal/models/unified_market.py,sha256=mNyUrGKWxvI2f2KCa2uhGwq22cBdDia2hD5AtzFw8fE,4487
-pmxt-1.0.3.dist-info/METADATA,sha256=wRbKtij3NsrlPLx9HmSho_HyDkfwe-nKjTnpWCWWtNA,6300
-pmxt-1.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pmxt-1.0.3.dist-info/top_level.txt,sha256=J_jrcouJ-x-5lpcXMxeW0GOSi1HsBVR5_PdSfvigVrw,19
-pmxt-1.0.3.dist-info/RECORD,,
+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.1.0.dist-info/METADATA,sha256=le091hHBquNEKcZpGKhRyouzBh1amaAcD7ZUkL7loPc,6288
+pmxt-1.1.0.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+pmxt-1.1.0.dist-info/top_level.txt,sha256=J_jrcouJ-x-5lpcXMxeW0GOSi1HsBVR5_PdSfvigVrw,19
+pmxt-1.1.0.dist-info/RECORD,,

{pmxt-1.0.3.dist-info → pmxt-1.1.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

pmxt_internal/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "1.0.3"
+__version__ = "1.1.0"
 
 # Define package exports
 __all__ = [
@@ -66,6 +66,9 @@ __all__ = [
     "SearchMarketsRequestArgsInner",
     "Trade",
     "UnifiedMarket",
+    "WatchOrderBookRequest",
+    "WatchOrderBookRequestArgsInner",
+    "WatchTradesRequest",
 ]
 
 # import apis into sdk package
@@ -121,4 +124,7 @@ from pmxt_internal.models.search_markets_request import SearchMarketsRequest as
 from pmxt_internal.models.search_markets_request_args_inner import SearchMarketsRequestArgsInner as SearchMarketsRequestArgsInner
 from pmxt_internal.models.trade import Trade as Trade
 from pmxt_internal.models.unified_market import UnifiedMarket as UnifiedMarket
+from pmxt_internal.models.watch_order_book_request import WatchOrderBookRequest as WatchOrderBookRequest
+from pmxt_internal.models.watch_order_book_request_args_inner import WatchOrderBookRequestArgsInner as WatchOrderBookRequestArgsInner
+from pmxt_internal.models.watch_trades_request import WatchTradesRequest as WatchTradesRequest