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

ansible: support become:True and become_user

This commit is contained in:
David Wilson 2018-02-15 00:27:23 +05:45
parent 8204dc69e7
commit 6059e0c1db
2 changed files with 19 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
ansible_mitogen/connection/mitogen.py
View File

@ -54,7 +54,7 @@ class Connection(ansible.plugins.connection.ConnectionBase):
router = None router = None
context = None context = None
become_methods = [] become_methods = ['sudo']
transport = 'mitogen' transport = 'mitogen'
@property @property
@ -66,10 +66,23 @@ class Connection(ansible.plugins.connection.ConnectionBase):
return return
self.router, self.parent = mitogen.unix.connect('/tmp/mitosock') self.router, self.parent = mitogen.unix.connect('/tmp/mitosock')
self.context = mitogen.service.call(self.parent, 500, { host = mitogen.service.call(self.parent, 500, {
'method': 'ssh',
'hostname': self._play_context.remote_addr, 'hostname': self._play_context.remote_addr,
'username': self._play_context.remote_user,
'port': self._play_context.port,
'ssh_path': self._play_context.ssh_executable,
}) })
if not self._play_context.become:
self.context = host
else:
self.context = mitogen.service.call(self.parent, 500, {
'method': 'sudo',
'username': self._play_context.become_user,
'via': host,
})
def py_call(self, func, *args, **kwargs): def py_call(self, func, *args, **kwargs):
self._connect() self._connect()
return self.context.call(func, *args, **kwargs) return self.context.call(func, *args, **kwargs)

7
ansible_mitogen/strategy/mitogen.py
View File

@ -49,13 +49,14 @@ class ContextProxyService(mitogen.service.Service):
self._context_by_id = {} self._context_by_id = {}
def validate_args(self, args): def validate_args(self, args):
return (isinstance(args, dict) and return isinstance(args, dict)
isinstance(args.get('hostname'), basestring))
def dispatch(self, dct, msg): def dispatch(self, dct, msg):
print dct.get('via')
key = repr(sorted(dct.items())) key = repr(sorted(dct.items()))
if key not in self._context_by_id: if key not in self._context_by_id:
self._context_by_id[key] = self.router.ssh(**dct) method = getattr(self.router, dct.pop('method'))
self._context_by_id[key] = method(**dct)
return self._context_by_id[key] return self._context_by_id[key]