override get_from_hint_text to include FINAL

Author: joe-clickhouse

clickhouse_connect/cc_sqlalchemy/sql/compiler.py

@@ -83,3 +83,8 @@ def visit_empty_set_expr(self, element_types, **kw):
 
     def visit_sequence(self, sequence, **kw):
         raise NotImplementedError("ClickHouse doesn't support sequences")
+
+    def get_from_hint_text(self, table, text):
+        if text == "FINAL":
+            return "FINAL"
+        return super().get_from_hint_text(table, text)

tests/integration_tests/test_sqlalchemy/test_ddl.py

@@ -123,11 +123,8 @@ def test_final_modifier_replacing_merge_tree(test_engine: Engine, test_db: str):
 
         query_with_final = select(test_table).final().order_by(test_table.c.id)
         compiled = query_with_final.compile(dialect=test_engine.dialect)
-
-        # In low-load test envs, the merge usually happens so fast that FINAL doesn't
-        #  seem to make a difference, but the important part is seeing FINAL in the
-        #  actual compiled compiled query.
-        assert "FINAL" in str(compiled).upper()
+        compiled_str = str(compiled)
+        assert " FINAL" in compiled_str
         result = conn.execute(query_with_final)
         rows = result.fetchall()
         assert len(rows) == 2