alembic_postgresql_enum_auto_migration_env_config.py

1# env.py
2from alembic_postgresql_enum import ConfigState
3import target_metadata_module # Import your SQLAlchemy metadata here
4
5# ... existing Alembic env.py code ...
6
7def run_migrations_online():
8    # ... existing setup ...
9    
10    connectable = engine_from_config(
11        config.get_section(config.config_ini_section),
12        prefix="sqlalchemy.",
13        poolclass=pool.NullPool,
14    )
15
16    with connectable.connect() as connection:
17        context.configure(
18            connection=connection,
19            target_metadata=target_metadata_module.metadata,
20            # This is the required configuration for alembic-postgresql-enum
21            process_revision_directives=ConfigState(
22                target_metadata_module.metadata,
23            ).process_revision_directives
24        )
25
26        with context.begin_transaction():
27            context.run_migrations()