From 0459bbfeb2f944029892aad3ee73914d49c2a4fe Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Tue, 28 Mar 2023 16:52:43 -1000 Subject: [PATCH] generate in stmt --- homeassistant/components/sql/sensor.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/sql/sensor.py b/homeassistant/components/sql/sensor.py index 911e5550f60d..e00d01692a29 100644 --- a/homeassistant/components/sql/sensor.py +++ b/homeassistant/components/sql/sensor.py @@ -6,7 +6,7 @@ import decimal import logging import sqlalchemy -from sqlalchemy import TextClause, lambda_stmt +from sqlalchemy import lambda_stmt from sqlalchemy.engine import Result from sqlalchemy.exc import SQLAlchemyError from sqlalchemy.orm import Session, scoped_session, sessionmaker @@ -206,8 +206,9 @@ def _validate_and_get_session_maker_for_db_url(db_url: str) -> scoped_session | sess.close() -def _generate_lambda_stmt(text: TextClause) -> StatementLambdaElement: +def _generate_lambda_stmt(query: str) -> StatementLambdaElement: """Generate the lambda statement.""" + text = sqlalchemy.text(query) return lambda_stmt(lambda: text, lambda_cache=_SQL_LAMBDA_CACHE) @@ -243,7 +244,7 @@ class SQLSensor(SensorEntity): self._attr_extra_state_attributes = {} self._attr_unique_id = unique_id self._use_database_executor = use_database_executor - self._lambda_stmt = _generate_lambda_stmt(sqlalchemy.text(query)) + self._lambda_stmt = _generate_lambda_stmt(query) if not yaml and unique_id: self._attr_device_info = DeviceInfo( entry_type=DeviceEntryType.SERVICE,