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())