Merge pull request #1150 from alphagov/imdad-migration-remove-month-year-monthly-billing

Drop unused month and year columns from MonthlyBilling

migrations/versions/0114_drop_monthly_billing_cols.py

@@ -0,0 +1,34 @@
+"""
+
+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
+    )