aboutsummaryrefslogtreecommitdiffstats
path: root/app/discover/link_finders/find_links_for_vservice_vnics.py
blob: ca9bc4a908eea5c2edbbdf56b75e16dead13d427 (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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
###############################################################################
# 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                                  #
###############################################################################
from discover.link_finders.find_links import FindLinks


class FindLinksForVserviceVnics(FindLinks):
    def __init__(self):
        super().__init__()

    def add_links(self, search=None):
        self.log.info("adding links of type: vservice-vnic")

        if search is None:
            search = {}

        search.update({"environment": self.get_env(),
                       "type": "vnic",
                       "vnic_type": "vservice_vnic"})

        vnics = self.inv.find_items(search)

        for v in vnics:
            self.add_link_for_vnic(v)

    def add_link_for_vnic(self, v):
        host = self.inv.get_by_id(self.get_env(), v["host"])
        if "Network" not in host["host_type"]:
            return
        if "network" not in v:
            return
        network = self.inv.get_by_id(self.get_env(), v["network"])
        if network == []:
            return
        vservice_id = v["parent_id"]
        vservice_id = vservice_id[:vservice_id.rindex('-')]
        vservice = self.inv.get_by_id(self.get_env(), vservice_id)
        source = vservice["_id"]
        source_id = vservice_id
        target = v["_id"]
        target_id = v["id"]
        link_type = "vservice-vnic"
        link_name = network["name"]
        state = "up"  # TBD
        link_weight = 0  # TBD
        self.create_link(self.get_env(),
                         source, source_id,
                         target, target_id,
                         link_type, link_name, state, link_weight,
                         host=v["host"],
                         extra_attributes={'network': v['network']})