109 lines
3.3 KiB
Python
109 lines
3.3 KiB
Python
#!/usr/bin/python2.7
|
|
"""
|
|
src/settingsmixin.py
|
|
====================
|
|
|
|
"""
|
|
|
|
from PyQt4 import QtCore, QtGui
|
|
|
|
|
|
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.toByteArray() if hasattr(geom, 'toByteArray') else 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.toByteArray() if hasattr(state, 'toByteArray') else state)
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
class SMainWindow(QtGui.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(QtGui.QTableWidget, SettingsMixin):
|
|
"""Table widget with Settings functionality"""
|
|
# pylint: disable=too-many-ancestors
|
|
def loadSettings(self):
|
|
"""Load table settings."""
|
|
self.readState(self.horizontalHeader())
|
|
|
|
def saveSettings(self):
|
|
"""Save table settings."""
|
|
self.writeState(self.horizontalHeader())
|
|
|
|
|
|
class SSplitter(QtGui.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(QtGui.QTreeWidget, SettingsMixin):
|
|
"""Tree widget with settings functionality."""
|
|
# pylint: disable=too-many-ancestors
|
|
def loadSettings(self):
|
|
"""Load tree settings."""
|
|
# recurse children
|
|
# self.readState(self)
|
|
pass
|
|
|
|
def saveSettings(self):
|
|
"""Save tree settings"""
|
|
# recurse children
|
|
# self.writeState(self)
|
|
pass
|