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

ansible: support local connections too

This commit is contained in:
David Wilson 2018-02-16 07:23:52 +05:45
parent 71a6b9e32a
commit 6e7bb4fd13
1 changed files with 29 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
ansible_mitogen/connection/mitogen.py
View File

@ -47,14 +47,13 @@ class Connection(ansible.plugins.connection.ConnectionBase):
def connected(self):
return self.router is not None
def _connect(self):
if self.connected:
return
def _connect_local(self):
return mitogen.service.call(self.parent, 500, {
'method': 'local',
}
path = os.environ['LISTENER_SOCKET_PATH']
self.router, self.parent = mitogen.unix.connect(path)
host = mitogen.service.call(self.parent, 500, cast({
def _connect_ssh(self):
return mitogen.service.call(self.parent, 500, cast({
'method': 'ssh',
'hostname': self._play_context.remote_addr,
'username': self._play_context.remote_user,
@ -64,17 +63,32 @@ class Connection(ansible.plugins.connection.ConnectionBase):
'ssh_path': self._play_context.ssh_executable,
}))
def _connect_sudo(self, via):
return mitogen.service.call(self.parent, 500, cast({
'method': 'sudo',
'username': self._play_context.become_user,
'password': self._play_context.password,
'python_path': '/usr/bin/python',
'via': via,
'debug': True,
}))
def _connect(self):
if self.connected:
return
path = os.environ['LISTENER_SOCKET_PATH']
self.router, self.parent = mitogen.unix.connect(path)
if self._play_context.connection == 'local':
host = self._connect_local()
else:
host = self._connect_ssh(self)
if not self._play_context.become:
self.context = host
else:
self.context = mitogen.service.call(self.parent, 500, cast({
'method': 'sudo',
'username': self._play_context.become_user,
'password': self._play_context.password,
'python_path': '/usr/bin/python',
'via': host,
'debug': True,
}))
self.context = self._connect_sudo(via=host)
def call_async(self, func, *args, **kwargs):
self._connect()