darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-07-08 10:28:57 -04:00
Code Issues Packages Projects Releases Wiki Activity

Refactor broadcast message model

Browse Source 
There was a lot of duplicate code here for updating the status and other
properties of a broadcast.

This commit abstracts them into reusable methods. They’re named with an
underscore to suggest that they shouldn’t be used externally.
This commit is contained in:
Chris Hill-Scott
2020-07-17 08:07:42 +01:00
parent 73c5aa9bb7
commit 4b1e3ec504
1 changed files with 22 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
app/models/broadcast_message.py
View File

@@ -108,49 +108,39 @@ class BroadcastMessage(JSONModel):
return User.from_id(self.cancelled_by_id)
def add_areas(self, *new_areas):
broadcast_message_api_client.update_broadcast_message(
broadcast_message_id=self.id,
service_id=self.service_id,
data={
'areas': list(OrderedSet(
self._dict['areas'] + list(new_areas)
))
},
)
self._update(areas=list(OrderedSet(
self._dict['areas'] + list(new_areas)
)))
def remove_area(self, area_to_remove):
self._update(areas=[
area for area in self._dict['areas']
if area != area_to_remove
])
def _set_status_to(self, status):
broadcast_message_api_client.update_broadcast_message_status(
status,
broadcast_message_id=self.id,
service_id=self.service_id,
)
def _update(self, **kwargs):
broadcast_message_api_client.update_broadcast_message(
broadcast_message_id=self.id,
service_id=self.service_id,
data={
'areas': [
area for area in self._dict['areas']
if area != area_to_remove
]
},
data=kwargs,
)
def start_broadcast(self):
broadcast_message_api_client.update_broadcast_message(
broadcast_message_id=self.id,
service_id=self.service_id,
data={
'starts_at': datetime.utcnow().isoformat(),
'finishes_at': (datetime.utcnow() + self.DEFAULT_TTL).isoformat(),
},
)
broadcast_message_api_client.update_broadcast_message_status(
'broadcasting',
broadcast_message_id=self.id,
service_id=self.service_id,
self._update(
starts_at=datetime.utcnow().isoformat(),
finishes_at=(datetime.utcnow() + self.DEFAULT_TTL).isoformat(),
)
self._set_status_to('broadcasting')
def cancel_broadcast(self):
broadcast_message_api_client.update_broadcast_message_status(
'cancelled',
broadcast_message_id=self.id,
service_id=self.service_id,
)
self._set_status_to('cancelled')
class BroadcastMessages(ModelList):