darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-09-10 18:23:01 -04:00
Code Issues Packages Projects Releases Wiki Activity

reformat

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-29 14:54:30 -07:00
parent 19dcd7a48b
commit 1ecb747c6d
588 changed files with 34091 additions and 23580 deletions
Download Patch File Download Diff File Expand all files Collapse all files
migrations/versions/0112_add_start_end_dates.py
+32 -23
View File
@@ -12,38 +12,47 @@ from sqlalchemy import text
from app.dao.date_util import get_month_start_and_end_date_in_utc
down_revision = '0111_drop_old_service_flags'
revision = '0112_add_start_end_dates'
down_revision = "0111_drop_old_service_flags"
revision = "0112_add_start_end_dates"
def upgrade():
op.drop_index('uix_monthly_billing', 'monthly_billing')
op.alter_column('monthly_billing', 'month', nullable=True)
op.alter_column('monthly_billing', 'year', nullable=True)
op.add_column('monthly_billing', sa.Column('start_date', sa.DateTime))
op.add_column('monthly_billing', sa.Column('end_date', sa.DateTime))
op.drop_index("uix_monthly_billing", "monthly_billing")
op.alter_column("monthly_billing", "month", nullable=True)
op.alter_column("monthly_billing", "year", nullable=True)
op.add_column("monthly_billing", sa.Column("start_date", sa.DateTime))
op.add_column("monthly_billing", sa.Column("end_date", sa.DateTime))
conn = op.get_bind()
results = conn.execute("Select id, month, year from monthly_billing")
res = results.fetchall()
for x in res:
start_date, end_date = get_month_start_and_end_date_in_utc(
datetime(int(x.year), datetime.strptime(x.month, '%B').month, 1))
input_params = {
"start_date": start_date,
"end_date": end_date,
"x_id": x.id
}
conn.execute(text("update monthly_billing set start_date = :start_date, end_date = :end_date where id = :x_id"),
input_params)
op.alter_column('monthly_billing', 'start_date', nullable=False)
op.alter_column('monthly_billing', 'end_date', nullable=False)
op.create_index(op.f('uix_monthly_billing'), 'monthly_billing', ['service_id', 'start_date', 'notification_type'],
unique=True)
datetime(int(x.year), datetime.strptime(x.month, "%B").month, 1)
)
input_params = {"start_date": start_date, "end_date": end_date, "x_id": x.id}
conn.execute(
text(
"update monthly_billing set start_date = :start_date, end_date = :end_date where id = :x_id"
),
input_params,
)
op.alter_column("monthly_billing", "start_date", nullable=False)
op.alter_column("monthly_billing", "end_date", nullable=False)
op.create_index(
op.f("uix_monthly_billing"),
"monthly_billing",
["service_id", "start_date", "notification_type"],
unique=True,
)
def downgrade():
op.drop_column('monthly_billing', 'start_date')
op.drop_column('monthly_billing', 'end_date')
op.drop_column("monthly_billing", "start_date")
op.drop_column("monthly_billing", "end_date")
op.create_index(op.f('uix_monthly_billing'), 'monthly_billing',
['service_id', 'month', 'year', 'notification_type'], unique=True)
op.create_index(
op.f("uix_monthly_billing"),
"monthly_billing",
["service_id", "month", "year", "notification_type"],
unique=True,
)