From 22851c4ef04e2401cfed7c3012b823344115ca7b Mon Sep 17 00:00:00 2001
From: Rebecca Law <rebecca.law@digital.cabinet-office.gov.uk>
Date: Tue, 10 Dec 2019 16:24:18 +0000
Subject: [PATCH] Migration script to populate the returned letters.

---
 .../0311_populate_returned_letters.py         | 30 +++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 migrations/versions/0311_populate_returned_letters.py

diff --git a/migrations/versions/0311_populate_returned_letters.py b/migrations/versions/0311_populate_returned_letters.py
new file mode 100644
index 000000000..a423c5757
--- /dev/null
+++ b/migrations/versions/0311_populate_returned_letters.py
@@ -0,0 +1,30 @@
+"""
+
+Revision ID: 0311_populate_returned_letters
+Revises: 0310_returned_letters_table
+Create Date: 2019-12-09 12:13:49.432993
+
+"""
+from alembic import op
+
+from app.dao.returned_letters_dao import insert_or_update_returned_letters
+
+revision = '0311_populate_returned_letters'
+down_revision = '0310_returned_letters_table'
+
+
+def upgrade():
+    conn = op.get_bind()
+    sql = """
+        select id, service_id, reference 
+        from notification_history 
+        where notification_type = 'letter'
+        and notification_status = 'returned-letter'"""
+    results = conn.execute(sql)
+    returned_letters = results.fetchall()
+    references = [x.reference for x in returned_letters]
+    insert_or_update_returned_letters(references)
+
+
+def downgrade():
+    pass