Fix PK reflection

Author: xzkostyan

CHANGELOG.md

@@ -1,13 +1,17 @@
 # Changelog
 
 ## [Unreleased]
+## [0.2.8] - 2025-01-30
+### Fixed
+- Table primary key reflection uses ``engine_reflection`` option. Enabled by default reflection leads to speed penalty during `system.tables` lookup.
+
 ## [0.2.7] - 2024-10-18
 ### Fixed
 - ClickHouse dialect extensions (FINAL, SAMPLE, LIMIT BY etc.) rendering in SQLAlchemy version 1.4.
 - Unhashable type dict caused by full join. Solves issue [#276](https://github.com/xzkostyan/clickhouse-sqlalchemy/issues/276).
 
 ## [0.2.6] - 2024-03-25
-### Fixed (backported from 00.3.x)
+### Fixed (backported from 0.3.x)
 - [alembic] Table reflection for alembic version < 1.11. Solves issue [#274](https://github.com/xzkostyan/clickhouse-sqlalchemy/issues/274).
 
 ## [0.2.5] - 2023-10-29
@@ -334,7 +338,8 @@ Log, TinyLog, Null.
 - Chunked `INSERT INTO` in one request.
 - Engines: MergeTree, CollapsingMergeTree, SummingMergeTree, Buffer, Memory. 
 
-[Unreleased]: https://github.com/xzkostyan/clickhouse-sqlalchemy/compare/0.2.7...HEAD
+[Unreleased]: https://github.com/xzkostyan/clickhouse-sqlalchemy/compare/0.2.8...HEAD
+[0.2.8]: https://github.com/xzkostyan/clickhouse-sqlalchemy/compare/0.2.7...0.2.8
 [0.2.7]: https://github.com/xzkostyan/clickhouse-sqlalchemy/compare/0.2.6...0.2.7
 [0.2.6]: https://github.com/xzkostyan/clickhouse-sqlalchemy/compare/0.2.5...0.2.6
 [0.2.5]: https://github.com/xzkostyan/clickhouse-sqlalchemy/compare/0.2.4...0.2.5

clickhouse_sqlalchemy/__init__.py

@@ -4,7 +4,7 @@
 from .sql import Table, MaterializedView, select
 
 
-VERSION = (0, 2, 7)
+VERSION = (0, 2, 8)
 __version__ = '.'.join(str(x) for x in VERSION)
 
 

clickhouse_sqlalchemy/drivers/base.py

@@ -360,7 +360,11 @@ def get_foreign_keys(self, connection, table_name, schema=None, **kw):
 
     @reflection.cache
     def get_pk_constraint(self, connection, table_name, schema=None, **kw):
-        if not self.supports_engine_reflection:
+        should_reflect = (
+            self.supports_engine_reflection and
+            self.engine_reflection
+        )
+        if not should_reflect:
             return {}
 
         if schema: