Skip to content

Commit

Permalink
Fixed set_index() and set_tickers() markets...
Browse files Browse the repository at this point in the history
  • Loading branch information
shner-elmo committed Nov 24, 2024
1 parent 7da5747 commit 9af99e2
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/tradingview_screener/query.py
Original file line number Diff line number Diff line change
Expand Up @@ -424,6 +424,7 @@ def set_tickers(self, *tickers: str) -> Self:
:return: Self
"""
self.query.setdefault('symbols', {})['tickers'] = list(tickers)
self.set_markets()
return self

def set_index(self, *indexes: str) -> Self:
Expand Down Expand Up @@ -471,7 +472,8 @@ def set_index(self, *indexes: str) -> Self:
"""
self.query.setdefault('preset', 'index_components_market_pages')
self.query.setdefault('symbols', {})['symbolset'] = list(indexes)
self.url = URL.format(market='global')
# reset markets list and URL to `/global`
self.set_markets()
return self

# def set_currency(self, currency: Literal['symbol', 'market'] | str) -> Self:
Expand Down Expand Up @@ -585,7 +587,6 @@ def __eq__(self, other) -> bool:
return isinstance(other, Query) and self.query == other.query and self.url == other.url


# TODO: should get_scanner_data() return the raw data instead of DF?
# TODO: Query should have no defaults (except limit), and a separate module should have all the
# default screeners
# TODO: add all presets

0 comments on commit 9af99e2

Please sign in to comment.