112 lines
3.1 KiB
Python
112 lines
3.1 KiB
Python
"""
|
|
src/settingsmixin.py
|
|
====================
|
|
|
|
"""
|
|
|
|
from qtpy import QtCore, QtWidgets
|
|
|
|
|
|
class SettingsMixin(object):
|
|
"""Mixin for adding geometry and state saving between restarts"""
|
|
|
|
def warnIfNoObjectName(self):
|
|
"""
|
|
Handle objects which don't have a name. Currently it ignores them. Objects without a name can't have their
|
|
state/geometry saved as they don't have an identifier.
|
|
"""
|
|
if self.objectName() == "":
|
|
# .. todo:: logger
|
|
pass
|
|
|
|
def writeState(self, source):
|
|
"""Save object state (e.g. relative position of a splitter)"""
|
|
self.warnIfNoObjectName()
|
|
settings = QtCore.QSettings()
|
|
settings.beginGroup(self.objectName())
|
|
settings.setValue("state", source.saveState())
|
|
settings.endGroup()
|
|
|
|
def writeGeometry(self, source):
|
|
"""Save object geometry (e.g. window size and position)"""
|
|
self.warnIfNoObjectName()
|
|
settings = QtCore.QSettings()
|
|
settings.beginGroup(self.objectName())
|
|
settings.setValue("geometry", source.saveGeometry())
|
|
settings.endGroup()
|
|
|
|
def readGeometry(self, target):
|
|
"""Load object geometry"""
|
|
self.warnIfNoObjectName()
|
|
settings = QtCore.QSettings()
|
|
try:
|
|
geom = settings.value(
|
|
"/".join([str(self.objectName()), "geometry"]))
|
|
target.restoreGeometry(geom)
|
|
except Exception:
|
|
pass
|
|
|
|
def readState(self, target):
|
|
"""Load object state"""
|
|
self.warnIfNoObjectName()
|
|
settings = QtCore.QSettings()
|
|
try:
|
|
state = settings.value("/".join([str(self.objectName()), "state"]))
|
|
target.restoreState(state)
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
class SMainWindow(QtWidgets.QMainWindow, SettingsMixin):
|
|
"""Main window with Settings functionality"""
|
|
|
|
def loadSettings(self):
|
|
"""Load main window settings."""
|
|
self.readGeometry(self)
|
|
self.readState(self)
|
|
|
|
def saveSettings(self):
|
|
"""Save main window settings"""
|
|
self.writeState(self)
|
|
self.writeGeometry(self)
|
|
|
|
|
|
class STableWidget(QtWidgets.QTableWidget, SettingsMixin):
|
|
"""Table widget with Settings functionality"""
|
|
|
|
def loadSettings(self):
|
|
"""Load table settings."""
|
|
self.readState(self.horizontalHeader())
|
|
|
|
def saveSettings(self):
|
|
"""Save table settings."""
|
|
self.writeState(self.horizontalHeader())
|
|
|
|
|
|
class SSplitter(QtWidgets.QSplitter, SettingsMixin):
|
|
"""Splitter with Settings functionality"""
|
|
|
|
def loadSettings(self):
|
|
"""Load splitter settings"""
|
|
self.readState(self)
|
|
|
|
def saveSettings(self):
|
|
"""Save splitter settings."""
|
|
self.writeState(self)
|
|
|
|
|
|
class STreeWidget(QtWidgets.QTreeWidget, SettingsMixin):
|
|
"""Tree widget with settings functionality"""
|
|
|
|
def loadSettings(self):
|
|
"""Load tree settings. Unimplemented."""
|
|
# recurse children
|
|
# self.readState(self)
|
|
pass
|
|
|
|
def saveSettings(self):
|
|
"""Save tree settings. Unimplemented."""
|
|
# recurse children
|
|
# self.writeState(self)
|
|
pass
|