Merge pull request #1653 from normanjaeckel/FixMotionPoll

Fixed changes in motion poll serializer.
This commit is contained in:
Oskar Hahn 2015-10-21 23:15:28 +02:00
commit 0e3688fc33
1 changed files with 15 additions and 3 deletions

View File

@ -106,13 +106,25 @@ class MotionPollSerializer(ModelSerializer):
'votes',)
def get_yes(self, obj):
return obj.get_votes().get(value='Yes').weight
try:
result = obj.get_votes().get(value='Yes').weight
except obj.get_vote_class().DoesNotExist:
result = None
return result
def get_no(self, obj):
return obj.get_votes().get(value='No').weight
try:
result = obj.get_votes().get(value='No').weight
except obj.get_vote_class().DoesNotExist:
result = None
return result
def get_abstain(self, obj):
return obj.get_votes().get(value='Abstain').weight
try:
result = obj.get_votes().get(value='Abstain').weight
except obj.get_vote_class().DoesNotExist:
result = None
return result
@transaction.atomic
def update(self, instance, validated_data):