mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-20 02:16:39 +00:00
fix: doctor looks for alembic_version where alembic puts it
env.py sets no version_table_schema, so the table follows search_path. Pinning public. in doctor's queries made them agree with each other and disagree with alembic: on an install using another schema it reported no schema at all and sent the user to migrate an already-migrated database. Both queries resolve the table the same way alembic does now.
This commit is contained in:
1 parent
bb50d0a91d
commit
48285f4405
2 files changed
+14
-7
No files matched your search
@@ -805,12 +805,14 @@ def _check_postgres(uri: Optional[str]) -> Check:
|
||||
with psycopg.connect(uri, connect_timeout=5) as connection, connection.cursor() as cursor:
|
||||
cursor.execute("select current_setting('server_version')")
|
||||
version = cursor.fetchone()[0]
|
||||
cursor.execute("select to_regclass('public.alembic_version')")
|
||||
# Unqualified, like alembic itself: env.py sets no version_table_schema, so the table
|
||||
# lives wherever search_path puts it. Asserting public would call a migrated database empty.
|
||||
cursor.execute("select to_regclass('alembic_version')")
|
||||
applied = cursor.fetchone()[0] is not None
|
||||
current = None
|
||||
if applied:
|
||||
# public, like the to_regclass check above: search_path could resolve another one.
|
||||
cursor.execute("select version_num from public.alembic_version")
|
||||
# The same relation the check above resolved, by the same rules.
|
||||
cursor.execute("select version_num from alembic_version")
|
||||
row = cursor.fetchone()
|
||||
current = row[0] if row else None
|
||||
except (psycopg.Error, OSError, ValueError) as exc:
|
||||
|
||||
Reference in new issue
Block a user