pass schema into table constructors.

Browse Source
This commit is contained in:
Friedrich Lindenberg 2015-05-21 22:07:03 +02:00
parent e662d8bde7
commit a95c7be33b
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
dataset/persistence/database.py
View File

@ -191,7 +191,7 @@ class Database(object):
raise DatasetException(
"The primary_type has to be either 'Integer' or 'String'.")
table = SQLATable(table_name, self.metadata)
table = SQLATable(table_name, self.metadata, schema=self.schema)
table.append_column(col)
table.create(self.engine)
self._tables[table_name] = table
@ -215,7 +215,8 @@ class Database(object):
self._acquire()
try:
log.debug("Loading table: %s on %r" % (table_name, self))
table = SQLATable(table_name, self.metadata, autoload=True)
table = SQLATable(table_name, self.metadata,
schema=self.schema, autoload=True)
self._tables[table_name] = table
return Table(self, table)
finally:
@ -226,7 +227,8 @@ class Database(object):
self.metadata = MetaData(schema=self.schema)
self.metadata.bind = self.engine
self.metadata.reflect(self.engine)
self._tables[table_name] = SQLATable(table_name, self.metadata)
self._tables[table_name] = SQLATable(table_name, self.metadata,
schema=self.schema)
return self._tables[table_name]
def get_table(self, table_name, primary_id='id', primary_type='Integer'):