From e6824dc3ff9b4d6b0866ca7e2f3b19cc6ec1520d Mon Sep 17 00:00:00 2001
From: Pea Tyczynska <pea.tyczynska@digital.cabinet-office.gov.uk>
Date: Wed, 2 Dec 2020 18:34:17 +0000
Subject: [PATCH] Broadcast provider message created with a sequential number

This is for the IBAG format (similar to CAP format, but proprietary)
used in the XMLs that we exchange with broadcast providers (specifically
Vodafone).
---
 app/dao/broadcast_message_dao.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/app/dao/broadcast_message_dao.py b/app/dao/broadcast_message_dao.py
index 3c05f2ab9..281e5b09a 100644
--- a/app/dao/broadcast_message_dao.py
+++ b/app/dao/broadcast_message_dao.py
@@ -1,6 +1,14 @@
+import uuid
+
 from app import db
 from app.dao.dao_utils import transactional
-from app.models import BroadcastMessage, BroadcastEvent, BroadcastProviderMessage, BroadcastProviderMessageStatus
+from app.models import (
+    BroadcastMessage,
+    BroadcastEvent,
+    BroadcastProviderMessage,
+    BroadcastProviderMessageNumber,
+    BroadcastProviderMessageStatus
+)
 
 
 def dao_get_broadcast_message_by_id_and_service_id(broadcast_message_id, service_id):
@@ -36,10 +44,18 @@ def get_earlier_events_for_broadcast_event(broadcast_event_id):
 
 @transactional
 def create_broadcast_provider_message(broadcast_event, provider):
+    broadcast_provider_message_id = uuid.uuid4()
     provider_message = BroadcastProviderMessage(
+        id=broadcast_provider_message_id,
         broadcast_event=broadcast_event,
         provider=provider,
         status=BroadcastProviderMessageStatus.SENDING,
     )
     db.session.add(provider_message)
+    db.session.commit()
+
+    provider_message_number = BroadcastProviderMessageNumber(
+        broadcast_provider_message_id=broadcast_provider_message_id)
+    db.session.add(provider_message_number)
+    db.session.commit()
     return provider_message