add alembic include scehmas (#6036)

adhami3310 · web-flow · commit 805ad97f54c1 · 2025-12-11T17:38:15.000-08:00
diff --git a/reflex/environment.py b/reflex/environment.py
@@ -583,6 +583,9 @@ class EnvironmentVariables:
     # Path to the alembic config file
     ALEMBIC_CONFIG: EnvVar[ExistingPath] = env_var(Path(constants.ALEMBIC_CONFIG))
 
+    # Include schemas in alembic migrations.
+    ALEMBIC_INCLUDE_SCHEMAS: EnvVar[bool] = env_var(False)
+
     # Disable SSL verification for HTTPX requests.
     SSL_NO_VERIFY: EnvVar[bool] = env_var(False)
 
diff --git a/reflex/model.py b/reflex/model.py
@@ -518,6 +518,7 @@ def run_autogenerate(rev: str, context: MigrationContext):
                     render_item=cls._alembic_render_item,
                     process_revision_directives=writer,
                     compare_type=False,
+                    include_schemas=environment.ALEMBIC_INCLUDE_SCHEMAS.get(),
                     render_as_batch=True,  # for sqlite compatibility
                 )
                 env.run_migrations()

Original file line number	Diff line number	Diff line change
`@@ -518,6 +518,7 @@ def run_autogenerate(rev: str, context: MigrationContext):`
`518`	`518`	`render_item=cls._alembic_render_item,`
`519`	`519`	`process_revision_directives=writer,`
`520`	`520`	`compare_type=False,`
	`521`	`+ include_schemas=environment.ALEMBIC_INCLUDE_SCHEMAS.get(),`
`521`	`522`	`render_as_batch=True, # for sqlite compatibility`
`522`	`523`	`)`
`523`	`524`	`env.run_migrations()`