mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-21 07:51:13 -05:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
|
|
"""
|
||
|
|
|
||
|
|
Revision ID: 0014_drop_monthly_billing_cols
|
||
|
|
Revises: 0113_job_created_by_nullable
|
||
|
|
Create Date: 2017-07-27 13:36:37.304344
|
||
|
|
|
||
|
|
"""
|
||
|
|
from alembic import op
|
||
|
|
import sqlalchemy as sa
|
||
|
|
from sqlalchemy.dialects import postgresql
|
||
|
|
|
||
|
|
revision = '0014_drop_monthly_billing_cols'
|
||
|
|
down_revision = '0113_job_created_by_nullable'
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade():
|
||
|
|
op.drop_index('uix_monthly_billing', table_name='monthly_billing')
|
||
|
|
op.create_unique_constraint(
|
||
|
|
'uix_monthly_billing', 'monthly_billing', ['service_id', 'start_date', 'notification_type']
|
||
|
|
)
|
||
|
|
op.drop_column('monthly_billing', 'year')
|
||
|
|
op.drop_column('monthly_billing', 'month')
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade():
|
||
|
|
op.add_column('monthly_billing', sa.Column('month', sa.VARCHAR(), autoincrement=False, nullable=True))
|
||
|
|
op.add_column(
|
||
|
|
'monthly_billing',
|
||
|
|
sa.Column('year', postgresql.DOUBLE_PRECISION(precision=53), autoincrement=False, nullable=True)
|
||
|
|
)
|
||
|
|
op.drop_constraint('uix_monthly_billing', 'monthly_billing', type_='unique')
|
||
|
|
op.create_index(
|
||
|
|
'uix_monthly_billing', 'monthly_billing', ['service_id', 'start_date', 'notification_type'], unique=True
|
||
|
|
)
|