class Attr(Entry):
def __init__(self, router=None, core=None, rdonly=True):
Entry.__init__(self, router, core)
self._temp = Temp(self, rdonly)
def _log(self, text):
if LOG_ATTR:
log_debug(self, text)
def may_update(self, flags):
return flags & os.O_APPEND or flags & os.O_RDWR or flags & os.O_TRUNC or flags & os.O_WRONLY
def can_invalidate(self):
return True
def truncate(self, uid, name, length):
if not self._core:
path = self.get_path(uid, name)
self._fs.truncate(uid, path, length)
self._temp.truncate(uid, name, length)
def is_expired(self, uid, name):
temp = get_temp(self.get_path(uid, name))
return self._fs.exist(uid, temp)
def getattr(self, uid, name):
return self.lsattr(uid, name)
def create(self, uid, name):
self.check_path(uid, name)
return self._temp.create(uid, name)
def open(self, uid, name, flags):
if self._core:
flags = 0
return self._temp.open(uid, name, flags)
def release(self, uid, name, fh):
return self._temp.release(uid, name, fh)
def _unlink(self, uid, name):
if not self._core:
return
child = self.child(name)
parent = self.parent(name)
if parent != child:
if child == ATTR_HANDLER:
self._core.remove_handler(parent)
elif child == ATTR_FILTER:
self._core.remove_filter(parent)
elif child == ATTR_DISPATCHER:
self._core.remove_dispatcher(parent)
elif child == ATTR_MODE:
self._core.remove_mode(parent)
elif child == ATTR_FREQ:
self._core.remove_freq(parent)
elif child == ATTR_TIMEOUT:
self._core.remove_timeout(parent)
def unlink(self, uid, name):
self.remove(uid, name)
self._unlink(uid, name)
def invalidate(self, uid, name):
self._log('invalidate, name=%s' % str(name))
path = self.get_path(uid, name)
temp = get_temp(path)
if self._fs.exist(uid, path):
self._fs.rename(uid, path, temp)
self._unlink(uid, name)
else:
self._fs.touch(uid, temp)
def signature(self, uid, name):
res = ''
if RSYNC:
temp = get_temp(self.get_path(uid, name))
with open(temp, 'rb') as f:
sigature = librsync.signature(f)
res = b64encode(sigature.read())
return res
def patch(self, uid, name, buf):
if not buf:
log_warnning(self, 'cannot patch, no content, name=%s' % str(name))
return
dest = self.get_path(uid, name)
src = get_temp(dest)
tmp = b64decode(buf)
if RSYNC:
delta = StringIO(tmp)
with open(dest, 'wb') as f_dest:
with open(src, 'rb') as f_src:
try:
librsync.patch(f_src, delta, f_dest)
except:
log_warnning(self, 'failed to patch, name=%s' % str(name))
self._fs.rename(uid, src, dest)
return
self._fs.remove(uid, src)
else:
with open(src, 'wb') as f:
f.write(tmp)
self._fs.rename(uid, src, dest)
def readdir(self, uid, name):
return self.lsdir(uid, name)
def _create_attr(self, uid, name, attr, val):
error = False
name = os.path.join(name, attr)
f = self.create(uid, name)
try:
os.write(f, str(val))
except:
error = True
finally:
os.close(f)
if error:
self.discard(uid, name)
return
self.commit(uid, name)
def initialize(self, uid, name, attr, val):
if attr not in ATTRIBUTES:
log_err(self, 'failed to initialize, invalid attribute %s, name=%s' % (str(attr), str(name)))
raise Exception(log_get(self, 'failed to initialize'))
if attr == ATTR_PROFILE:
val = json.dumps(val)
self._create_attr(uid, name, attr, val)
def discard(self, uid, name):
return self._temp.discard(uid, name)
def commit(self, uid, name):
return self._temp.commit(uid, name)
class Data(Entry):
def __init__(self, vrtx, edge, attr, router=None, core=None, rdonly=True):
Entry.__init__(self, router, core)
self._temp = Temp(self, rdonly)
self._vrtx = vrtx
self._edge = edge
self._attr = attr
def can_load(self):
return True
def can_scan(self):
return True
def can_enable(self):
return True
def can_disable(self):
return True
def truncate(self, uid, name, length):
path = self.get_path(uid, name)
self._fs.truncate(uid, path, length)
self._temp.truncate(uid, name, length)
def getattr(self, uid, name):
return self.lsattr(uid, name)
def create(self, uid, name):
self.check_path(uid, name)
self._vrtx.check_path(uid, parent=name)
self._edge.check_path(uid, parent=name)
self._attr.check_path(uid, parent=name)
return self._temp.create(uid, name)
def open(self, uid, name, flags):
return self._temp.open(uid, name, flags)
def release(self, uid, name, fh):
return self._temp.release(uid, name, fh)
def _unlink(self, uid, name):
if self._core:
self._core.remove(name)
def unlink(self, uid, name):
self._vrtx.remove(uid, name)
self._edge.unlink(uid, name)
self._attr.unlink(uid, name)
self._temp.remove(uid, name)
self._unlink(uid, name)
self.remove(uid, name)
def readdir(self, uid, name):
if not name:
return self.lsdir(uid, name)
else:
return []
def initialize(self, uid, name):
f = self.create(uid, name)
self.release(uid, name, f)
def discard(self, uid, name):
return self._temp.discard(uid, name)
def commit(self, uid, name):
return self._temp.commit(uid, name)