diff options
Diffstat (limited to 'lib/sqlalchemy')
| -rw-r--r-- | lib/sqlalchemy/engine/ddl.py | 4 | ||||
| -rw-r--r-- | lib/sqlalchemy/schema.py | 10 |
2 files changed, 8 insertions, 6 deletions
diff --git a/lib/sqlalchemy/engine/ddl.py b/lib/sqlalchemy/engine/ddl.py index dca183f9a..79958baae 100644 --- a/lib/sqlalchemy/engine/ddl.py +++ b/lib/sqlalchemy/engine/ddl.py @@ -51,7 +51,7 @@ class SchemaGenerator(DDLBase): tables = metadata.tables.values() collection = [t for t in sql_util.sort_tables(tables) if self._can_create_table(t)] - seq_coll = [s for s in metadata._sequences + seq_coll = [s for s in metadata._sequences.values() if s.column is None and self._can_create_sequence(s)] metadata.dispatch.before_create(metadata, self.connection, @@ -112,7 +112,7 @@ class SchemaDropper(DDLBase): tables = metadata.tables.values() collection = [t for t in reversed(sql_util.sort_tables(tables)) if self._can_drop_table(t)] - seq_coll = [s for s in metadata._sequences + seq_coll = [s for s in metadata._sequences.values() if s.column is None and self._can_drop_sequence(s)] metadata.dispatch.before_drop(metadata, self.connection, diff --git a/lib/sqlalchemy/schema.py b/lib/sqlalchemy/schema.py index c8d00f407..1da43c6b1 100644 --- a/lib/sqlalchemy/schema.py +++ b/lib/sqlalchemy/schema.py @@ -1486,6 +1486,7 @@ class Sequence(DefaultGenerator): self.quote = quote self.schema = schema self.metadata = metadata + self._key = _get_table_key(name, schema) if metadata: self._set_metadata(metadata) @@ -1520,7 +1521,7 @@ class Sequence(DefaultGenerator): def _set_metadata(self, metadata): self.metadata = metadata - self.metadata._sequences.add(self) + self.metadata._sequences[self._key] = self @property def bind(self): @@ -2099,7 +2100,7 @@ class MetaData(SchemaItem): """ self.tables = util.immutabledict() self._schemas = set() - self._sequences = set() + self._sequences = {} self.bind = bind if reflect: if not bind: @@ -2131,12 +2132,13 @@ class MetaData(SchemaItem): if t.schema is not None]) def __getstate__(self): - return {'tables': self.tables, 'schemas':self._schemas} + return {'tables': self.tables, 'schemas':self._schemas, + 'sequences':self._sequences} def __setstate__(self, state): self.tables = state['tables'] self._bind = None - self._sequences = set() + self._sequences = state['sequences'] self._schemas = state['schemas'] def is_bound(self): |