From 57223ee224d35a2e99a7dc82668a1e7b7ea041d8 Mon Sep 17 00:00:00 2001 From: Michael Pivato Date: Thu, 17 Jan 2019 14:49:31 +1030 Subject: [PATCH] Add vote property to submit vote and change null values of init parameters --- DecisionSystem/messages.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/DecisionSystem/messages.py b/DecisionSystem/messages.py index 2a8a8c1..34016c3 100644 --- a/DecisionSystem/messages.py +++ b/DecisionSystem/messages.py @@ -1,7 +1,7 @@ import umsgpack class Message: - def __init__(self, sender = None, data = None): + def __init__(self, sender = "", data = {}}): self._sender = sender self._data = data @@ -59,4 +59,14 @@ class CommanderWill(Message): _type = "disconnectedcommander" class SubmitVote(Message): - _type = "vote" \ No newline at end of file + _type = "vote" + + @property + def vote(self): + return self._data["vote"] + + @vote.setter + def vote(self, value): + self._data["vote"] = value + + \ No newline at end of file