Use create dir when saving state to config dir (#90)

skorokithakis · Apr 22, 2018 · 6d86ef7 · 6d86ef7
1 parent 4b141ae
commit 6d86ef7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/catt/cli.py b/catt/cli.py
@@ -290,7 +290,7 @@ def save(settings, path):
     if path and path.exists():
         click.confirm("File already exists. Overwrite?", abort=True)
     click.echo("Saving...")
-    state = CastState(path or STATE_PATH)
+    state = CastState(path or STATE_PATH, create_dir=True if not path else False)
     state.set_data(cst.cc_name, {"controller": cst.name, "data": cst.media_info})
 
 

diff --git a/catt/controllers.py b/catt/controllers.py
@@ -201,9 +201,10 @@ def get_data(self, name):
 
 
 class CastState(CattStore):
-    def __init__(self, state_path):
+    def __init__(self, state_path, create_dir=False):
         super(CastState, self).__init__(state_path)
-
+        if create_dir:
+            self._create_store_dir()
         if not self.store_path.exists():
             self._write_store({})