From 7acd0d7a1dadd25df5f3e5ea249eb752ff70561b Mon Sep 17 00:00:00 2001 From: jfkirk Date: Wed, 30 Dec 2015 11:50:48 -0500 Subject: [PATCH] MAINT: Refactors table assignment in AssetDBWriter --- zipline/assets/asset_writer.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/zipline/assets/asset_writer.py b/zipline/assets/asset_writer.py index 0fd4ba72..593a0ea0 100644 --- a/zipline/assets/asset_writer.py +++ b/zipline/assets/asset_writer.py @@ -345,12 +345,8 @@ class AssetDBWriter(with_metaclass(ABCMeta)): tables_already_exist = self.check_for_tables(engine) metadata = generate_asset_db_metadata(bind=engine) - self.version_info = metadata.tables['version_info'] - self.equities = metadata.tables['equities'] - self.futures_exchanges = metadata.tables['futures_exchanges'] - self.futures_root_symbols = metadata.tables['futures_root_symbols'] - self.futures_contracts = metadata.tables['futures_contracts'] - self.asset_router = metadata.tables['asset_router'] + for table_name in asset_db_table_names: + setattr(self, table_name, metadata.tables[table_name]) # Create the SQL tables if they do not already exist. metadata.create_all(checkfirst=True)