Include 'p' for get_indexes() query

Added support for reflection of indexes on PostgreSQL partitioned tables,
which was added to PostgreSQL as of version 11.

Fixes: #4771
Change-Id: I9e8e75c4d8a667b4d52d12afbd384e0a8db00466

diff --git a/doc/build/changelog/unreleased_13/4771.rst b/doc/build/changelog/unreleased_13/4771.rst
new file mode 100644 (file)
index 0000000..e40797d
--- /dev/null
+++ b/doc/build/changelog/unreleased_13/4771.rst
@@ -0,0 +1,6 @@
+.. change::
+    :tags: usecase, postgresql
+    :tickets: 4771
+
+    Added support for reflection of indexes on PostgreSQL partitioned tables,
+    which was added to PostgreSQL as of version 11.

diff --git a/lib/sqlalchemy/dialects/postgresql/base.py b/lib/sqlalchemy/dialects/postgresql/base.py
index 5aea02cfcaa3adb26a4931d909b054669ddc4386..8fbd97ebe1f4496f0519b18aa6be267140824c0d 100644 (file)
--- a/lib/sqlalchemy/dialects/postgresql/base.py
+++ b/lib/sqlalchemy/dialects/postgresql/base.py
@@ -3255,7 +3255,7 @@ class PGDialect(default.DefaultDialect):
                             pg_am am
                             on i.relam = am.oid
               WHERE
-                  t.relkind IN ('r', 'v', 'f', 'm')
+                  t.relkind IN ('r', 'v', 'f', 'm', 'p')
                   and t.oid = :table_oid
                   and ix.indisprimary = 'f'
               ORDER BY

diff --git a/test/dialect/postgresql/test_reflection.py b/test/dialect/postgresql/test_reflection.py
index ea72d571056f29101015a3cec9036014515cbb03..6e7d7202ad5f30152af461f948d63dee4b97a8c9 100644 (file)
--- a/test/dialect/postgresql/test_reflection.py
+++ b/test/dialect/postgresql/test_reflection.py
@@ -111,6 +111,7 @@ class PartitionedReflectionTest(fixtures.TablesTest, AssertsExecutionResults):
             metadata,
             Column("modulus", Integer, nullable=False),
             Column("data", String(30)),
+            Column("q", Integer),
             postgresql_partition_by="range(modulus)",
         )
 
@@ -124,6 +125,9 @@ class PartitionedReflectionTest(fixtures.TablesTest, AssertsExecutionResults):
             ),
         )
 
+        if testing.against("postgresql >= 11"):
+            Index("my_index", dv.c.q)
+
     def test_get_tablenames(self):
         assert {"data_values", "data_values_4_10"}.issubset(
             inspect(testing.db).get_table_names()
@@ -131,11 +135,25 @@ class PartitionedReflectionTest(fixtures.TablesTest, AssertsExecutionResults):
 
     def test_reflect_cols(self):
         cols = inspect(testing.db).get_columns("data_values")
-        eq_([c["name"] for c in cols], ["modulus", "data"])
+        eq_([c["name"] for c in cols], ["modulus", "data", "q"])
 
     def test_reflect_cols_from_partition(self):
         cols = inspect(testing.db).get_columns("data_values_4_10")
-        eq_([c["name"] for c in cols], ["modulus", "data"])
+        eq_([c["name"] for c in cols], ["modulus", "data", "q"])
+
+    @testing.only_on("postgresql >= 11")
+    def test_reflect_index(self):
+        idx = inspect(testing.db).get_indexes("data_values")
+        eq_(
+            idx, [{"column_names": ["q"], "name": "my_index", "unique": False}]
+        )
+
+    @testing.only_on("postgresql >= 11")
+    def test_reflect_index_from_partition(self):
+        idx = inspect(testing.db).get_indexes("data_values_4_10")
+        # note the name appears to be generated by PG, currently
+        # 'data_values_4_10_q_idx'
+        eq_(idx, [{"column_names": ["q"], "name": mock.ANY, "unique": False}])
 
 
 class MaterializedViewReflectionTest(