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

ENH: Add init_db to transaction so table exists=>it has also been written to

Browse Source
This commit is contained in:
Stewart Douglas
2015-08-13 10:14:38 -04:00
committed by jfkirk
parent 667f0695d5
commit aa110dd149
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/assets/asset_writer.py
+6 -4
View File
@@ -110,11 +110,13 @@ class AssetDBWriter(with_metaclass(ABCMeta)):
"""
self.allow_sid_assignment = allow_sid_assignment
# Create SQL tables
self.init_db(engine, constraints)
# Get the data to add to SQL
data = self.load_data()
with engine.begin() as txn:
# Create SQL tables.
self.init_db(txn, constraints)
# Get the data to add to SQL.
data = self.load_data()
# Write the data to SQL.
self._write_exchanges(data.exchanges, txn)
self._write_root_symbols(data.root_symbols, txn)
self._write_futures(data.futures, txn)