Import Generator from typing_extensions (1) (#118986)

This commit is contained in:
Marc Mueller
2024-06-06 17:02:13 +02:00
committed by GitHub
parent 69708db8e0
commit fe21e2b8ba
34 changed files with 134 additions and 118 deletions

View File

@ -2,7 +2,7 @@
from __future__ import annotations
from collections.abc import Callable, Generator, Sequence
from collections.abc import Callable, Sequence
import contextlib
from contextlib import contextmanager
from datetime import date, datetime, timedelta
@ -25,6 +25,7 @@ from sqlalchemy.exc import OperationalError, SQLAlchemyError, StatementError
from sqlalchemy.orm.query import Query
from sqlalchemy.orm.session import Session
from sqlalchemy.sql.lambdas import StatementLambdaElement
from typing_extensions import Generator
import voluptuous as vol
from homeassistant.core import HomeAssistant, callback
@ -118,7 +119,7 @@ def session_scope(
session: Session | None = None,
exception_filter: Callable[[Exception], bool] | None = None,
read_only: bool = False,
) -> Generator[Session, None, None]:
) -> Generator[Session]:
"""Provide a transactional scope around a series of operations.
read_only is used to indicate that the session is only used for reading
@ -714,7 +715,7 @@ def periodic_db_cleanups(instance: Recorder) -> None:
@contextmanager
def write_lock_db_sqlite(instance: Recorder) -> Generator[None, None, None]:
def write_lock_db_sqlite(instance: Recorder) -> Generator[None]:
"""Lock database for writes."""
assert instance.engine is not None
with instance.engine.connect() as connection: