summaryrefslogtreecommitdiff
path: root/lib/utils
diff options
context:
space:
mode:
Diffstat (limited to 'lib/utils')
-rw-r--r--lib/utils/cache.py6
-rwxr-xr-xlib/utils/paths.py11
-rw-r--r--lib/utils/settings.py58
3 files changed, 74 insertions, 1 deletions
diff --git a/lib/utils/cache.py b/lib/utils/cache.py
index 767978ca..1ede8ce1 100644
--- a/lib/utils/cache.py
+++ b/lib/utils/cache.py
@@ -10,6 +10,8 @@ import pickle
import appdirs
import diskcache
+from lib.utils.settings import global_settings
+
try:
from functools import lru_cache
except ImportError:
@@ -29,7 +31,9 @@ def get_stitch_plan_cache():
if __stitch_plan_cache is None:
cache_dir = os.path.join(appdirs.user_config_dir('inkstitch'), 'cache', 'stitch_plan')
- __stitch_plan_cache = diskcache.Cache(cache_dir, size=1024 * 1024 * 100)
+ size_limit = global_settings['cache_size'] * 1024 * 1024
+ __stitch_plan_cache = diskcache.Cache(cache_dir, size=size_limit)
+ __stitch_plan_cache.size_limit = size_limit
atexit.register(__stitch_plan_cache.close)
return __stitch_plan_cache
diff --git a/lib/utils/paths.py b/lib/utils/paths.py
index 2a95f6e7..10d72de9 100755
--- a/lib/utils/paths.py
+++ b/lib/utils/paths.py
@@ -7,6 +7,8 @@ import sys
import os
from os.path import dirname, realpath
+import appdirs
+
def get_bundled_dir(name):
if getattr(sys, 'frozen', None) is not None:
@@ -26,3 +28,12 @@ def get_resource_dir(name):
return realpath(os.path.join(sys._MEIPASS, name))
else:
return realpath(os.path.join(dirname(realpath(__file__)), '..', '..', name))
+
+
+def get_user_dir(name=None):
+ path = appdirs.user_config_dir("inkstitch")
+
+ if name is not None:
+ path = os.path.join(path, name)
+
+ return path
diff --git a/lib/utils/settings.py b/lib/utils/settings.py
new file mode 100644
index 00000000..f2ce276d
--- /dev/null
+++ b/lib/utils/settings.py
@@ -0,0 +1,58 @@
+from collections.abc import MutableMapping
+import json
+import os
+
+from .paths import get_user_dir
+
+# These settings are the defaults for SVG metadata settings of the same name in
+# lib.extensions.base.InkstitchMetadata
+DEFAULT_METADATA = {
+ "min_stitch_len_mm": 0,
+ "collapse_len_mm": 3,
+}
+
+DEFAULT_SETTINGS = {
+ "cache_size": 100
+}
+
+
+class GlobalSettings(MutableMapping):
+ def __init__(self):
+ super().__init__()
+ self.__settings_file = os.path.join(get_user_dir(), "settings.json")
+ self.__settings = {}
+
+ for name, value in DEFAULT_METADATA.items():
+ self.__settings[f"default_{name}"] = value
+
+ self.__settings.update(DEFAULT_SETTINGS)
+
+ try:
+ with open(self.__settings_file, 'r') as settings_file:
+ self.__settings.update(json.load(settings_file))
+ except (OSError, json.JSONDecodeError, ValueError):
+ pass
+
+ def __setitem__(self, item, value):
+ self.__settings[item] = value
+
+ with open(self.__settings_file, 'w') as settings_file:
+ json.dump(self.__settings, settings_file)
+
+ def __getitem__(self, item):
+ return self.__settings[item]
+
+ def __delitem__(self, item):
+ del self.__settings[item]
+
+ def __iter__(self):
+ return iter(self.__settings)
+
+ def __len__(self):
+ return len(self.__settings)
+
+ def __json__(self):
+ return self.__settings
+
+
+global_settings = GlobalSettings()