wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-08-06 13:00:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

BUG: Avoid NOT NULL violation by checking for data first

Browse Source
This commit is contained in:
Stewart Douglas
2015-09-10 11:53:26 -04:00
committed by jfkirk
parent e5875af784
commit 667f0695d5
1 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/assets/asset_writer.py
+14 -12
View File
@@ -121,20 +121,22 @@ class AssetDBWriter(with_metaclass(ABCMeta)):
self._write_equities(data.equities, fuzzy_char, txn)
def _write_exchanges(self, exchanges, bind=None):
self.futures_exchanges.insert().values(
exchanges.reset_index().rename_axis(
{'index': 'exchange_id'},
1,
).to_dict('records'),
).execute(bind=bind)
recs = exchanges.reset_index().rename_axis(
{'index': 'exchange_id'},
1,
).to_dict('records')
# In SQLAlchemy, insert().values([]) will insert NULLs,
# hence we check first to avoid violating NOT NULL constraints.
if recs:
self.futures_exchanges.insert().values(recs).execute(bind=bind)
def _write_root_symbols(self, root_symbols, bind=None):
self.futures_root_symbols.insert().values(
root_symbols.reset_index().rename_axis(
{'index': 'root_symbol_id'},
1,
).to_dict('records'),
).execute(bind=bind)
recs = root_symbols.reset_index().rename_axis(
{'index': 'root_symbol_id'},
1,
).to_dict('records')
if recs:
self.futures_root_symbols.insert().values(recs).execute(bind=bind)
def _write_futures(self, futures, bind=None):
recs = futures.reset_index().rename_axis(