ansible/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2024-09-14 20:13:21 +02:00
Code Releases Activity

Export SSH port number as host variable.

Browse source
This commit is contained in:
Jeroen Hoekx 2012-04-14 13:12:32 +02:00
parent f04041b37d
commit 54f4526160
2 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
lib/ansible/inventory.py
View file

@ -35,6 +35,7 @@ class Inventory(object):
def __init__(self, host_list=C.DEFAULT_HOST_LIST, extra_vars=None): def __init__(self, host_list=C.DEFAULT_HOST_LIST, extra_vars=None):
self._restriction = None self._restriction = None
self._variables = {}
if type(host_list) == list: if type(host_list) == list:
self.host_list = host_list self.host_list = host_list
@ -80,6 +81,9 @@ class Inventory(object):
def get_variables(self, host, extra_vars=None): def get_variables(self, host, extra_vars=None):
""" Return the variables associated with this host. """ """ Return the variables associated with this host. """
if host in self._variables:
return self._variables[host]
if not self._is_script: if not self._is_script:
return {} return {}
@ -108,6 +112,11 @@ class Inventory(object):
if ":" in item: if ":" in item:
# a port was specified # a port was specified
item, port = item.split(":") item, port = item.split(":")
try:
port = int(port)
except ValueError:
raise errors.AnsibleError("SSH port for %s in inventory (%s) should be numerical."%(item, port))
self._set_variable(item, "ansible_ssh_port", port)
groups[group_name].append(item) groups[group_name].append(item)
if not item in results: if not item in results:
results.append(item) results.append(item)
@ -170,6 +179,11 @@ class Inventory(object):
)) ))
return variables return variables
def _set_variable(self, host, key, value):
if not host in self._variables:
self._variables[host] = {}
self._variables[host][key] = value
def _matches(self, host_name, pattern): def _matches(self, host_name, pattern):
''' returns if a hostname is matched by the pattern ''' ''' returns if a hostname is matched by the pattern '''

6
test/TestInventory.py
View file

@ -82,6 +82,12 @@ class TestInventory(unittest.TestCase):
assert vars == {} assert vars == {}
def test_simple_port(self):
inventory = self.simple_inventory()
vars = inventory.get_variables('hera')
assert vars == {'ansible_ssh_port': 3000}
### Inventory API tests ### Inventory API tests
def test_script(self): def test_script(self):