1 files changed, 60 insertions, 0 deletions
diff --git a/app/discover/fetchers/db/db_fetch_instances.py b/app/discover/fetchers/db/db_fetch_instances.py
new file mode 100644
index 0000000..54c4114
--- /dev/null
+++ b/app/discover/fetchers/db/db_fetch_instances.py
@@ -0,0 +1,60 @@
+###############################################################################
+# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems)   #
+# and others                                                                  #
+#                                                                             #
+# All rights reserved. This program and the accompanying materials            #
+# are made available under the terms of the Apache License, Version 2.0       #
+# which accompanies this distribution, and is available at                    #
+# http://www.apache.org/licenses/LICENSE-2.0                                  #
+###############################################################################
+import json
+
+from discover.fetchers.db.db_access import DbAccess
+
+
+class DbFetchInstances(DbAccess):
+    def get_instance_data(self, instances):
+        instances_hash = {}
+        for doc in instances:
+            instances_hash[doc["id"]] = doc
+
+        query = """
+          SELECT DISTINCT i.uuid AS id, i.display_name AS name,
+            i.host AS host, host_ip AS ip_address,
+            network_info, project_id,
+            IF(p.name IS NULL, "Unknown", p.name) AS project
+          FROM nova.instances i
+            LEFT JOIN keystone.project p ON p.id = i.project_id
+            JOIN nova.instance_info_caches ic ON i.uuid = ic.instance_uuid
+            JOIN nova.compute_nodes cn ON i.node = cn.hypervisor_hostname
+          WHERE i.deleted = 0
+        """
+        results = self.get_objects_list(query, "instance")
+        for result in results:
+            id = result["id"]
+            if id not in instances_hash:
+                continue
+            self.build_instance_details(result)
+            doc = instances_hash[id]
+            doc.update(result)
+
+    def build_instance_details(self, result):
+        network_info_str = result.pop("network_info", None)
+        result["network_info"] = json.loads(network_info_str)
+
+        # add network as an array to allow constraint checking
+        # when building clique
+        networks = []
+        for net in result["network_info"]:
+            if "network" not in net or "id" not in net["network"]:
+                continue
+            network_id = net["network"]["id"]
+            if network_id in networks:
+                continue
+            networks.append(network_id)
+        result["network"] = networks
+
+        result["type"] = "instance"
+        result["parent_type"] = "instances_folder"
+        result["parent_id"] = result["host"] + "-instances"
+        result["in_project-" + result.pop("project", None)] = "1"