Apply black

Author: Code0x58

jsonstore/__init__.py

@@ -10,61 +10,61 @@
 from collections import OrderedDict
 from copy import deepcopy
 
-__all__ = ['JsonStore']
+__all__ = ["JsonStore"]
 
 
 class JsonStore(object):
     """A class to provide object based access to a JSON file"""
+
     def __enter__(self):
-        current_state = self.__dict__['_data']
-        self.__dict__['_states'].append(current_state)
-        self.__dict__['_data'] = deepcopy(current_state)
+        current_state = self.__dict__["_data"]
+        self.__dict__["_states"].append(current_state)
+        self.__dict__["_data"] = deepcopy(current_state)
         return self
 
     def __exit__(self, *args):
-        previous_state = self.__dict__['_states'].pop()
+        previous_state = self.__dict__["_states"].pop()
         if any(args):
-            self.__dict__['_data'] = previous_state
-        elif not self.__dict__['_states']:
+            self.__dict__["_data"] = previous_state
+        elif not self.__dict__["_states"]:
             self._save()
 
     def _do_auto_commit(self):
-        if self._auto_commit and not self.__dict__['_states']:
+        if self._auto_commit and not self.__dict__["_states"]:
             self._save()
 
     def _load(self):
         if not os.path.exists(self._path):
-            with open(self._path, 'w+b') as store:
-                store.write('{}'.encode('utf-8'))
-        with open(self._path, 'r+b') as store:
-            raw_data = store.read().decode('utf-8')
+            with open(self._path, "w+b") as store:
+                store.write("{}".encode("utf-8"))
+        with open(self._path, "r+b") as store:
+            raw_data = store.read().decode("utf-8")
         if not raw_data:
             data = OrderedDict()
         else:
             data = json.loads(raw_data, object_pairs_hook=OrderedDict)
 
         if not isinstance(data, dict):
             raise ValueError("Root element is not an object")
-        self.__dict__['_data'] = data
+        self.__dict__["_data"] = data
 
     def _save(self):
-        temp = self._path + '~'
-        with open(temp, 'wb') as store:
-            output = json.dumps(
-                self._data,
-                indent=self._indent,
-                )
-            store.write(output.encode('utf-8'))
+        temp = self._path + "~"
+        with open(temp, "wb") as store:
+            output = json.dumps(self._data, indent=self._indent)
+            store.write(output.encode("utf-8"))
         os.rename(temp, self._path)
 
     def __init__(self, path, indent=2, auto_commit=True):
-        self.__dict__.update({
-            '_auto_commit': auto_commit,
-            '_data': None,
-            '_path': path,
-            '_indent': indent,
-            '_states': [],
-        })
+        self.__dict__.update(
+            {
+                "_auto_commit": auto_commit,
+                "_data": None,
+                "_path": path,
+                "_indent": indent,
+                "_states": [],
+            }
+        )
         self._load()
 
     def __getattr__(self, key):
@@ -90,7 +90,7 @@ def _valid_object(cls, obj, parents=None):
             return all(
                 cls._valid_string(k) and cls._valid_object(v, parents)
                 for k, v in obj.items()
-                )
+            )
         elif isinstance(obj, (list, tuple)):
             return all(cls._valid_object(o, parents) for o in obj)
         else:
@@ -100,7 +100,7 @@ def _valid_object(cls, obj, parents=None):
     def _valid_value(cls, value):
         if isinstance(value, (bool, int, float, type(None))):
             return True
-        elif sys.version_info < (3, ) and isinstance(value, long):
+        elif sys.version_info < (3,) and isinstance(value, long):
             return True
         else:
             return cls._valid_string(value)
@@ -109,7 +109,7 @@ def _valid_value(cls, value):
     def _valid_string(cls, value):
         if isinstance(value, str):
             return True
-        elif sys.version_info < (3, ):
+        elif sys.version_info < (3,):
             return isinstance(value, unicode)
         else:
             return False
@@ -127,7 +127,7 @@ def __get_obj(self, full_path):
         """
         Returns the object which is under the given path
         """
-        steps = full_path.split('.')
+        steps = full_path.split(".")
         path = []
         obj = self._data
         if not full_path:
@@ -137,11 +137,11 @@ def __get_obj(self, full_path):
             try:
                 obj = obj[step]
             except KeyError:
-                raise KeyError('.'.join(path))
+                raise KeyError(".".join(path))
         return obj
 
     def __setitem__(self, name, value):
-        path, _, key = name.rpartition('.')
+        path, _, key = name.rpartition(".")
         if self._valid_object(value):
             dictionary = self.__get_obj(path)
             dictionary[key] = deepcopy(value)
@@ -156,7 +156,7 @@ def __getitem__(self, key):
         return deepcopy(obj)
 
     def __delitem__(self, name):
-        path, _, key = name.rpartition('.')
+        path, _, key = name.rpartition(".")
         obj = self.__get_obj(path)
         del obj[key]