blob: 39d0065a4a0bc2ff22fcf2ab03657df55548fa50 (
plain)
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
|
# Copyright (C) 2012 Codethink Limited
#
import json
import os
import logging
class WorkingStateManager(object):
'''Manage the working state of lorry-controller'''
def __init__(self, app):
self.app = app
self.workdir = os.path.join(self.app.settings['work-area'], 'work')
def __enter__(self):
self._load_state()
return self
def __exit__(self, exctype, excvalue, exctraceback):
self.save_state()
def _load_state(self):
self.lorry_state_file = os.path.join(self.workdir,
"last-lorry-state.json")
if os.path.exists(self.lorry_state_file):
logging.debug("Loading state file: %s" % self.lorry_state_file)
with open(self.lorry_state_file, "r") as fh:
self.lorry_state = json.load(fh)
else:
self.lorry_state = dict()
def save_state(self):
logging.debug("Serialising state: %s" % self.lorry_state_file)
with open(self.lorry_state_file, "w") as fh:
json.dump(self.lorry_state, fh, sort_keys=True, indent=4)
fh.write("\n")
|