forked from M-Labs/artiq
1
0
Fork 0

gui: add state manager

This commit is contained in:
Sebastien Bourdeauducq 2015-08-01 16:48:27 +08:00
parent 4a7a4acf07
commit 9e24b56099
1 changed files with 63 additions and 0 deletions

63
artiq/gui/state.py Normal file
View File

@ -0,0 +1,63 @@
import asyncio
from collections import OrderedDict
from artiq.tools import TaskObject
from artiq.protocols import pyon
# support Qt CamelCase naming scheme for save/restore state
def _save_state(obj):
method = getattr(obj, "save_state", None)
if method is None:
method = obj.saveState
return method()
def _restore_state(obj, state):
method = getattr(obj, "restore_state", None)
if method is None:
method = obj.restoreState
method(state)
class StateManager(TaskObject):
def __init__(self, filename, autosave_period=30):
self.filename = filename
self.autosave_period = autosave_period
self.stateful_objects = OrderedDict()
def register(self, obj, name=None):
if name is None:
name = obj.__class__.__name__
if name in self.stateful_objects:
raise RuntimeError("Name '{}' already exists in state"
.format(name))
self.stateful_objects[name] = obj
def load(self):
try:
data = pyon.load_file(self.filename)
except FileNotFoundError:
return
# The state of one object may depend on the state of another,
# e.g. the display state may create docks that are referenced in
# the area state.
# To help address this problem, state is restored in the opposite
# order as the stateful objects are registered.
for name, obj in reversed(list(self.stateful_objects.items())):
state = data.get(name, None)
if state is not None:
_restore_state(obj, state)
def save(self):
data = {k: _save_state(v) for k, v in self.stateful_objects.items()}
pyon.store_file(self.filename, data)
@asyncio.coroutine
def _do(self):
try:
while True:
yield from asyncio.sleep(self.autosave_period)
self.save()
finally:
self.save()