Rooba
/
mitogen
mirror of https://github.com/mitogen-hq/mitogen.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

issue #106: get FileService working.

This commit is contained in:
David Wilson 2018-04-01 11:32:29 +01:00
parent 6aac37e157
commit 1a040cf5c0
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ansible_mitogen/services.py
View File

@ -128,17 +128,21 @@ class FileService(mitogen.service.Service):
isinstance(args, tuple) and isinstance(args, tuple) and
len(args) == 2 and len(args) == 2 and
args[0] in ('register', 'fetch') and args[0] in ('register', 'fetch') and
isinstance(args[1], str) isinstance(args[1], basestring)
) )
def dispatch(self, args, msg): def dispatch(self, args, msg):
cmd, path = msg cmd, path = args
return getattr(self, cmd)(path, msg) return getattr(self, cmd)(path, msg)
def register(self, path, msg): def register(self, path, msg):
if not mitogen.core.has_parent_authority(msg): if not mitogen.core.has_parent_authority(msg):
raise mitogen.core.CallError(self.unprivileged_msg) raise mitogen.core.CallError(self.unprivileged_msg)
if path in self._paths:
return
LOG.info('%r: registering %r', self, path)
with open(path, 'rb') as fp: with open(path, 'rb') as fp:
self._paths[path] = zlib.compress(fp.read()) self._paths[path] = zlib.compress(fp.read())