mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-11 15:52:21 -05:00
93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
"""empty message
|
|
|
|
Revision ID: 0039_fix_notifications
|
|
Revises: 0038_test_api_key_type
|
|
Create Date: 2016-07-06 13:28:48.381278
|
|
|
|
"""
|
|
|
|
# revision identifiers, used by Alembic.
|
|
from sqlalchemy import text
|
|
|
|
revision = "0039_fix_notifications"
|
|
down_revision = "0038_test_api_key_type"
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
|
|
def upgrade():
|
|
op.execute(
|
|
"update notifications set notification_type = (select cast(cast(template_type as text) as notification_type) from templates where templates.id= notifications.template_id)"
|
|
)
|
|
conn = op.get_bind()
|
|
reset_counts = (
|
|
"update notification_statistics set emails_requested = 0, emails_delivered = 0, emails_failed=0,"
|
|
"sms_requested = 0, sms_delivered = 0, sms_failed=0 where day > '2016-06-30'"
|
|
)
|
|
op.execute(reset_counts)
|
|
all_notifications = text(
|
|
"select * from notifications where date(created_at) > '2016-06-30' order by created_at;"
|
|
)
|
|
|
|
results = conn.execute(all_notifications)
|
|
res = results.fetchall()
|
|
|
|
for x in res:
|
|
created = x.created_at.strftime("%Y-%m-%d")
|
|
input_params = {"created": created, "service_id": x.service_id}
|
|
if x.notification_type == "email" and x.status == "delivered":
|
|
sql = text(
|
|
"update notification_statistics set emails_requested = emails_requested + 1, "
|
|
"emails_delivered = emails_delivered + 1 where day = date(:created) and service_id = :service_id"
|
|
)
|
|
if x.notification_type == "sms" and x.status == "delivered":
|
|
sql = text(
|
|
"update notification_statistics set sms_requested = sms_requested + 1, "
|
|
"sms_delivered = sms_delivered + 1 where day = date(:created) and service_id = :service_id"
|
|
)
|
|
if x.notification_type == "email" and x.status in [
|
|
"technical-failure",
|
|
"temporary-failure",
|
|
"permanent-failure",
|
|
]:
|
|
sql = text(
|
|
"update notification_statistics set emails_requested = emails_requested + 1, "
|
|
"emails_failed = emails_failed + 1 where day = date(:created) and service_id = :service_id"
|
|
)
|
|
if x.notification_type == "sms" and x.status in [
|
|
"technical-failure",
|
|
"temporary-failure",
|
|
"permanent-failure",
|
|
]:
|
|
sql = text(
|
|
"update notification_statistics set sms_requested = sms_requested + 1, "
|
|
"sms_failed = sms_failed + 1 where day = date(:created) and service_id = :service_id"
|
|
)
|
|
if x.notification_type == "email" and x.status in [
|
|
"created",
|
|
"sending",
|
|
"pending",
|
|
]:
|
|
sql = text(
|
|
"update notification_statistics set emails_requested = emails_requested + 1 "
|
|
" where day = date(:created) and service_id = :service_id"
|
|
)
|
|
if x.notification_type == "sms" and x.status in [
|
|
"created",
|
|
"sending",
|
|
"pending",
|
|
]:
|
|
sql = text(
|
|
"update notification_statistics set sms_requested = sms_requested + 1 "
|
|
" where day = date(:created) and service_id = :service_id"
|
|
)
|
|
print(sql)
|
|
conn.execute(sql, input_params)
|
|
|
|
|
|
def downgrade():
|
|
### commands auto generated by Alembic - please adjust! ###
|
|
pass
|
|
### end Alembic commands ###
|