Save current status

Author: ElusiveSpirit

django_dev_protector/app_data.json

@@ -0,0 +1 @@
+{"status": false}

django_dev_protector/middleware.py

@@ -16,7 +16,7 @@ def process_request(self, request):
             if data['key'] == settings.SECRET_KEY:
                 from .setup import save_status
                 environ[PROTECT_STATUS_VARIABLE] = str(data['status'])
-                save_status(str(data['status']))
+                save_status(data['status'])
                 return redirect('/')
 
         if environ.get(PROTECT_STATUS_VARIABLE) == 'True':

django_dev_protector/protect_status.conf

@@ -6,7 +6,7 @@
 PROTECT_STATUS_DEFAULT = False
 
 PROTECT_STATUS_VARIABLE = 'PROTECT_STATUS_VARIABLE'
-PROTECT_STATUS_FILE = '/protect_status.conf'
+PROTECT_STATUS_FILE = '/app_data.json'
 
 TEMPLATE_NAME = 'django_dev_protector/index.html' if not hasattr(settings, 'PROTECT_TEMPLATE_NAME') else settings.PROTECT_TEMPLATE_NAME
 

django_dev_protector/settings.py

@@ -1,43 +1,32 @@
 import os
+import json
 
 from .settings import (PROTECT_STATUS_FILE, PROTECT_STATUS_VARIABLE,
                        PROTECT_STATUS_DEFAULT, DIR_NAME)
 
 STATUS_FILE_NAME = DIR_NAME + PROTECT_STATUS_FILE
 
 
-def save_to_file(text, file_name):
+def save_status(status):
+    """Save current status to file"""
     try:
-        f = open(file_name, 'w')
-        f.write(str(text))
+        f = open(STATUS_FILE_NAME, 'w')
+        f.write(json.dumps({
+            'status': status
+        }))
         f.close()
-    except:
+    except (NameError, FileNotFoundError):
         pass
 
 
-def get_from_file(file_name):
-    """Gets first word from file"""
+def get_status():
+    """Get status from file"""
     try:
-        f = open(file_name, 'r')
+        f = open(STATUS_FILE_NAME, 'r')
         data = f.read()
         f.close()
-
-        import re
-        data = re.findall('\w+', data)
-        return data[0]
-    except:
-        return ''
-
-
-def save_status(status):
-    """Save current status to file"""
-    save_to_file(status, STATUS_FILE_NAME)
-
-
-def get_status():
-    """Get status from file"""
-    res = get_from_file(STATUS_FILE_NAME)
-    if not res:
+        res = json.loads(data)['status']
+    except (NameError, FileNotFoundError):
         save_status(PROTECT_STATUS_DEFAULT)
-        return str(PROTECT_STATUS_DEFAULT)
-    return res
+        res = (PROTECT_STATUS_DEFAULT)
+    return str(res)