Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions src/polymarket/models/data/activity.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,7 @@ class TradeActivity(_KnownActivityBase):
outcome_index: int = Field(validation_alias="outcomeIndex")
title: str
slug: str
icon: str
icon: str | None = None
event_slug: str = Field(validation_alias="eventSlug")

@field_validator("condition_id", mode="before")
Expand All @@ -140,7 +140,7 @@ class _MarketEventActivity(_KnownActivityBase):
amount: Decimal
title: str
slug: str
icon: str
icon: str | None = None
event_slug: str = Field(validation_alias="eventSlug")

@field_validator("condition_id", mode="before")
Expand Down Expand Up @@ -284,7 +284,7 @@ def _normalize_activity_payload(data: dict[str, Any]) -> dict[str, Any]:
if normalized.get("outcomeIndex") == 999:
normalized.pop("outcomeIndex", None)

for sentinel_key in ("conditionId", "asset", "side", "outcome"):
for sentinel_key in ("conditionId", "asset", "side", "outcome", "icon"):
if normalized.get(sentinel_key) == "":
normalized.pop(sentinel_key, None)

Expand Down
12 changes: 12 additions & 0 deletions tests/unit/test_data_activity.py
Original file line number Diff line number Diff line change
Expand Up @@ -125,6 +125,18 @@ def test_empty_string_sentinels_drop_to_none_on_trade_raises() -> None:
parse_activity(payload)


def test_trade_empty_icon_normalizes_to_none() -> None:
activity = parse_activity(_trade_payload(icon=""))
assert isinstance(activity, TradeActivity)
assert activity.icon is None


def test_market_event_empty_icon_normalizes_to_none() -> None:
activity = parse_activity(_market_event_payload("SPLIT", icon=""))
assert isinstance(activity, SplitActivity)
assert activity.icon is None


def test_market_event_variants_parse() -> None:
for activity_type, expected_class in [
("SPLIT", SplitActivity),
Expand Down
Loading