Skip to content

Commit 70e2072

Browse files
committed
Review tweaks
1 parent cd5799f commit 70e2072

2 files changed

Lines changed: 6 additions & 9 deletions

File tree

blockapi/v2/api/debank.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -627,9 +627,6 @@ def _parse_app(self, raw_app: dict) -> Optional[DebankApp]:
627627
predictions = []
628628

629629
chain = DEBANK_APP_CHAIN_MAP.get(model.id)
630-
if not chain:
631-
logger.warning(f'Unknown chain for app {model.id}, using default')
632-
chain = Blockchain.ETHEREUM
633630

634631
for portfolio_item in model.portfolio_item_list:
635632
detail_types = portfolio_item.detail_types
@@ -655,7 +652,7 @@ def _parse_app(self, raw_app: dict) -> Optional[DebankApp]:
655652
)
656653

657654
def _parse_prediction(
658-
self, item: DebankModelAppPortfolioItem, chain: Blockchain
655+
self, item: DebankModelAppPortfolioItem, chain: Optional[Blockchain]
659656
) -> Optional[DebankPrediction]:
660657
"""Parse a prediction market position."""
661658
try:
@@ -679,7 +676,7 @@ def _parse_prediction(
679676
)
680677

681678
def _parse_deposit(
682-
self, item: DebankModelAppPortfolioItem, chain: Blockchain
679+
self, item: DebankModelAppPortfolioItem, chain: Optional[Blockchain]
683680
) -> Optional[DebankAppDeposit]:
684681
"""Parse a deposit/common type portfolio item."""
685682
return DebankAppDeposit.from_api(

blockapi/v2/models.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1246,7 +1246,7 @@ class DebankPrediction:
12461246
claimable: bool
12471247
event_end_at: Optional[datetime]
12481248
is_market_closed: bool
1249-
chain: Blockchain
1249+
chain: Optional[Blockchain]
12501250
position_index: Optional[str]
12511251
update_at: Optional[datetime]
12521252

@@ -1261,7 +1261,7 @@ def from_api(
12611261
usd_value: Union[str, float, int],
12621262
claimable: bool,
12631263
is_market_closed: bool,
1264-
chain: Blockchain,
1264+
chain: Optional[Blockchain] = None,
12651265
event_end_at: Optional[Union[int, float]] = None,
12661266
position_index: Optional[str] = None,
12671267
update_at: Optional[Union[int, float]] = None,
@@ -1290,7 +1290,7 @@ class DebankAppDeposit:
12901290
debt_usd_value: Decimal
12911291
net_usd_value: Decimal
12921292
tokens: list[DebankDepositToken]
1293-
chain: Blockchain
1293+
chain: Optional[Blockchain]
12941294
position_index: Optional[str]
12951295
update_at: Optional[datetime]
12961296

@@ -1304,7 +1304,7 @@ def from_api(
13041304
net_usd_value: Union[str, float, int],
13051305
position_index: str,
13061306
tokens: Optional[list[DebankDepositToken]] = None,
1307-
chain: Blockchain,
1307+
chain: Optional[Blockchain] = None,
13081308
update_at: Optional[Union[int, float]] = None,
13091309
) -> 'DebankAppDeposit':
13101310
return cls(

0 commit comments

Comments
 (0)