prediction-market-agent-tooling 0.65.5__py3-none-any.whl → 0.69.17.dev1149__py3-none-any.whl

prediction_market_agent_tooling/markets/polymarket/data_models_web.py

@@ -1,420 +0,0 @@
-"""
-Autogenerated using `datamodel-codegen` from a single Polymarket response. Then adjusted, fixed.
-Keep in mind that not all fields were used so far, so there might be some bugs.
-
-These models are based on what Polymarket's website returns in the response, and `prediction_market_agent_tooling/markets/polymarket/data_models.py` are based on what their API returns.
-"""
-
-import json
-import typing as t
-
-import requests
-from pydantic import BaseModel, field_validator
-
-from prediction_market_agent_tooling.gtypes import USDC, HexAddress, OutcomeStr
-from prediction_market_agent_tooling.loggers import logger
-from prediction_market_agent_tooling.markets.data_models import Resolution
-from prediction_market_agent_tooling.tools.utils import DatetimeUTC
-
-POLYMARKET_BASE_URL = "https://polymarket.com"
-POLYMARKET_TRUE_OUTCOME = "Yes"
-POLYMARKET_FALSE_OUTCOME = "No"
-
-
-class ImageOptimized(BaseModel):
-    imageUrlOptimized: str
-
-
-class IconOptimized(BaseModel):
-    imageUrlOptimized: str
-
-
-class Event(BaseModel):
-    id: str
-    slug: str
-    ticker: str
-    title: str
-    series: t.Any | None = None
-
-
-class Event1(BaseModel):
-    startDate: DatetimeUTC | None = None
-    slug: str
-
-
-class Market1(BaseModel):
-    slug: str
-    question: str
-    image: str
-    volume: USDC | None = None
-    outcomes: t.Sequence[OutcomeStr]
-    outcomePrices: list[USDC]
-    active: bool
-    archived: bool
-    closed: bool
-    orderPriceMinTickSize: USDC
-    clobTokenIds: str
-    events: list[Event1] | None = None
-
-
-class ResolutionData(BaseModel):
-    id: str
-    author: str
-    lastUpdateTimestamp: int
-    status: str
-    wasDisputed: bool
-    price: str
-    proposedPrice: str
-    reproposedPrice: str
-    updates: str
-    newVersionQ: bool
-    transactionHash: str
-    logIndex: str
-
-
-class Market(BaseModel):
-    id: str
-    question: str
-    conditionId: str
-    slug: str
-    twitterCardImage: t.Any | None = None
-    resolutionSource: str | None = None
-    endDate: DatetimeUTC
-    category: t.Any | None = None
-    ammType: t.Any | None = None
-    description: str
-    liquidity: str | None = None
-    startDate: DatetimeUTC | None = None
-    createdAt: DatetimeUTC
-    xAxisValue: t.Any | None = None
-    yAxisValue: t.Any | None = None
-    denominationToken: t.Any | None = None
-    fee: str | None = None
-    lowerBound: t.Any | None = None
-    upperBound: t.Any | None = None
-    outcomes: t.Sequence[OutcomeStr]
-    image: str | None = None
-    icon: str | None = None
-    imageOptimized: t.Any | None = None
-    iconOptimized: t.Any | None = None
-    outcomePrices: list[USDC]
-    volume: USDC | None = None
-    active: bool
-    marketType: str | None = None
-    formatType: t.Any | None = None
-    lowerBoundDate: t.Any | None = None
-    upperBoundDate: t.Any | None = None
-    closed: bool
-    marketMakerAddress: HexAddress
-    closedTime: DatetimeUTC | None = None
-    wideFormat: bool | None = None
-    new: bool | None = None
-    sentDiscord: t.Any | None = None
-    mailchimpTag: t.Any | None = None
-    featured: bool | None = None
-    submitted_by: str | None = None
-    subcategory: t.Any | None = None
-    categoryMailchimpTag: t.Any | None = None
-    archived: bool | None = None
-    resolvedBy: str | None = None
-    restricted: bool | None = None
-    groupItemTitle: str | None = None
-    groupItemThreshold: str | None = None
-    questionID: str | None = None
-    umaEndDate: t.Any | None = None
-    enableOrderBook: bool | None = None
-    orderPriceMinTickSize: float | None = None
-    orderMinSize: int | None = None
-    umaResolutionStatus: t.Any | None = None
-    curationOrder: t.Any | None = None
-    volumeNum: USDC | None = None
-    liquidityNum: float | None = None
-    endDateIso: DatetimeUTC | None = None
-    startDateIso: DatetimeUTC | None = None
-    umaEndDateIso: DatetimeUTC | None = None
-    commentsEnabled: bool | None = None
-    disqusThread: t.Any | None = None
-    gameStartTime: t.Any | None = None
-    secondsDelay: int | None = None
-    clobTokenIds: list[str] | None = None
-    liquidityAmm: float | None = None
-    liquidityClob: float | None = None
-    makerBaseFee: int | None = None
-    takerBaseFee: int | None = None
-    negRisk: t.Any | None = None
-    negRiskRequestID: t.Any | None = None
-    negRiskMarketID: t.Any | None = None
-    events: list[Event] | None = None
-    markets: list[Market1] | None = None
-    lower_bound_date: t.Any | None = None
-    upper_bound_date: t.Any | None = None
-    market_type: str | None = None
-    resolution_source: str | None = None
-    end_date: str | None = None
-    amm_type: t.Any | None = None
-    x_axis_value: t.Any | None = None
-    y_axis_value: t.Any | None = None
-    denomination_token: t.Any | None = None
-    resolved_by: str | None = None
-    upper_bound: t.Any | None = None
-    lower_bound: t.Any | None = None
-    created_at: str | None = None
-    updated_at: t.Any | None = None
-    closed_time: t.Any | None = None
-    wide_format: bool | None = None
-    volume_num: USDC | None = None
-    liquidity_num: USDC | None = None
-    image_raw: str | None = None
-    resolutionData: ResolutionData
-
-    @field_validator("closedTime", mode="before")
-    def field_validator_closedTime(cls, v: str | None = None) -> str | None:
-        return v.replace("+00", "") if v else None
-
-    @property
-    def resolution(self) -> Resolution | None:
-        # If the market is not closed, it doesn't have a resolution.
-        if not self.closed:
-            return None
-
-        outcome_to_outcome_price: dict[str, USDC] = dict(
-            zip(self.outcomes, self.outcomePrices)
-        )
-
-        # On Polymarket, we can find out binary market resolution by checking for the outcome prices.
-        # E.g. if `Yes` price (probability) is 1$ and `No` price (probability) is 0$, it means the resolution is `Yes`.
-        if (
-            outcome_to_outcome_price[POLYMARKET_TRUE_OUTCOME] == 1.0
-            and outcome_to_outcome_price[POLYMARKET_FALSE_OUTCOME] == 0.0
-        ):
-            return Resolution(
-                outcome=OutcomeStr(POLYMARKET_TRUE_OUTCOME), invalid=False
-            )
-
-        elif (
-            outcome_to_outcome_price[POLYMARKET_TRUE_OUTCOME] == 0.0
-            and outcome_to_outcome_price[POLYMARKET_FALSE_OUTCOME] == 1.0
-        ):
-            return Resolution(
-                outcome=OutcomeStr(POLYMARKET_FALSE_OUTCOME), invalid=False
-            )
-
-        else:
-            raise ValueError(
-                f"Unexpected outcome prices {outcome_to_outcome_price} or outcomes {self.outcomes}, please debug."
-            )
-
-
-class Category(BaseModel):
-    id: str
-    label: str
-    parentCategory: str
-    slug: str
-
-
-class Bid(BaseModel):
-    price: str
-    size: str
-
-
-class Ask(BaseModel):
-    price: str
-    size: str
-
-
-class MarketBidsAndAsks(BaseModel):
-    market: str
-    asset_id: str
-    hash: str
-    bids: list[Bid]
-    asks: list[Ask]
-    lastUpdated: int
-
-
-class PolymarketFullMarket(BaseModel):
-    id: str
-    ticker: str
-    slug: str
-    title: str
-    subtitle: t.Any | None = None
-    description: str
-    commentCount: int | None = None
-    resolutionSource: str | None = None
-    startDate: DatetimeUTC | None = None
-    endDate: DatetimeUTC
-    image: str | None = None
-    icon: str | None = None
-    featuredImage: str | None = None
-    active: bool
-    closed: bool
-    archived: bool | None = None
-    new: bool | None = None
-    featured: bool | None = None
-    restricted: bool | None = None
-    liquidity: USDC | None = None
-    volume: USDC | None = None
-    volume24hr: USDC | None = None
-    competitive: float | None = None
-    openInterest: int | None = None
-    sortBy: str | None = None
-    createdAt: DatetimeUTC
-    commentsEnabled: bool | None = None
-    disqusThread: t.Any | None = None
-    updatedAt: DatetimeUTC
-    enableOrderBook: bool | None = None
-    liquidityAmm: float | None = None
-    liquidityClob: float | None = None
-    imageOptimized: ImageOptimized | None = None
-    iconOptimized: IconOptimized | None = None
-    featuredImageOptimized: str | None = None
-    negRisk: t.Any | None = None
-    negRiskMarketID: t.Any | None = None
-    negRiskFeeBips: t.Any | None = None
-    markets: list[Market]
-    categories: list[Category] | None = None
-    series: t.Any | None = None
-    image_raw: str | None = None
-
-    @property
-    def url(self) -> str:
-        return construct_polymarket_url(self.slug)
-
-    @property
-    def is_main_market(self) -> bool:
-        # On Polymarket, there are markets that are actually a group of multiple Yes/No markets, for example https://polymarket.com/event/presidential-election-winner-2024.
-        # But if there is only 1 market in this list, it should mean it's a "main market", e.g. with only one question.
-        return len(self.markets) == 1
-
-    @property
-    def main_market(self) -> Market:
-        if not self.is_main_market:
-            raise ValueError(
-                f"Unexpected number of markets in the Polymarket's response, shouldn't happen if you used `is_main_market` filter, please debug: {self.id=}, {self.title=}"
-            )
-        return self.markets[0]
-
-    @staticmethod
-    def fetch_from_url(url: str) -> "PolymarketFullMarket | None":
-        """
-        Get the full market data from the Polymarket website.
-
-        Returns None if this market's url returns "Oops...we didn't forecast this", see `check_if_its_a_main_market` method for more details.
-
-        Warning: This is a very slow operation, as it requires fetching the website. Use it only when necessary.
-        """
-        logger.info(f"Fetching full market from {url}")
-
-        # Fetch the website as a normal browser would.
-        headers = {
-            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"
-        }
-        content = requests.get(url, headers=headers).text
-
-        # Find the JSON with the data within the content.
-        start_tag = """<script id="__NEXT_DATA__" type="application/json" crossorigin="anonymous">"""
-        start_idx = content.find(start_tag) + len(start_tag)
-        end_idx = content.find("</script>", start_idx)
-        response_data = content[start_idx:end_idx]
-
-        # Parsing.
-        response_dict = json.loads(response_data)
-        response_model = PolymarketWebResponse.model_validate(response_dict)
-
-        full_market_queries = [
-            q
-            for q in response_model.props.pageProps.dehydratedState.queries
-            if isinstance(q.state.data, PolymarketFullMarket)
-        ]
-
-        # We expect either 0 markets (if it doesn't exist) or 1 market.
-        if len(full_market_queries) not in (0, 1):
-            raise ValueError(
-                f"Unexpected number of queries in the response, please check it out and modify the code accordingly: `{response_dict}`"
-            )
-
-        # It will be `PolymarketFullMarket` thanks to the filter above.
-        market = (
-            t.cast(PolymarketFullMarket, full_market_queries[0].state.data)
-            if full_market_queries
-            else None
-        )
-
-        if market is None:
-            logger.warning(f"No polymarket found for {url}")
-
-        return market
-
-
-class PriceSide(BaseModel):
-    price: USDC
-    side: t.Literal["BUY", "SELL"]
-
-
-class State(BaseModel):
-    data: (
-        PolymarketFullMarket
-        | MarketBidsAndAsks
-        | PriceSide
-        | dict[str, MarketBidsAndAsks]
-        | dict[str, PriceSide]
-        | None
-        # It's none if you go to the website and it says "Oops...we didn't forecast this".
-    )
-    dataUpdateCount: int
-    dataUpdatedAt: int
-    error: t.Any | None = None
-    errorUpdateCount: int
-    errorUpdatedAt: int
-    fetchFailureCount: int
-    fetchFailureReason: t.Any | None = None
-    fetchMeta: t.Any | None = None
-    isInvalidated: bool
-    status: str
-    fetchStatus: str
-
-
-class Query(BaseModel):
-    state: State
-    queryKey: list[str]
-    queryHash: str
-
-
-class DehydratedState(BaseModel):
-    mutations: list[t.Any]
-    queries: list[Query]
-
-
-class PageProps(BaseModel):
-    key: str
-    dehydratedState: DehydratedState
-    eslug: str
-    mslug: str | None = None
-    isSingleMarket: bool
-
-
-class Props(BaseModel):
-    pageProps: PageProps
-
-
-class Query1(BaseModel):
-    slug: list[str]
-
-
-class PolymarketWebResponse(BaseModel):
-    props: Props
-    page: str
-    query: Query1
-    buildId: str
-    isFallback: bool
-    gsp: bool
-    locale: str
-    locales: list[str]
-    defaultLocale: str
-    scriptLoader: list[t.Any]
-
-
-def construct_polymarket_url(slug: str) -> str:
-    """
-    Note: This works only if it's a single main market, not sub-market of some more general question.
-    """
-    return f"{POLYMARKET_BASE_URL}/event/{slug}"