New upstream version 0.9.9upstream/0.9.9upstream/latest

1 files changed, 55 insertions, 0 deletions

diff --git a/src/config.py b/src/config.py
new file mode 100644
index 0000000..85b405b
--- /dev/null
+++ b/src/config.py
@@ -0,0 +1,55 @@
+import configparser
+import distutils
+import distutils.util
+import os
+from pathlib import Path
+
+
+class config:
+    # Class constants
+    default_config = {
+        "deploy_type": "quick",
+        "last_paired_device": "None",
+        "paired": "False",
+        "adapter": "None",
+    }
+    config_dir = os.environ.get("XDG_CONFIG_HOME") or os.path.join(
+        os.path.expanduser("~"), ".config"
+    )
+    config_file = os.path.join(config_dir, "siglo.ini")
+
+    def load_defaults(self):
+        if not Path(self.config_dir).is_dir():
+            Path.mkdir(Path(self.config_dir))
+        # if config file is not valid, load defaults
+        if not self.file_valid():
+            config = configparser.ConfigParser()
+            config["settings"] = self.default_config
+            with open(self.config_file, "w") as f:
+                config.write(f)
+
+    def file_valid(self):
+        if not Path(self.config_file).is_file():
+            return False
+        else:
+            config = configparser.ConfigParser()
+            config.read(self.config_file)
+            for key in list(self.default_config.keys()):
+                if not key in config["settings"]:
+                    return False
+            return True
+
+    def get_property(self, key):
+        config = configparser.ConfigParser()
+        config.read(self.config_file)
+        prop = config["settings"][key]
+        if key == "paired":
+            prop = bool(distutils.util.strtobool(prop))
+        return prop
+
+    def set_property(self, key, val):
+        config = configparser.ConfigParser()
+        config.read(self.config_file)
+        config["settings"][key] = val
+        with open(self.config_file, "w") as f:
+            config.write(f)