mirror of
https://github.com/wassname/catalyst.git
synced 2026-08-07 11:20:19 +08:00
Tested ingestion after refactoring
This commit is contained in:
@@ -141,12 +141,16 @@ class Exchange:
|
||||
|
||||
return symbols
|
||||
|
||||
def get_assets(self, symbols):
|
||||
def get_assets(self, symbols=None):
|
||||
assets = []
|
||||
|
||||
for symbol in symbols:
|
||||
asset = self.get_asset(symbol)
|
||||
assets.append(asset)
|
||||
if symbols is not None:
|
||||
for symbol in symbols:
|
||||
asset = self.get_asset(symbol)
|
||||
assets.append(asset)
|
||||
else:
|
||||
for key in self.assets:
|
||||
assets.append(self.assets[key])
|
||||
|
||||
return assets
|
||||
|
||||
|
||||
@@ -39,8 +39,14 @@ class ExchangeBundle:
|
||||
self._reader = None
|
||||
|
||||
def get_assets(self, include_symbols, exclude_symbols):
|
||||
# TODO: filter assets
|
||||
return self.exchange.assets
|
||||
# TODO: filter exclude symbols assets
|
||||
if include_symbols is not None:
|
||||
include_symbols_list = include_symbols.split(',')
|
||||
|
||||
return self.exchange.get_assets(include_symbols_list)
|
||||
|
||||
else:
|
||||
return self.exchange.get_assets()
|
||||
|
||||
def get_adj_dates(self, start, end):
|
||||
now = pd.Timestamp.utcnow()
|
||||
|
||||
@@ -18,10 +18,11 @@ class ExchangeBundleTestCase:
|
||||
exchange_bundle = ExchangeBundle(
|
||||
exchange_name=exchange_name,
|
||||
data_frequency='minute',
|
||||
include_symbols=None,
|
||||
include_symbols='neo_btc',
|
||||
exclude_symbols=None,
|
||||
start=start,
|
||||
end=end,
|
||||
show_progress=True
|
||||
)
|
||||
exchange_bundle.ingest()
|
||||
pass
|
||||
|
||||
Reference in New Issue
Block a user