From 4d188e2a4a719869f00a2955263acef3dc609e13 Mon Sep 17 00:00:00 2001
From: Myned <alphasea14@gmail.com>
Date: Thu, 21 Dec 2017 22:39:20 -0500
Subject: [PATCH] Added json to setdefault helper method

---
 src/utils/utils.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/utils/utils.py b/src/utils/utils.py
index 19a5948..5ca1346 100644
--- a/src/utils/utils.py
+++ b/src/utils/utils.py
@@ -36,7 +36,8 @@ except FileNotFoundError:
             'FILE NOT FOUND : config.json created with abstract values. Restart run.py with correct values')
 
 
-def setdefault(filename, default=None):
+def setdefault(filename, default=None, json=False):
+    if not json:
     try:
         with open(filename, 'rb') as infile:
             print('LOADED : {}'.format(filename))
@@ -47,6 +48,17 @@ def setdefault(filename, default=None):
             pkl.dump(default, iofile)
             iofile.seek(0)
             return pkl.load(iofile)
+    else:
+        try:
+            with open(filename, 'rb') as infile:
+                print(f'LOADED : {filename}')
+                return jsn.load(infile)
+        except FileNotFoundError:
+            with open(filename, 'wb+') as iofile:
+                print(f'FILE NOT FOUND : {filename} created and loaded with default values')
+                jsn.dump(default, iofile)
+                iofile.seek(0)
+                return jsn.load(iofile)
 
 
 def load(filename, *, json=False):