inkstitch/lib/gui/presets.py

184 wiersze
6.2 KiB
Python

import json
import os
import re
import wx
from ..i18n import _
from ..utils import cache
from .dialogs import info_dialog
class PresetsPanel(wx.Panel):
"""A wx.Panel for loading, saving, and applying presets.
A preset is a named collection of settings. From the perspective of this
class, a preset is an opaque JSON-serializable object.
The PresetsPanel will handle interaction with the user and inform the
instantiator of events such as a preset being loaded. Presets starting
and ending with "__" will not be shown to the user. This allows for the
instantiator to manage hidden presets such as "__LAST__".
"""
HIDDEN_PRESET_RE = re.compile('^__.*__$')
def __init__(self, parent, *args, **kwargs):
"""Construct a PresetsPanel.
The parent is the parent window for this wx.Panel. The parent is
expected to implement the following methods:
def get_preset_data(self)
returns a JSON object representing the current state as a preset
def apply_preset_data(self, preset_data):
apply the preset data to the GUI, updating GUI elements as necessary
def get_preset_suite_name(self):
Return a string used in the presets filename, e.g. "lettering" -> "lettering_presets.json".
If not defined, "presets.json" will be used.
"""
kwargs.setdefault('style', wx.BORDER_NONE)
wx.Panel.__init__(self, parent, wx.ID_ANY, *args, **kwargs)
self.parent = parent
self.presets_box = wx.StaticBox(self, wx.ID_ANY, label=_("Presets"))
self.preset_chooser = wx.ComboBox(self, wx.ID_ANY)
self.update_preset_list()
self.preset_chooser.SetSelection(-1)
self.load_preset_button = wx.Button(self, wx.ID_ANY, _("Load"))
self.load_preset_button.Bind(wx.EVT_BUTTON, self.load_selected_preset)
self.add_preset_button = wx.Button(self, wx.ID_ANY, _("Add"))
self.add_preset_button.Bind(wx.EVT_BUTTON, self.add_preset)
self.overwrite_preset_button = wx.Button(self, wx.ID_ANY, _("Overwrite"))
self.overwrite_preset_button.Bind(wx.EVT_BUTTON, self.overwrite_preset)
self.delete_preset_button = wx.Button(self, wx.ID_ANY, _("Delete"))
self.delete_preset_button.Bind(wx.EVT_BUTTON, self.delete_preset)
presets_sizer = wx.StaticBoxSizer(self.presets_box, wx.HORIZONTAL)
presets_sizer.Add(self.preset_chooser, 1, wx.ALIGN_CENTER_VERTICAL | wx.ALL, 10)
presets_sizer.Add(self.load_preset_button, 0, wx.ALIGN_CENTER_VERTICAL | wx.ALL, 10)
presets_sizer.Add(self.add_preset_button, 0, wx.ALIGN_CENTER_VERTICAL | wx.ALL, 10)
presets_sizer.Add(self.overwrite_preset_button, 0, wx.ALIGN_CENTER_VERTICAL | wx.ALL, 10)
presets_sizer.Add(self.delete_preset_button, 0, wx.ALIGN_CENTER_VERTICAL | wx.ALL, 10)
self.SetSizerAndFit(presets_sizer)
self.Layout()
@property
@cache
def suite_name(self):
try:
return self.parent.get_preset_suite_name() + "_presets"
except AttributeError:
return "presets"
@cache
def presets_path(self):
try:
import appdirs
config_path = appdirs.user_config_dir('inkstitch')
except ImportError:
config_path = os.path.expanduser('~/.inkstitch')
if not os.path.exists(config_path):
os.makedirs(config_path)
return os.path.join(config_path, '%s.json' % self.suite_name)
def _load_presets(self):
try:
with open(self.presets_path(), 'r') as presets:
presets = json.load(presets)
return presets
except IOError:
return {}
def _save_presets(self, presets):
with open(self.presets_path(), 'w') as presets_file:
json.dump(presets, presets_file)
def update_preset_list(self):
preset_names = self._load_presets().keys()
preset_names = [preset for preset in preset_names if not self.is_hidden(preset)]
self.preset_chooser.SetItems(sorted(preset_names))
def is_hidden(self, preset_name):
return self.HIDDEN_PRESET_RE.match(preset_name)
def get_preset_name(self):
preset_name = self.preset_chooser.GetValue().strip()
if preset_name:
return preset_name
else:
info_dialog(self, _("Please enter or select a preset name first."), caption=_('Preset'))
return
def check_and_load_preset(self, preset_name):
preset = self._load_presets().get(preset_name)
if not preset:
info_dialog(self, _('Preset "%s" not found.') % preset_name, caption=_('Preset'))
return preset
def store_preset(self, preset_name, data):
presets = self._load_presets()
presets[preset_name] = data
self._save_presets(presets)
self.update_preset_list()
def add_preset(self, event, overwrite=False):
preset_name = self.get_preset_name()
if not preset_name:
return
if not overwrite and preset_name in self._load_presets():
info_dialog(self, _('Preset "%s" already exists. Please use another name or press "Overwrite"') % preset_name, caption=_('Preset'))
self.store_preset(self, preset_name, self.parent.get_preset_data())
event.Skip()
def overwrite_preset(self, event):
self.add_preset(event, overwrite=True)
def load_preset(self, preset_name):
preset = self.check_and_load_preset(preset_name)
if not preset:
return
self.parent.apply_preset_data(preset)
def load_selected_preset(self, event):
preset_name = self.get_preset_name()
if not preset_name:
return
self.load_preset(preset_name)
event.Skip()
def delete_preset(self, event):
preset_name = self.get_preset_name()
if not preset_name:
return
preset = self.check_and_load_preset(preset_name)
if not preset:
return
presets = self._load_presets()
presets.pop(preset_name, None)
self._save_presets(presets)
self.update_preset_list()
self.preset_chooser.SetValue("")
event.Skip()