prediction-market-agent-tooling 0.69.9__py3-none-any.whl → 0.69.10.dev1119__py3-none-any.whl

prediction_market_agent_tooling/tools/cow/cow_order.py

@@ -399,6 +399,22 @@ def get_orders_by_owner(
     return [Order.model_validate(i) for i in items]
 
 
+@tenacity.retry(
+    stop=stop_after_attempt(3),
+    wait=wait_fixed(1),
+    after=lambda x: logger.debug(f"get_order_by_uid failed, {x.attempt_number=}."),
+)
+async def get_order_by_uid(
+    uid: HexBytes,
+) -> Order:
+    async with httpx.AsyncClient() as client:
+        response = await client.get(
+            f"https://api.cow.fi/xdai/api/v1/orders/{uid.to_0x_hex()}"
+        )
+        response.raise_for_status()
+        return Order.model_validate(response.json())
+
+
 @tenacity.retry(
     stop=stop_after_attempt(3),
     wait=wait_fixed(1),

prediction_market_agent_tooling/tools/cow/models.py

@@ -1,23 +1,131 @@
-from pydantic import BaseModel
+from enum import Enum
+from typing import Annotated, Optional
+
+from pydantic import BaseModel, BeforeValidator, ConfigDict
 from sqlmodel import Field, SQLModel
+from web3 import Web3
 
-from prediction_market_agent_tooling.gtypes import ChecksumAddress, HexBytes
+from prediction_market_agent_tooling.gtypes import ChecksumAddress, HexBytes, Wei
 from prediction_market_agent_tooling.tools.datetime_utc import DatetimeUTC
 from prediction_market_agent_tooling.tools.utils import utcnow
 
 
 class MinimalisticTrade(BaseModel):
-    sellToken: ChecksumAddress
-    buyToken: ChecksumAddress
+    sellToken: Annotated[ChecksumAddress, BeforeValidator(Web3.to_checksum_address)]
+    buyToken: Annotated[ChecksumAddress, BeforeValidator(Web3.to_checksum_address)]
     orderUid: HexBytes
     txHash: HexBytes
 
 
+class EthFlowData(BaseModel):
+    refundTxHash: Optional[HexBytes]
+    userValidTo: int
+
+
+class PlacementError(str, Enum):
+    QuoteNotFound = "QuoteNotFound"
+    ValidToTooFarInFuture = "ValidToTooFarInFuture"
+    PreValidationError = "PreValidationError"
+
+
+class OnchainOrderData(BaseModel):
+    sender: Annotated[ChecksumAddress, BeforeValidator(Web3.to_checksum_address)]
+    placementError: Optional[PlacementError]
+
+
+class OrderKind(str, Enum):
+    buy = "buy"
+    sell = "sell"
+
+
+class SellTokenBalance(str, Enum):
+    external = "external"
+    internal = "internal"
+    erc20 = "erc20"
+
+
+class BuyTokenBalance(str, Enum):
+    internal = "internal"
+    erc20 = "erc20"
+
+
+class SigningScheme(str, Enum):
+    eip712 = "eip712"
+    ethsign = "ethsign"
+    presign = "presign"
+    eip1271 = "eip1271"
+
+
+class OrderClass(str, Enum):
+    limit = "limit"
+    liquidity = "liquidity"
+    market = "market"
+
+
+class OrderStatus(str, Enum):
+    presignaturePending = "presignaturePending"
+    open = "open"
+    fulfilled = "fulfilled"
+    cancelled = "cancelled"
+    expired = "expired"
+
+
 class Order(BaseModel):
+    model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)
+
     uid: str
-    sellToken: str
-    buyToken: str
+    quoteId: int | None = None
+    validTo: int
+    sellAmount: Wei
+    sellToken: Annotated[ChecksumAddress, BeforeValidator(Web3.to_checksum_address)]
+    buyAmount: Wei
+    buyToken: Annotated[ChecksumAddress, BeforeValidator(Web3.to_checksum_address)]
+    receiver: Annotated[
+        ChecksumAddress | None,
+        BeforeValidator(
+            lambda x: Web3.to_checksum_address(x) if x is not None else None
+        ),
+    ]
+    feeAmount: Wei
     creationDate: DatetimeUTC
+    kind: OrderKind
+    partiallyFillable: bool
+    sellTokenBalance: SellTokenBalance | None
+    buyTokenBalance: BuyTokenBalance | None
+    signingScheme: SigningScheme
+    signature: HexBytes
+    from_: Annotated[
+        ChecksumAddress | None,
+        BeforeValidator(
+            lambda x: Web3.to_checksum_address(x) if x is not None else None
+        ),
+    ] = Field(None, alias="from")
+    appData: str
+    fullAppData: str | None
+    appDataHash: HexBytes | None = None
+    class_: str = Field(None, alias="class")
+    owner: Annotated[ChecksumAddress, BeforeValidator(Web3.to_checksum_address)]
+    executedSellAmount: Wei
+    executedSellAmountBeforeFees: Wei
+    executedBuyAmount: Wei
+    executedFeeAmount: Wei | None
+    invalidated: bool
+    status: str
+    isLiquidityOrder: bool | None
+    ethflowData: EthFlowData | None = None
+    onchainUser: Annotated[
+        ChecksumAddress | None,
+        BeforeValidator(
+            lambda x: Web3.to_checksum_address(x) if x is not None else None
+        ),
+    ] = None
+    executedFee: Wei
+    executedFeeToken: Annotated[
+        ChecksumAddress | None,
+        BeforeValidator(
+            lambda x: Web3.to_checksum_address(x) if x is not None else None
+        ),
+    ]
 
 
 class RateLimit(SQLModel, table=True):

{prediction_market_agent_tooling-0.69.9.dist-info → prediction_market_agent_tooling-0.69.10.dev1119.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prediction-market-agent-tooling
-Version: 0.69.9
+Version: 0.69.10.dev1119
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 License-File: LICENSE
 Author: Gnosis

{prediction_market_agent_tooling-0.69.9.dist-info → prediction_market_agent_tooling-0.69.10.dev1119.dist-info}/RECORD

@@ -95,8 +95,8 @@ prediction_market_agent_tooling/tools/caches/serializers.py,sha256=vFDx4fsPxclXp
 prediction_market_agent_tooling/tools/contract.py,sha256=BzpAFcbKl_KqwgAlaXx63Fg8jzr0EO3qEeOs1K11CPA,33905
 prediction_market_agent_tooling/tools/contract_utils.py,sha256=9X9raICUZkPDShilt02aYzS_ILZ62u0vG5081uWLdqk,2152
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
-prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=kdvhWVTb31oHVspQFowDNEIoDtx8hwGTlKsWYGRh3oQ,14050
-prediction_market_agent_tooling/tools/cow/models.py,sha256=2hTW-Ye6SamBBAO32nVo4f3vjAOOySw8ehECReRcA3o,756
+prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=DN_8cPrr4jWVpXdS4D0j1QB19nB8fxDoSheo2BFMc8M,14523
+prediction_market_agent_tooling/tools/cow/models.py,sha256=y_zsPEw7sG-OV6ooycQ9mx75GtnG3SuwXc6xuxL_o3I,3799
 prediction_market_agent_tooling/tools/cow/semaphore.py,sha256=LPB-4wNQf7El7dgBD4Tmol6vr5j1CP9qMeDm8dcs6RI,3741
 prediction_market_agent_tooling/tools/custom_exceptions.py,sha256=Fh8z1fbwONvP4-j7AmV_PuEcoqb6-QXa9PJ9m7guMcM,93
 prediction_market_agent_tooling/tools/datetime_utc.py,sha256=_oO6mMc28C9aSmQfrG-S7UQy5uMHVEQqia8arnscVCk,3213
@@ -137,8 +137,8 @@ prediction_market_agent_tooling/tools/tokens/usd.py,sha256=DPO-4HBTy1-TZHKL_9CnH
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=ruq6P5TFs8CBHxeBLj1Plpx7kuNFPpDgMsJGQgDiRNs,8785
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=CDbaidlLeQ4VHzSg150L7QNfHfGveljSePGuDVFEYqc,13963
-prediction_market_agent_tooling-0.69.9.dist-info/METADATA,sha256=fBezbmaBxLVzpMNtCsmCnE2Pc8ywbXJTnCRF7HMJcYw,8890
-prediction_market_agent_tooling-0.69.9.dist-info/WHEEL,sha256=M5asmiAlL6HEcOq52Yi5mmk9KmTVjY2RDPtO4p9DMrc,88
-prediction_market_agent_tooling-0.69.9.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.69.9.dist-info/licenses/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.69.9.dist-info/RECORD,,
+prediction_market_agent_tooling-0.69.10.dev1119.dist-info/METADATA,sha256=4cM7-CJ3tpEUsBYQasv-387chs1IoRw0L1l5Xs7CqIU,8899
+prediction_market_agent_tooling-0.69.10.dev1119.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+prediction_market_agent_tooling-0.69.10.dev1119.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.69.10.dev1119.dist-info/licenses/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.69.10.dev1119.dist-info/RECORD,,

{prediction_market_agent_tooling-0.69.9.dist-info → prediction_market_agent_tooling-0.69.10.dev1119.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.2.0
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any