diff options
author | tangkikodo <allmonday@126.com> | 2023-05-11 16:43:05 -0400 |
---|---|---|
committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2023-05-12 12:14:33 -0400 |
commit | 647190e82d4e7b9d7fe24a891af6556ed9d651b8 (patch) | |
tree | 06c89d927879ecf5420d0268fe5c062a58966bbf /alembic | |
parent | 92e54a0e1c96cecd99397cb1aee9c3bb28f780c6 (diff) | |
download | alembic-647190e82d4e7b9d7fe24a891af6556ed9d651b8.tar.gz |
add user_module_prefix param for render_python_code
Fixed issue where :func:`.autogenerate.render_python_code` function did not
provide a default value for the ``user_module_prefix`` variable, leading to
``NoneType`` errors when autogenerate structures included user-defined
types. Added new parameter
:paramref:`.autogenerate.render_python_code.user_module_prefix` to allow
this to be set as well as to default to ``None``. Pull request courtesy
tangkikodo.
Fixes #1235
Closes: #1233
Pull-request: https://github.com/sqlalchemy/alembic/pull/1233
Pull-request-sha: 6fa3bc1ce4f2130bf7a28282a84e3bb9a3223304
Change-Id: Ic21eec87f373a9518513a6a308f16e5d2b5ebf81
Diffstat (limited to 'alembic')
-rw-r--r-- | alembic/autogenerate/api.py | 21 |
1 files changed, 18 insertions, 3 deletions
diff --git a/alembic/autogenerate/api.py b/alembic/autogenerate/api.py index 95fdede..c4ec5c1 100644 --- a/alembic/autogenerate/api.py +++ b/alembic/autogenerate/api.py @@ -6,9 +6,10 @@ from typing import Callable from typing import Dict from typing import Iterator from typing import Optional +from typing import Sequence from typing import Set -from typing import Tuple from typing import TYPE_CHECKING +from typing import Union from sqlalchemy import inspect @@ -28,6 +29,7 @@ if TYPE_CHECKING: from sqlalchemy.sql.schema import SchemaItem from ..config import Config + from ..operations.ops import DowngradeOps from ..operations.ops import MigrationScript from ..operations.ops import UpgradeOps from ..runtime.environment import NameFilterParentNames @@ -195,13 +197,14 @@ def produce_migrations( def render_python_code( - up_or_down_op: UpgradeOps, + up_or_down_op: Union[UpgradeOps, DowngradeOps], sqlalchemy_module_prefix: str = "sa.", alembic_module_prefix: str = "op.", render_as_batch: bool = False, - imports: Tuple[str, ...] = (), + imports: Sequence[str] = (), render_item: Optional[RenderItemFn] = None, migration_context: Optional[MigrationContext] = None, + user_module_prefix: Optional[str] = None, ) -> str: """Render Python code given an :class:`.UpgradeOps` or :class:`.DowngradeOps` object. @@ -209,12 +212,24 @@ def render_python_code( This is a convenience function that can be used to test the autogenerate output of a user-defined :class:`.MigrationScript` structure. + :param up_or_down_op: :class:`.UpgradeOps` or :class:`.DowngradeOps` object + :param sqlalchemy_module_prefix: module prefix for SQLAlchemy objects + :param alembic_module_prefix: module prefix for Alembic constructs + :param render_as_batch: use "batch operations" style for rendering + :param imports: sequence of import symbols to add + :param render_item: callable to render items + :param migration_context: optional :class:`.MigrationContext` + :param user_module_prefix: optional string prefix for user-defined types + + .. versionadded:: 1.11.0 + """ opts = { "sqlalchemy_module_prefix": sqlalchemy_module_prefix, "alembic_module_prefix": alembic_module_prefix, "render_item": render_item, "render_as_batch": render_as_batch, + "user_module_prefix": user_module_prefix, } if migration_context is None: |