From 55276a8d87dc4dc6dcaf9bc7fa0aadfc19fb518d Mon Sep 17 00:00:00 2001
From: Abhimanyu Saharan <desk.abhimanyu@gmail.com>
Date: Sat, 13 May 2023 01:41:29 +0530
Subject: [PATCH] adds maintenance exempt paths #11233

---
 netbox/netbox/middleware.py | 14 +++++++++-----
 netbox/netbox/settings.py   |  6 ++++++
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/netbox/netbox/middleware.py b/netbox/netbox/middleware.py
index 461c018b9..7efc0708a 100644
--- a/netbox/netbox/middleware.py
+++ b/netbox/netbox/middleware.py
@@ -181,19 +181,23 @@ class MaintenanceModeMiddleware:
 
     def __call__(self, request):
         if get_config().MAINTENANCE_MODE:
-            self._prevent_db_write_operations()
+            self._prevent_db_write_operations(
+                allow_write=request.path_info.startswith(settings.MAINTENANCE_EXEMPT_PATHS)
+            )
 
         return self.get_response(request)
 
     @staticmethod
-    def _prevent_db_write_operations():
+    def _prevent_db_write_operations(allow_write=False):
         """
         Prevent any write-related database operations.
+
+        Args:
+            allow_write (bool): Whether to allow write operations.
         """
         with connection.cursor() as cursor:
-            cursor.execute(
-                'SET SESSION CHARACTERISTICS AS TRANSACTION READ ONLY;'
-            )
+            mode = 'READ WRITE' if allow_write else 'READ ONLY'
+            cursor.execute(f'SET SESSION CHARACTERISTICS AS TRANSACTION {mode};')
 
     def process_exception(self, request, exception):
         """
diff --git a/netbox/netbox/settings.py b/netbox/netbox/settings.py
index 575755d2b..426456672 100644
--- a/netbox/netbox/settings.py
+++ b/netbox/netbox/settings.py
@@ -478,6 +478,12 @@ AUTH_EXEMPT_PATHS = (
     f'/{BASE_PATH}metrics',
 )
 
+# All URLs starting with a string listed here are exempt from maintenance mode enforcement
+MAINTENANCE_EXEMPT_PATHS = (
+    f'/{BASE_PATH}admin/extras/configrevision/',
+    f'/{BASE_PATH}login/',
+)
+
 SERIALIZATION_MODULES = {
     'json': 'utilities.serializers.json',
 }