set maintenance mode via admin

Author: mkovalua

admin/maintenance/views.py

@@ -1,7 +1,7 @@
 import pytz
 import datetime
 
-from osf.models import MaintenanceState
+from osf.models import MaintenanceState, MaintenanceMode
 import website.maintenance as maintenance
 from admin.maintenance.forms import MaintenanceForm
 
@@ -36,15 +36,17 @@ def get_context_data(self, **kwargs):
         maintenance = MaintenanceState.objects.first()
         kwargs['form'] = MaintenanceForm()
         kwargs['current_alert'] = model_to_dict(maintenance) if maintenance else None
+        kwargs['maintenance_mode'] = MaintenanceMode.is_under_maintenance()
         return super().get_context_data(**kwargs)
 
     def post(self, request, *args, **kwargs):
         data = request.POST
-
-        start = convert_eastern_to_utc(data['start']).isoformat() if data.get('start') else None
-        end = convert_eastern_to_utc(data['end']).isoformat() if data.get('end') else None
-
-        maintenance.set_maintenance(data.get('message', ''), data['level'], start, end)
+        if maintenance_mode := data.get('maintenance_mode'):
+            MaintenanceMode(maintenance_mode=False if maintenance_mode == 'True' else True).save()
+        else:
+            start = convert_eastern_to_utc(data['start']).isoformat() if data.get('start') else None
+            end = convert_eastern_to_utc(data['end']).isoformat() if data.get('end') else None
+            maintenance.set_maintenance(data.get('message', ''), data['level'], start, end)
         return redirect('maintenance:display')
 
 

admin/templates/maintenance/display.html

@@ -66,6 +66,21 @@ <h4>Put up an alert:</h4>
                 </form>
             </div>
         </div>
+
+
+    <div class="row">
+         <div class="col-md-9" style="margin-top: 15px;">
+         <form action="" method="post">
+             {% csrf_token %}
+             <input type="hidden" name="maintenance_mode" value={{maintenance_mode}}>
+             {% if maintenance_mode %}
+             <input class="btn btn-success" type="submit" value="Turn off Maintenance Mode" />
+             {% else %}
+             <input class="btn btn-danger" type="submit" value="Turn on Maintenance Mode" />
+             {% endif %}
+         </form>
+         </div>
+    </div>
     {% endif %}
 
 {% endblock content %}