cmueller/ansible
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

added become check back to connections

Browse source
This commit is contained in:
Brian Coca 2015-06-14 23:45:56 -04:00
parent de82c953f2
commit c3ccf26b70
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
lib/ansible/plugins/connections/__init__.py
View file

@ -63,10 +63,10 @@ class ConnectionBase(with_metaclass(ABCMeta, object)):
if not hasattr(self, '_connected'): if not hasattr(self, '_connected'):
self._connected = False self._connected = False
def _become_method_supported(self, become_method): def _become_method_supported(self):
''' Checks if the current class supports this privilege escalation method ''' ''' Checks if the current class supports this privilege escalation method '''
if become_method in self.__class__.become_methods: if self._connection_info.become_method in self.__class__.become_methods:
return True return True
raise AnsibleError("Internal Error: this connection module does not support running commands via %s" % become_method) raise AnsibleError("Internal Error: this connection module does not support running commands via %s" % become_method)
@ -90,7 +90,10 @@ class ConnectionBase(with_metaclass(ABCMeta, object)):
@abstractmethod @abstractmethod
def _connect(self): def _connect(self):
"""Connect to the host we've been initialized with""" """Connect to the host we've been initialized with"""
pass
# Check if PE is supported
if self._connection_info.become:
self.__become_method_supported()
@ensure_connect @ensure_connect
@abstractmethod @abstractmethod