Files
notifications-api/migrations/versions/0204_service_data_retention.py
2018-07-11 10:09:33 +01:00

36 lines
1.5 KiB
Python

"""
Revision ID: 0204_service_data_retention
Revises: 0203_fix_old_incomplete_jobs
Create Date: 2018-07-10 11:22:01.761829
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
revision = '0204_service_data_retention'
down_revision = '0203_fix_old_incomplete_jobs'
def upgrade():
op.create_table('service_data_retention',
sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
sa.Column('service_id', postgresql.UUID(as_uuid=True), nullable=False),
sa.Column('notification_type', postgresql.ENUM(name='notification_type', create_type=False),
nullable=False),
sa.Column('days_of_retention', sa.Integer(), nullable=False),
sa.Column('created_at', sa.DateTime(), nullable=False),
sa.Column('updated_at', sa.DateTime(), nullable=True),
sa.ForeignKeyConstraint(['service_id'], ['services.id'], ),
sa.PrimaryKeyConstraint('id'),
sa.UniqueConstraint('service_id', 'notification_type', name='uix_service_data_retention')
)
op.create_index(op.f('ix_service_data_retention_service_id'), 'service_data_retention', ['service_id'],
unique=False)
def downgrade():
op.drop_index(op.f('ix_service_data_retention_service_id'), table_name='service_data_retention')
op.drop_table('service_data_retention')