67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
# (c) 2012-2014, Michael DeHaan <michael.dehaan@gmail.com>
|
|
#
|
|
# This file is part of Ansible
|
|
#
|
|
# Ansible is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Ansible is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
# Make coding more python3-ish
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
from multiprocessing.managers import SyncManager, BaseProxy
|
|
from ansible.playbook.handler import Handler
|
|
from ansible.playbook.task import Task
|
|
from ansible.playbook.play import Play
|
|
from ansible.errors import AnsibleError
|
|
|
|
__all__ = ['AnsibleManager']
|
|
|
|
|
|
class VariableManagerWrapper:
|
|
'''
|
|
This class simply acts as a wrapper around the VariableManager class,
|
|
since manager proxies expect a new object to be returned rather than
|
|
any existing one. Using this wrapper, a shared proxy can be created
|
|
and an existing VariableManager class assigned to it, which can then
|
|
be accessed through the exposed proxy methods.
|
|
'''
|
|
|
|
def __init__(self):
|
|
self._vm = None
|
|
|
|
def get_vars(self, loader, play=None, host=None, task=None):
|
|
return self._vm.get_vars(loader=loader, play=play, host=host, task=task)
|
|
|
|
def set_variable_manager(self, vm):
|
|
self._vm = vm
|
|
|
|
def set_host_variable(self, host, varname, value):
|
|
self._vm.set_host_variable(host, varname, value)
|
|
|
|
def set_host_facts(self, host, facts):
|
|
self._vm.set_host_facts(host, facts)
|
|
|
|
class AnsibleManager(SyncManager):
|
|
'''
|
|
This is our custom manager class, which exists only so we may register
|
|
the new proxy below
|
|
'''
|
|
pass
|
|
|
|
AnsibleManager.register(
|
|
typeid='VariableManagerWrapper',
|
|
callable=VariableManagerWrapper,
|
|
)
|
|
|