Spaces:
Runtime error
Runtime error
File size: 1,419 Bytes
2de3774 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import json
from pathlib import Path
import shutil
class OneButtonPresets:
DEFAULT_OBP_FILE = Path("random_prompt/presets/obp_presets.default")
OBP_FILE = Path("random_prompt/userfiles/obp_presets.json")
CUSTOM_OBP = "Custom..."
RANDOM_PRESET_OBP = "All (random)..."
def __init__(self):
self.opb_presets = self.load_obp_presets()
def load_obp_presets(self):
default_data = self._load_data(self.DEFAULT_OBP_FILE)
data = self._load_data(self.OBP_FILE)
for name, settings in default_data.items():
if name not in data:
data[name] = settings
# Sanity check
for name, settings in data.items():
if settings['subject'] == '------ all':
settings['subject'] = 'all'
self._save_data(self.OBP_FILE, data)
return data
def _load_data(self, file_path):
if not file_path.exists():
shutil.copy(self.DEFAULT_OBP_FILE, file_path)
return json.load(open(file_path))
def _save_data(self, file_path, data):
with open(file_path, "w") as f:
json.dump(data, f, indent=2)
def save_obp_preset(self, perf_options):
with open(self.OBP_FILE, "w") as f:
json.dump(perf_options, f, indent=2)
self.opb_presets = self.load_obp_presets()
def get_obp_preset(self, name):
return self.opb_presets[name]
|