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

prediction_market_agent_tooling/gtypes.py

@@ -1,6 +1,6 @@
 import typing as t
 from decimal import Decimal
-from typing import NewType
+from typing import Annotated, NewType, TypeAlias
 
 from eth_typing.evm import (  # noqa: F401  # Import for the sake of easy importing with others from here.
     Address,
@@ -8,6 +8,7 @@ from eth_typing.evm import (  # noqa: F401  # Import for the sake of easy import
     HexAddress,
     HexStr,
 )
+from pydantic import BeforeValidator
 from pydantic.types import SecretStr
 from pydantic.v1.types import SecretStr as SecretStrV1
 from web3 import Web3
@@ -18,6 +19,7 @@ from web3.types import (  # noqa: F401  # Import for the sake of easy importing
 )
 from web3.types import Wei as Web3Wei
 
+from prediction_market_agent_tooling.gtypes import ChecksumAddress
 from prediction_market_agent_tooling.tools._generic_value import _GenericValue
 from prediction_market_agent_tooling.tools.datetime_utc import (  # noqa: F401  # Import for the sake of easy importing with others from here.
     DatetimeUTC,
@@ -26,6 +28,14 @@ from prediction_market_agent_tooling.tools.hexbytes_custom import (  # noqa: F40
     HexBytes,
 )
 
+VerifiedChecksumAddress: TypeAlias = Annotated[
+    ChecksumAddress, BeforeValidator(Web3.to_checksum_address)
+]
+VerifiedChecksumAddressOrNone: TypeAlias = Annotated[
+    ChecksumAddress | None,
+    BeforeValidator(lambda x: Web3.to_checksum_address(x) if x is not None else None),
+]
+
 
 class CollateralToken(_GenericValue[int | float | str | Decimal, float], parser=float):
     """

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,115 @@
-from pydantic import BaseModel
+from enum import Enum
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict
 from sqlmodel import Field, SQLModel
 
-from prediction_market_agent_tooling.gtypes import ChecksumAddress, HexBytes
+from prediction_market_agent_tooling.gtypes import (
+    HexBytes,
+    VerifiedChecksumAddress,
+    VerifiedChecksumAddressOrNone,
+    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: VerifiedChecksumAddress
+    buyToken: VerifiedChecksumAddress
     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: VerifiedChecksumAddress
+    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: VerifiedChecksumAddress
+    buyAmount: Wei
+    buyToken: VerifiedChecksumAddress
+    receiver: VerifiedChecksumAddressOrNone
+    feeAmount: Wei
     creationDate: DatetimeUTC
+    kind: OrderKind
+    partiallyFillable: bool
+    sellTokenBalance: SellTokenBalance | None
+    buyTokenBalance: BuyTokenBalance | None
+    signingScheme: SigningScheme
+    signature: HexBytes
+    from_: VerifiedChecksumAddressOrNone = Field(None, alias="from")
+    appData: str
+    fullAppData: str | None
+    appDataHash: HexBytes | None = None
+    class_: str = Field(None, alias="class")
+    owner: VerifiedChecksumAddress
+    executedSellAmount: Wei
+    executedSellAmountBeforeFees: Wei
+    executedBuyAmount: Wei
+    executedFeeAmount: Wei | None
+    invalidated: bool
+    status: str
+    isLiquidityOrder: bool | None
+    ethflowData: EthFlowData | None = None
+    onchainUser: VerifiedChecksumAddressOrNone = None
+    executedFee: Wei
+    executedFeeToken: VerifiedChecksumAddressOrNone
 
 
 class RateLimit(SQLModel, table=True):

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prediction-market-agent-tooling
-Version: 0.69.9
+Version: 0.69.10.dev1120
 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.dev1120.dist-info}/RECORD

@@ -36,7 +36,7 @@ prediction_market_agent_tooling/deploy/gcp/deploy.py,sha256=CYUgnfy-9XVk04kkxA_5
 prediction_market_agent_tooling/deploy/gcp/kubernetes_models.py,sha256=OsPboCFGiZKsvGyntGZHwdqPlLTthITkNF5rJFvGgU8,2582
 prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=WI2ycX1X-IlTRoNoG4ggFlRwPL28kwM9VGDFD2fePLo,5699
 prediction_market_agent_tooling/deploy/trade_interval.py,sha256=Xk9j45alQ_vrasGvsNyuW70XHIQ7wfvjoxNR3F6HYCw,1155
-prediction_market_agent_tooling/gtypes.py,sha256=bUIZfZIGvIi3aiZNu5rVE9kRevw8sfMa4bcze6QeBg8,6058
+prediction_market_agent_tooling/gtypes.py,sha256=tHXMBjh0iAEJEcBywFvwzwPqDkBUpjEyeb7ynh7fOKc,6469
 prediction_market_agent_tooling/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/jobs/jobs_models.py,sha256=4KZ0iaKZx5zEZBBiEo9CHk0OpX3bKjoYA0wmptR0Wyk,2455
 prediction_market_agent_tooling/jobs/omen/omen_jobs.py,sha256=CojpgghRs0lFEf5wl-oWhGaQozG616YD0cXdlmFO-U0,5116
@@ -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=EpCziNs4sGmN7ZvPKpqdqjhVd_-zPZHsJVeVfFu4Y60,3005
 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.dev1120.dist-info/METADATA,sha256=_zBgziT22Ha55DnWQSUlO2ITl7W1-zIXoNY19nKoGqs,8899
+prediction_market_agent_tooling-0.69.10.dev1120.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+prediction_market_agent_tooling-0.69.10.dev1120.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.69.10.dev1120.dist-info/licenses/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.69.10.dev1120.dist-info/RECORD,,

{prediction_market_agent_tooling-0.69.9.dist-info → prediction_market_agent_tooling-0.69.10.dev1120.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