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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
import fnmatch
import pickle
import pprint
import threading
class ArloStorage(object):
def __init__(self, arlo):
self._arlo = arlo
self._state_file = self._arlo.cfg.state_file
self.db = {}
self.lock = threading.Lock()
self.load()
def _ekey(self, key):
return key if not isinstance(key, list) else "/".join(key)
def _keys_matching(self, key):
mkeys = []
ekey = self._ekey(key)
for mkey in self.db:
if fnmatch.fnmatch(mkey, ekey):
mkeys.append(mkey)
return mkeys
def load(self):
if self._state_file is not None:
try:
with self.lock:
with open(self._state_file, "rb") as dump:
self.db = pickle.load(dump)
except Exception:
self._arlo.debug("storage: file not read")
def save(self):
if self._state_file is not None:
try:
with self.lock:
with open(self._state_file, "wb") as dump:
pickle.dump(self.db, dump)
except Exception:
self._arlo.warning("storage: file not written")
def file_name(self):
return self._state_file
def get(self, key, default=None):
with self.lock:
ekey = self._ekey(key)
return self.db.get(ekey, default)
def get_matching(self, key, default=None):
with self.lock:
gets = []
for mkey in self._keys_matching(key):
gets.append((mkey, self.db.get(mkey, default)))
return gets
def keys_matching(self, key):
with self.lock:
return self._keys_matching(key)
def set(self, key, value, prefix=""):
ekey = self._ekey(key)
output = "set:" + ekey + "=" + str(value)
self._arlo.debug(f"{prefix}: {output[:80]}")
with self.lock:
self.db[ekey] = value
return value
def unset(self, key):
with self.lock:
del self.db[self._ekey(key)]
def clear(self):
with self.lock:
self.db = {}
def dump(self):
with self.lock:
pprint.pprint(self.db)
|
