diff options
Diffstat (limited to 'lib/ansible/inventory/host.py')
-rw-r--r-- | lib/ansible/inventory/host.py | 169 |
1 files changed, 169 insertions, 0 deletions
diff --git a/lib/ansible/inventory/host.py b/lib/ansible/inventory/host.py new file mode 100644 index 0000000..18569ce --- /dev/null +++ b/lib/ansible/inventory/host.py @@ -0,0 +1,169 @@ +# (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 collections.abc import Mapping, MutableMapping + +from ansible.inventory.group import Group +from ansible.parsing.utils.addresses import patterns +from ansible.utils.vars import combine_vars, get_unique_id + + +__all__ = ['Host'] + + +class Host: + ''' a single ansible host ''' + + # __slots__ = [ 'name', 'vars', 'groups' ] + + def __getstate__(self): + return self.serialize() + + def __setstate__(self, data): + return self.deserialize(data) + + def __eq__(self, other): + if not isinstance(other, Host): + return False + return self._uuid == other._uuid + + def __ne__(self, other): + return not self.__eq__(other) + + def __hash__(self): + return hash(self.name) + + def __str__(self): + return self.get_name() + + def __repr__(self): + return self.get_name() + + def serialize(self): + groups = [] + for group in self.groups: + groups.append(group.serialize()) + + return dict( + name=self.name, + vars=self.vars.copy(), + address=self.address, + uuid=self._uuid, + groups=groups, + implicit=self.implicit, + ) + + def deserialize(self, data): + self.__init__(gen_uuid=False) # used by __setstate__ to deserialize in place # pylint: disable=unnecessary-dunder-call + + self.name = data.get('name') + self.vars = data.get('vars', dict()) + self.address = data.get('address', '') + self._uuid = data.get('uuid', None) + self.implicit = data.get('implicit', False) + + groups = data.get('groups', []) + for group_data in groups: + g = Group() + g.deserialize(group_data) + self.groups.append(g) + + def __init__(self, name=None, port=None, gen_uuid=True): + + self.vars = {} + self.groups = [] + self._uuid = None + + self.name = name + self.address = name + + if port: + self.set_variable('ansible_port', int(port)) + + if gen_uuid: + self._uuid = get_unique_id() + self.implicit = False + + def get_name(self): + return self.name + + def populate_ancestors(self, additions=None): + # populate ancestors + if additions is None: + for group in self.groups: + self.add_group(group) + else: + for group in additions: + if group not in self.groups: + self.groups.append(group) + + def add_group(self, group): + added = False + # populate ancestors first + for oldg in group.get_ancestors(): + if oldg not in self.groups: + self.groups.append(oldg) + + # actually add group + if group not in self.groups: + self.groups.append(group) + added = True + return added + + def remove_group(self, group): + removed = False + if group in self.groups: + self.groups.remove(group) + removed = True + + # remove exclusive ancestors, xcept all! + for oldg in group.get_ancestors(): + if oldg.name != 'all': + for childg in self.groups: + if oldg in childg.get_ancestors(): + break + else: + self.remove_group(oldg) + return removed + + def set_variable(self, key, value): + if key in self.vars and isinstance(self.vars[key], MutableMapping) and isinstance(value, Mapping): + self.vars = combine_vars(self.vars, {key: value}) + else: + self.vars[key] = value + + def get_groups(self): + return self.groups + + def get_magic_vars(self): + results = {} + results['inventory_hostname'] = self.name + if patterns['ipv4'].match(self.name) or patterns['ipv6'].match(self.name): + results['inventory_hostname_short'] = self.name + else: + results['inventory_hostname_short'] = self.name.split('.')[0] + + results['group_names'] = sorted([g.name for g in self.get_groups() if g.name != 'all']) + + return results + + def get_vars(self): + return combine_vars(self.vars, self.get_magic_vars()) |