darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-24 01:11:38 -05:00
Code Issues Packages Projects Releases Wiki Activity

Don't drop the columns yet

Browse Source
This commit is contained in:
Rebecca Law
2017-07-26 14:46:40 +01:00
parent 90ef82ef92
commit 5669d0475f
1 changed files with 0 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
migrations/versions/0112_add_start_end_dates.py
View File

@@ -29,24 +29,11 @@ def upgrade():
x.id))
op.alter_column('monthly_billing', 'start_date', nullable=False)
op.alter_column('monthly_billing', 'end_date', nullable=False)
op.drop_column('monthly_billing', 'month')
op.drop_column('monthly_billing', 'year')
op.create_index(op.f('uix_monthly_billing'), 'monthly_billing', ['service_id', 'start_date', 'notification_type'],
unique=True)
def downgrade():
op.add_column('monthly_billing', sa.Column('month', sa.String(), nullable=True))
op.add_column('monthly_billing', sa.Column('year', sa.Float(), nullable=True))
conn = op.get_bind()
results = conn.execute("Select id, start_date, end_date from monthly_billing")
res = results.fetchall()
for x in res:
year = datetime.strftime(x.end_date, '%Y')
month = datetime.strftime(x.end_date, '%B')
conn.execute("update monthly_billing set month = '{}', year = {} where id = '{}'".format(month, year, x.id))
op.drop_column('monthly_billing', 'start_date')
op.drop_column('monthly_billing', 'end_date')