39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
"""create fruits table
|
|
Revision ID: 0002
|
|
Revises: 0001
|
|
Create Date: 2024-01-20 10:00:00.000000
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = '0002'
|
|
down_revision = '0001'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
def upgrade():
|
|
op.create_table(
|
|
'fruits',
|
|
sa.Column('id', sa.String(36), primary_key=True),
|
|
sa.Column('name', sa.String(), nullable=False),
|
|
sa.Column('color', sa.String(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP')),
|
|
sa.Column('updated_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'))
|
|
)
|
|
op.create_index(op.f('ix_fruits_name'), 'fruits', ['name'], unique=False)
|
|
|
|
# Create trigger for updated_at
|
|
op.execute('''
|
|
CREATE TRIGGER trig_fruits_updated_at
|
|
AFTER UPDATE ON fruits
|
|
FOR EACH ROW
|
|
BEGIN
|
|
UPDATE fruits SET updated_at = CURRENT_TIMESTAMP
|
|
WHERE id = NEW.id;
|
|
END;
|
|
''')
|
|
|
|
def downgrade():
|
|
op.execute('DROP TRIGGER IF EXISTS trig_fruits_updated_at')
|
|
op.drop_index(op.f('ix_fruits_name'), table_name='fruits')
|
|
op.drop_table('fruits') |