blob: 59ba5d0bf2431011060791e21997a17e06c3a333 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
###############################################################################
# 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 bson
from discover.fetchers.db.db_access import DbAccess
from utils.inventory_mgr import InventoryMgr
class DbFetchAggregateHosts(DbAccess):
def get(self, id):
query = """
SELECT CONCAT('aggregate-', a.name, '-', host) AS id, host AS name
FROM nova.aggregate_hosts ah
JOIN nova.aggregates a ON a.id = ah.aggregate_id
WHERE ah.deleted = 0 AND aggregate_id = %s
"""
hosts = self.get_objects_list_for_id(query, "host", id)
if hosts:
inv = InventoryMgr()
for host_rec in hosts:
host_id = host_rec['name']
host = inv.get_by_id(self.get_env(), host_id)
if not host:
self.log.error('unable to find host {} '
'from aggregate {} in inventory'
.format(host_id, id))
continue
host_rec['ref_id'] = bson.ObjectId(host['_id'])
return hosts
|
