diff options
Diffstat (limited to 'alembic/runtime')
| -rw-r--r-- | alembic/runtime/environment.py | 3 | ||||
| -rw-r--r-- | alembic/runtime/migration.py | 3 |
2 files changed, 4 insertions, 2 deletions
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, |