Source code for oioioi.maintenancemode.management.commands.maintenance_mode

from __future__ import absolute_import

from django.core.management.base import BaseCommand, CommandError
from django.utils.translation import gettext as _

from oioioi.maintenancemode.models import set_maintenance_mode


[docs]class Command(BaseCommand):
[docs]    help = _("Sets maintenance mode state")

[docs]    def add_arguments(self, parser):
        parser.add_argument('mode', type=str, choices=('on', 'off'), help='Mode')
        parser.add_argument('message', nargs='?', default="", type=str)

[docs]    def handle(self, *args, **options):
        state_value = options['mode']

        if state_value == 'on':
            message = options['message']
            set_maintenance_mode(True, message)
        elif state_value == 'off':
            set_maintenance_mode(False)