diff options
author | Pete Hollobon <phollobon@renshawbay.com> | 2015-06-04 15:12:09 +0100 |
---|---|---|
committer | Pete Hollobon <phollobon@renshawbay.com> | 2015-06-04 15:12:09 +0100 |
commit | b03ee45f32e53416ce1c73fa15966db17ed4bbf9 (patch) | |
tree | f4b308a868841804adee29bfc508ac6e57101908 | |
parent | dff81500b161cf41ff5d9e77c21a24010e1bd93b (diff) | |
download | sqlalchemy-b03ee45f32e53416ce1c73fa15966db17ed4bbf9.tar.gz |
Add reflection of PostgreSQL index storage options
-rw-r--r-- | lib/sqlalchemy/dialects/postgresql/base.py | 12 | ||||
-rw-r--r-- | test/dialect/postgresql/test_reflection.py | 20 |
2 files changed, 29 insertions, 3 deletions
diff --git a/lib/sqlalchemy/dialects/postgresql/base.py b/lib/sqlalchemy/dialects/postgresql/base.py index e0926a852..b9690e214 100644 --- a/lib/sqlalchemy/dialects/postgresql/base.py +++ b/lib/sqlalchemy/dialects/postgresql/base.py @@ -2624,7 +2624,8 @@ class PGDialect(default.DefaultDialect): SELECT i.relname as relname, ix.indisunique, ix.indexprs, ix.indpred, - a.attname, a.attnum, NULL, ix.indkey%s + a.attname, a.attnum, NULL, ix.indkey%s, + i.reloptions FROM pg_class t join pg_index ix on t.oid = ix.indrelid @@ -2651,7 +2652,8 @@ class PGDialect(default.DefaultDialect): SELECT i.relname as relname, ix.indisunique, ix.indexprs, ix.indpred, - a.attname, a.attnum, c.conrelid, ix.indkey::varchar + a.attname, a.attnum, c.conrelid, ix.indkey::varchar, + i.reloptions FROM pg_class t join pg_index ix on t.oid = ix.indrelid @@ -2680,7 +2682,7 @@ class PGDialect(default.DefaultDialect): sv_idx_name = None for row in c.fetchall(): - idx_name, unique, expr, prd, col, col_num, conrelid, idx_key = row + idx_name, unique, expr, prd, col, col_num, conrelid, idx_key, options = row if expr: if idx_name != sv_idx_name: @@ -2706,6 +2708,8 @@ class PGDialect(default.DefaultDialect): index['unique'] = unique if conrelid is not None: index['duplicates_constraint'] = idx_name + if options: + index['options'] = dict([option.split("=") for option in options]) result = [] for name, idx in indexes.items(): @@ -2716,6 +2720,8 @@ class PGDialect(default.DefaultDialect): } if 'duplicates_constraint' in idx: entry['duplicates_constraint'] = idx['duplicates_constraint'] + if 'options' in idx: + entry.setdefault('dialect_options', {})["postgresql_with"] = idx['options'] result.append(entry) return result diff --git a/test/dialect/postgresql/test_reflection.py b/test/dialect/postgresql/test_reflection.py index 0ebe68cba..70b81219e 100644 --- a/test/dialect/postgresql/test_reflection.py +++ b/test/dialect/postgresql/test_reflection.py @@ -673,6 +673,26 @@ class ReflectionTest(fixtures.TestBase): conn.close() @testing.provide_metadata + def test_index_reflection_with_storage_options(self): + """reflect indexes with storage options set""" + + metadata = self.metadata + + t1 = Table('t', metadata, + Column('id', Integer, primary_key=True), + Column('x', Integer) + ) + metadata.create_all() + conn = testing.db.connect().execution_options(autocommit=True) + conn.execute("CREATE INDEX idx1 ON t (x) WITH (fillfactor = 50)") + + ind = testing.db.dialect.get_indexes(conn, "t", None) + eq_(ind, [{'unique': False, 'column_names': ['x'], 'name': 'idx1', + 'dialect_options': {"postgresql_with": {"fillfactor": "50"}}}]) + conn.close() + + + @testing.provide_metadata def test_foreign_key_option_inspection(self): metadata = self.metadata Table( |