7

Use constants for enabled/disabling status on Segment

This commit is contained in:
Michael van Tellingen
2017-05-31 14:38:58 +02:00
parent 3d920d8ed8
commit 31f8a329f2
6 changed files with 22 additions and 18 deletions

View File

@ -97,10 +97,10 @@ def toggle(request, segment_id):
if request.user.has_perm('wagtailadmin.access_admin'):
segment = get_object_or_404(Segment, pk=segment_id)
if segment.status == 'enabled':
segment.status = 'disabled'
elif segment.status == 'disabled':
segment.status = 'enabled'
if segment.status == Segment.STATUS_ENABLED:
segment.status = Segment.STATUS_DISABLED
elif segment.status == Segment.STATUS_DISABLED:
segment.status = Segment.STATUS_ENABLED
segment.save()