diff options
Diffstat (limited to 'alembic')
-rw-r--r-- | alembic/context.pyi | 3 | ||||
-rw-r--r-- | alembic/runtime/environment.py | 3 | ||||
-rw-r--r-- | alembic/runtime/migration.py | 3 |
3 files changed, 6 insertions, 3 deletions
diff --git a/alembic/context.pyi b/alembic/context.pyi index 142a0c7..a262638 100644 --- a/alembic/context.pyi +++ b/alembic/context.pyi @@ -17,6 +17,7 @@ from typing import Union if TYPE_CHECKING: from sqlalchemy.engine.base import Connection + from sqlalchemy.engine.url import URL from sqlalchemy.sql.elements import ClauseElement from sqlalchemy.sql.schema import MetaData @@ -75,7 +76,7 @@ config: Config def configure( connection: Optional[Connection] = None, - url: Optional[str] = None, + url: Union[str, URL, None] = None, dialect_name: Optional[str] = None, dialect_opts: Optional[Dict[str, Any]] = None, transactional_ddl: Optional[bool] = None, diff --git a/alembic/runtime/environment.py b/alembic/runtime/environment.py index 0f9d3a5..c2fa11a 100644 --- a/alembic/runtime/environment.py +++ b/alembic/runtime/environment.py @@ -20,6 +20,7 @@ from ..operations import Operations if TYPE_CHECKING: from typing import Literal + from sqlalchemy.engine import URL from sqlalchemy.engine.base import Connection from sqlalchemy.sql.elements import ClauseElement from sqlalchemy.sql.schema import MetaData @@ -334,7 +335,7 @@ class EnvironmentContext(util.ModuleClsProxy): def configure( self, connection: Optional[Connection] = None, - url: Optional[str] = None, + url: Optional[Union[str, URL]] = None, dialect_name: Optional[str] = None, dialect_opts: Optional[Dict[str, Any]] = None, transactional_ddl: Optional[bool] = None, diff --git a/alembic/runtime/migration.py b/alembic/runtime/migration.py index a615651..4e2d062 100644 --- a/alembic/runtime/migration.py +++ b/alembic/runtime/migration.py @@ -34,6 +34,7 @@ from ..util.compat import EncodedIO if TYPE_CHECKING: from sqlalchemy.engine import Dialect + from sqlalchemy.engine import URL from sqlalchemy.engine.base import Connection from sqlalchemy.engine.base import Transaction from sqlalchemy.engine.mock import MockConnection @@ -217,7 +218,7 @@ class MigrationContext: def configure( cls, connection: Optional[Connection] = None, - url: Optional[str] = None, + url: Optional[Union[str, URL]] = None, dialect_name: Optional[str] = None, dialect: Optional[Dialect] = None, environment_context: Optional[EnvironmentContext] = None, |