holetron-lab/fleet-memory
1
0
Fork 0
Code Releases Packages Activity Actions

fix migration

Browse source
This commit is contained in:
Nicolò Boschi 2025-12-04 21:44:33 +01:00
parent ab10162d51
commit f69ea6ee61
2 changed files with 25 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
hindsight-api/alembic/versions/5a366d414dce_initial_schema.py
View file

@ -31,7 +31,7 @@ def upgrade() -> None:
'banks', 'banks',
sa.Column('bank_id', sa.Text(), nullable=False), sa.Column('bank_id', sa.Text(), nullable=False),
sa.Column('name', sa.Text(), nullable=True), sa.Column('name', sa.Text(), nullable=True),
sa.Column('disposition', postgresql.JSONB(astext_type=sa.Text()), server_default=sa.text("'{}'::jsonb"), nullable=False), sa.Column('personality', postgresql.JSONB(astext_type=sa.Text()), server_default=sa.text("'{}'::jsonb"), nullable=False),
sa.Column('background', sa.Text(), nullable=True), sa.Column('background', sa.Text(), nullable=True),
sa.Column('created_at', postgresql.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False), sa.Column('created_at', postgresql.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
sa.Column('updated_at', postgresql.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False), sa.Column('updated_at', postgresql.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),

27
hindsight-api/alembic/versions/rename_personality_to_disposition.py
View file

@ -9,6 +9,7 @@ from typing import Sequence, Union
from alembic import op from alembic import op
import sqlalchemy as sa import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic. # revision identifiers, used by Alembic.
@ -20,16 +21,36 @@ depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None: def upgrade() -> None:
"""Rename personality column to disposition in banks table (if it exists).""" """Rename personality column to disposition in banks table (if it exists)."""
# Check if 'personality' column exists before renaming
# This handles both old databases (with personality) and new databases (with disposition)
conn = op.get_bind() conn = op.get_bind()
# Check if 'personality' column exists (old database)
result = conn.execute(sa.text(""" result = conn.execute(sa.text("""
SELECT column_name SELECT column_name
FROM information_schema.columns FROM information_schema.columns
WHERE table_name = 'banks' AND column_name = 'personality' WHERE table_name = 'banks' AND column_name = 'personality'
""")) """))
if result.fetchone(): has_personality = result.fetchone() is not None
# Check if 'disposition' column exists (new database)
result = conn.execute(sa.text("""
SELECT column_name
FROM information_schema.columns
WHERE table_name = 'banks' AND column_name = 'disposition'
"""))
has_disposition = result.fetchone() is not None
if has_personality and not has_disposition:
# Old database: rename personality -> disposition
op.alter_column('banks', 'personality', new_column_name='disposition') op.alter_column('banks', 'personality', new_column_name='disposition')
elif not has_personality and not has_disposition:
# Neither exists (shouldn't happen, but be safe): add disposition column
op.add_column('banks', sa.Column(
'disposition',
postgresql.JSONB(astext_type=sa.Text()),
server_default=sa.text("'{}'::jsonb"),
nullable=False
))
# else: disposition already exists, nothing to do
def downgrade() -> None: def downgrade() -> None: