Skip to content

Feat/adapt gateway upgrade #97

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
cardosofede merged 3 commits into from
Dec 12, 2025
Merged

Feat/adapt gateway upgrade #97

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
7ea190a
(feat) prioritize sell tokens
cardosofede Dec 9, 2025
0269f94
(feat) remove dead code
cardosofede Dec 9, 2025
0e0800d
(feat) improve portfolio query by connector name
cardosofede Dec 10, 2025
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
1 change: 1 addition & 0 deletions models/trading.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -190,6 +190,7 @@ class PortfolioStateFilterRequest(BaseModel):
account_names: Optional[List[str]] = Field(default=None, description="List of account names to filter by")
connector_names: Optional[List[str]] = Field(default=None, description="List of connector names to filter by")
skip_gateway: bool = Field(default=False, description="Skip Gateway wallet balance updates for faster CEX-only queries")
refresh: bool = Field(default=False, description="If True, refresh balances before returning. If False, return cached state")


class PortfolioHistoryFilterRequest(TimeRangePaginationParams):
Expand Down
16 changes: 12 additions & 4 deletions routers/portfolio.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,21 +29,29 @@ async def get_portfolio_state(
- account_names: Optional list of account names to filter by
- connector_names: Optional list of connector names to filter by
- skip_gateway: If True, skip Gateway wallet balance updates for faster CEX-only queries
- refresh: If True, refresh balances before returning. If False (default), return cached state

Returns:
Dict containing account states with connector balances and token information
"""
await accounts_service.update_account_state(skip_gateway=filter_request.skip_gateway)
# Only refresh balances if explicitly requested
if filter_request.refresh:
await accounts_service.update_account_state(
skip_gateway=filter_request.skip_gateway,
account_names=filter_request.account_names,
connector_names=filter_request.connector_names
)

all_states = accounts_service.get_accounts_state()

# Apply account name filter first
if filter_request.account_names:
filtered_states = {}
for account_name in filter_request.account_names:
if account_name in all_states:
filtered_states[account_name] = all_states[account_name]
all_states = filtered_states

# Apply connector filter if specified
if filter_request.connector_names:
for account_name, account_data in all_states.items():
Expand All @@ -54,7 +62,7 @@ async def get_portfolio_state(
filtered_connectors[connector_name] = account_data[connector_name]
# Replace account_data with only filtered connectors
all_states[account_name] = filtered_connectors

return all_states


Expand Down
Loading