polymarket-apis 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

polymarket_apis/types/clob_types.py

@@ -21,6 +21,7 @@ from ..utilities.constants import ZERO_ADDRESS
 
 logger = logging.getLogger(__name__)
 
+
 class ApiCreds(BaseModel):
     key: str = Field(alias="apiKey")
     secret: str
@@ -122,11 +123,13 @@ class Rewards(BaseModel):
     rewards_min_size: int = Field(alias="min_size")
     rewards_max_spread: float = Field(alias="max_spread")
 
+
 class EarnedReward(BaseModel):
     asset_address: EthAddress
     earnings: float
     asset_rate: float
 
+
 class DailyEarnedReward(BaseModel):
     date: datetime
     asset_address: EthAddress
@@ -134,12 +137,12 @@ class DailyEarnedReward(BaseModel):
     earnings: float
     asset_rate: float
 
-class PolymarketRewardItem(BaseModel):
+
+class RewardMarket(BaseModel):
     market_id: str
     condition_id: Keccak256
     question: str
     market_slug: str
-    market_description: str
     event_slug: str
     image: str
     maker_address: EthAddress
@@ -165,6 +168,7 @@ class MarketRewards(BaseModel):
     rewards_min_size: int
     market_competitiveness: float
 
+
 class ClobMarket(BaseModel):
     # Core market information
     token_ids: list[Token] = Field(alias="tokens")
@@ -215,7 +219,9 @@ class ClobMarket(BaseModel):
 
     @field_validator("neg_risk_market_id", "neg_risk_request_id", mode="wrap")
     @classmethod
-    def validate_neg_risk_fields(cls, value: str, handler: ValidatorFunctionWrapHandler, info: ValidationInfo) -> Optional[str]:
+    def validate_neg_risk_fields(
+        cls, value: str, handler: ValidatorFunctionWrapHandler, info: ValidationInfo
+    ) -> Optional[str]:
         try:
             return handler(value)
         except ValidationError as e:
@@ -227,12 +233,17 @@ class ClobMarket(BaseModel):
                 return value
             if neg_risk and value == "":
                 for _ in e.errors():
-                    msg = ("Poorly setup market: negative risk is True, but either neg_risk_market_id or neg_risk_request_id is missing. "
-                           f" Question: {info.data.get("question")}; Market slug: {info.data.get('market_slug')} \n")
+                    msg = (
+                        "Poorly setup market: negative risk is True, but either neg_risk_market_id or neg_risk_request_id is missing. "
+                        f" Question: {info.data.get('question')}; Market slug: {info.data.get('market_slug')} \n"
+                    )
                     logger.warning(msg)
+
     @field_validator("condition_id", "question_id", mode="wrap")
     @classmethod
-    def validate_condition_fields(cls, value: str, handler: ValidatorFunctionWrapHandler, info: ValidationInfo) -> str:
+    def validate_condition_fields(
+        cls, value: str, handler: ValidatorFunctionWrapHandler, info: ValidationInfo
+    ) -> str:
         try:
             return handler(value)
         except ValueError:
@@ -241,6 +252,7 @@ class ClobMarket(BaseModel):
                 return value
             raise
 
+
 class OpenOrder(BaseModel):
     order_id: Keccak256 = Field(alias="id")
     status: str
@@ -258,6 +270,7 @@ class OpenOrder(BaseModel):
     associate_trades: list[str]
     created_at: datetime
 
+
 class MakerOrder(BaseModel):
     token_id: str = Field(alias="asset_id")
     order_id: Keccak256
@@ -268,6 +281,7 @@ class MakerOrder(BaseModel):
     outcome: str
     fee_rate_bps: float
 
+
 class PolygonTrade(BaseModel):
     trade_id: str = Field(alias="id")
     taker_order_id: Keccak256
@@ -277,7 +291,7 @@ class PolygonTrade(BaseModel):
     size: float
     fee_rate_bps: float
     price: float
-    status: str # change to literals MINED, CONFIRMED
+    status: str  # change to literals MINED, CONFIRMED
     match_time: datetime
     last_update: datetime
     outcome: str
@@ -288,6 +302,7 @@ class PolygonTrade(BaseModel):
     maker_orders: list[MakerOrder]
     trader_side: Literal["TAKER", "MAKER"]
 
+
 class TradeParams(BaseModel):
     id: Optional[str] = None
     maker_address: Optional[str] = None
@@ -304,13 +319,14 @@ class OpenOrderParams(BaseModel):
 
 
 class DropNotificationParams(BaseModel):
-    ids: Optional[list[str]]= None
+    ids: Optional[list[str]] = None
 
 
 class OrderSummary(BaseModel):
     price: Optional[float] = None
     size: Optional[float] = None
 
+
 class PriceLevel(OrderSummary):
     side: Literal["BUY", "SELL"]
 
@@ -375,6 +391,7 @@ class RoundConfig(BaseModel):
     size: int
     amount: int
 
+
 class OrderArgs(BaseModel):
     token_id: str
     """
@@ -456,12 +473,14 @@ class MarketOrderArgs(BaseModel):
 
     order_type: OrderType = OrderType.FOK
 
+
 class PostOrdersArgs(BaseModel):
     order: SignedOrder
     order_type: OrderType = OrderType.GTC
 
     model_config = ConfigDict(arbitrary_types_allowed=True)
 
+
 class ContractConfig(BaseModel):
     """Contract Configuration."""
 
@@ -489,6 +508,7 @@ class OrderPostResponse(BaseModel):
     status: str = Literal["live", "matched", "delayed"]
     success: bool
 
+
 class OrderCancelResponse(BaseModel):
     not_canceled: Optional[dict[Keccak256, str]]
     canceled: Optional[list[Keccak256]]

polymarket_apis/types/common.py

@@ -1,7 +1,8 @@
 import re
-from datetime import datetime
+from datetime import UTC, datetime
 from typing import Annotated
 
+from dateutil import parser
 from pydantic import AfterValidator, BaseModel, BeforeValidator, Field
 
 
@@ -11,41 +12,23 @@ def validate_keccak256(v: str) -> str:
         raise ValueError(msg)
     return v
 
-def parse_timestamp(v: str) -> datetime:
-    if isinstance(v, datetime):
-        return v
-    # Normalize '+00' to '+0000' for timezone
-    if v.endswith("+00"):
-        v = v[:-3] + "+0000"
-
-    # Pad fractional seconds to 6 digits if present
-    if "." in v:
-        dot_pos = v.find(".")
-        tz_pos = v.find("+", dot_pos)  # Find timezone start after '.'
-        if tz_pos == -1:
-            tz_pos = v.find("-", dot_pos)
-
-        if tz_pos != -1:
-            frac = v[dot_pos+1:tz_pos]
-            if len(frac) < 6:
-                frac = frac.ljust(6, "0")
-                v = f"{v[:dot_pos+1]}{frac}{v[tz_pos:]}"
-
-    # Try parsing with and without microseconds
-    for fmt in ("%Y-%m-%d %H:%M:%S.%f%z", "%Y-%m-%d %H:%M:%S%z"):
-        try:
-            return datetime.strptime(v, fmt) # noqa: DTZ007
-        except ValueError:
-            continue
-    msg = f"Time data '{v}' does not match expected formats."
-    raise ValueError(msg)
-
-
-TimestampWithTZ = Annotated[datetime, BeforeValidator(parse_timestamp)]
+
+def parse_flexible_datetime(v: str | datetime) -> datetime:
+    """Parse datetime from multiple formats using dateutil."""
+    if v in {"NOW*()", "NOW()"}:
+        return datetime.fromtimestamp(0, tz=UTC)
+
+    if isinstance(v, str):
+        return parser.parse(v)
+    return v
+
+
+FlexibleDatetime = Annotated[datetime, BeforeValidator(parse_flexible_datetime)]
 EthAddress = Annotated[str, Field(pattern=r"^0x[A-Fa-f0-9]{40}$")]
 Keccak256 = Annotated[str, AfterValidator(validate_keccak256)]
 EmptyString = Annotated[str, Field(pattern=r"^$", description="An empty string")]
 
+
 class TimeseriesPoint(BaseModel):
     value: float
     timestamp: datetime

polymarket_apis/types/data_types.py

@@ -1,5 +1,5 @@
 from datetime import UTC, datetime
-from typing import Literal
+from typing import Literal, Optional
 
 from pydantic import BaseModel, Field, field_validator
 
@@ -44,7 +44,7 @@ class Position(BaseModel):
     @field_validator("end_date", mode="before")
     def handle_empty_end_date(cls, v):
         if v == "":
-            return datetime(2099,12,31, tzinfo=UTC)
+            return datetime(2099, 12, 31, tzinfo=UTC)
         return v
 
 
@@ -145,6 +145,7 @@ class ValueResponse(BaseModel):
     # Value information
     value: float
 
+
 class User(BaseModel):
     proxy_wallet: EthAddress = Field(alias="proxyWallet")
     name: str
@@ -152,10 +153,22 @@ class User(BaseModel):
     profile_image: str = Field(alias="profileImage")
     profile_image_optimized: str = Field(alias="profileImageOptimized")
 
+
 class UserMetric(User):
     amount: float
     pseudonym: str
 
+
 class UserRank(User):
     amount: float
     rank: int
+
+
+class MarketValue(BaseModel):
+    condition_id: Keccak256 = Field(alias="market")
+    value: float
+
+
+class EventLiveVolume(BaseModel):
+    total: Optional[float]
+    markets: Optional[list[MarketValue]]