Source code for oioioi.maintenancemode.models

from django.db import models


[docs]class MaintenanceModeInfo(models.Model):
    """Represents maintenance state of the site. 'state'
    specifies if it's on or off. 'message' contains
    message to users (cause of the maintenance)
    """

[docs]    state = models.BooleanField(default=False)
[docs]    message = models.TextField(default="")


[docs]def get_maintenance_mode():
    info, _ = MaintenanceModeInfo.objects.get_or_create()
    return {'state': info.state, 'message': info.message}


[docs]def set_maintenance_mode(new_state, new_message=''):
    MaintenanceModeInfo.objects.update_or_create(
        {'state': new_state, 'message': new_message}
    )


[docs]def is_maintenance_mode_enabled():
    info, _ = MaintenanceModeInfo.objects.get_or_create()
    return info.state