aljaz/tiingo-python
1
1
Fork 0
mirror of https://github.com/hydrosquall/tiingo-python.git synced 2025-12-18 04:14:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add support for multi tickers in list_tickers function

Browse Source
This commit is contained in:
n1rna
2019-10-18 19:06:35 +03:30
parent 5490a3c85c
commit c3394c8cdc
2 changed files with 23 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_tiingo.py
View File

@@ -107,6 +107,21 @@ class TestTickerPrices(TestCase):
frequency="30Min") frequency="30Min")
self.assertGreater(len(prices), 1) self.assertGreater(len(prices), 1)
@vcr.use_cassette('tests/fixtures/list_stock_tickers.yaml')
def test_list_all_stock_tickers(self):
tickers = self._client.list_tickers()
assert len(tickers) > 1
assert any(ticker['assetType' == 'Stock' for ticker in tickers)
assert any(ticker['assetType' == 'Mutual Fund' for ticker in tickers)
assert any(ticker['assetType' == 'ETF' for ticker in tickers)
def test_list_multi_stock_tickers(self):
tickers = self._client.list_tickers(['Stock', 'ETF'])
assert len(tickers) > 1
assert any(ticker['assetType' == 'Stock' for ticker in tickers)
assert any(ticker['assetType' == 'ETF' for ticker in tickers)
assert all(ticker['assetType' != 'Mutual Fund' for ticker in tickers)
@vcr.use_cassette('tests/fixtures/list_stock_tickers.yaml') @vcr.use_cassette('tests/fixtures/list_stock_tickers.yaml')
def test_list_stock_tickers(self): def test_list_stock_tickers(self):
tickers = self._client.list_stock_tickers() tickers = self._client.list_stock_tickers()

13
tiingo/api.py
View File

@@ -94,7 +94,7 @@ class TiingoClient(RestClient):
# TICKER PRICE ENDPOINTS # TICKER PRICE ENDPOINTS
# https://api.tiingo.com/docs/tiingo/daily # https://api.tiingo.com/docs/tiingo/daily
def list_tickers(self, assetType): def list_tickers(self, assetTypes=[]):
"""Return a list of dicts of metadata tickers for all supported tickers """Return a list of dicts of metadata tickers for all supported tickers
of the specified asset type, as well as metadata about each ticker. of the specified asset type, as well as metadata about each ticker.
This includes supported date range, the exchange the ticker is traded This includes supported date range, the exchange the ticker is traded
@@ -108,17 +108,20 @@ class TiingoClient(RestClient):
raw_csv = get_buffer_from_zipfile(zipdata, 'supported_tickers.csv') raw_csv = get_buffer_from_zipfile(zipdata, 'supported_tickers.csv')
reader = csv.DictReader(raw_csv) reader = csv.DictReader(raw_csv)
if not len(assetTypes):
return [row for row in reader]
return [row for row in reader return [row for row in reader
if row.get('assetType') == assetType] if row.get('assetType') in assetTypes]
def list_stock_tickers(self): def list_stock_tickers(self):
return self.list_tickers('Stock') return self.list_tickers(['Stock'])
def list_etf_tickers(self): def list_etf_tickers(self):
return self.list_tickers('ETF') return self.list_tickers(['ETF'])
def list_fund_tickers(self): def list_fund_tickers(self):
return self.list_tickers('Mutual Fund') return self.list_tickers(['Mutual Fund'])
def get_ticker_metadata(self, ticker, fmt='json'): def get_ticker_metadata(self, ticker, fmt='json'):
"""Return metadata for 1 ticker """Return metadata for 1 ticker