Skip to content

Commit 70e4ffc

Browse files
committed
refac
1 parent 0883638 commit 70e4ffc

1 file changed

Lines changed: 40 additions & 0 deletions

File tree

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
"""add memory path and meta
2+
3+
Revision ID: 42e2978c7933
4+
Revises: 7b3f2a9c1d4e
5+
Create Date: 2026-06-29 05:35:50.565887
6+
7+
"""
8+
9+
from typing import Sequence, Union
10+
11+
from alembic import op
12+
import sqlalchemy as sa
13+
14+
15+
revision: str = '42e2978c7933'
16+
down_revision: Union[str, None] = '7b3f2a9c1d4e'
17+
branch_labels: Union[str, Sequence[str], None] = None
18+
depends_on: Union[str, Sequence[str], None] = None
19+
20+
21+
def upgrade() -> None:
22+
conn = op.get_bind()
23+
inspector = sa.inspect(conn)
24+
columns = {column['name'] for column in inspector.get_columns('memory')}
25+
26+
if 'path' not in columns:
27+
op.add_column('memory', sa.Column('path', sa.Text(), nullable=True))
28+
if 'meta' not in columns:
29+
op.add_column('memory', sa.Column('meta', sa.JSON(), nullable=True))
30+
31+
32+
def downgrade() -> None:
33+
conn = op.get_bind()
34+
inspector = sa.inspect(conn)
35+
columns = {column['name'] for column in inspector.get_columns('memory')}
36+
37+
if 'meta' in columns:
38+
op.drop_column('memory', 'meta')
39+
if 'path' in columns:
40+
op.drop_column('memory', 'path')

0 commit comments

Comments
 (0)