few changes on the ipv6.py logic, and a bit of code cleanup in tools.py,
[nodemanager.git] / plugins / update_ipv6addr_slivertag.py
1 """
2 Description: Update the IPv6 Address sliver tag accordingly to the IPv6 address set
3 update_ipv6addr_slivertag nodemanager plugin
4 Version: 0.5
5 Author: Guilherme Sperb Machado <gsm@machados.org>
6 """
7
8 import logger
9 import os
10 import socket
11 import re
12
13 import tools
14 import libvirt
15 import uuid
16 from sliver_libvirt import Sliver_Libvirt
17 from xml.dom.minidom import parseString
18
19 priority=150
20
21 ipv6addrtag = 'ipv6_address'
22
23 def start():
24     logger.log("update_ipv6addr_slivertag: plugin starting up...")
25
26
27 def getSliverTagId(slivertags):
28     for slivertag in slivertags:
29         if slivertag['tagname']==ipv6addrtag:
30             return slivertag['slice_tag_id']
31     return None
32
33 def SetSliverTag(plc, data, tagname):
34
35     for slice in data['slivers']:
36         logger.log("update_ipv6addr_slivertag: starting with slice=%s" % (slice['name']) )
37
38         # TODO: what about the prefixlen? should we add on it as well?
39         # here, I'm just taking the ipv6addr (value)
40         value,prefixlen = tools.get_sliver_ipv6(slice['name'])
41
42         node_id = tools.node_id()
43         slivertags = plc.GetSliceTags({"name":slice['name'],"node_id":node_id,"tagname":tagname})
44         #logger.log(repr(str(slivertags)))
45         #for tag in slivertags:
46             #    logger.log(repr(str(tag)))
47
48         ipv6addr = plc.GetSliceIPv6Address(slice['name'])
49         logger.log("update_ipv6addr_slivertag: slice=%s getSliceIPv6Address=%s" % \
50                (slice['name'],ipv6addr) )
51         # if the value to set is null...
52         if value is None:
53             if ipv6addr is not None:
54                 # then, let's remove the slice tag
55                 slivertag_id = getSliverTagId(slivertags)
56                 if slivertag_id:
57                     try:
58                         plc.DeleteSliceTag(slivertag_id)
59                         logger.log("update_ipv6addr_slivertag: slice tag deleted for slice=%s" % \
60                                (slice['name']) )
61                     except:
62                         logger.log("update_ipv6addr_slivertag: slice tag not deleted for slice=%s" % \
63                                (slice['name']) )
64             # if there's no ipv6 address anymore, then remove everything from the /etc/hosts
65             tools.remove_all_ipv6addr_hosts(slice['name'], data['hostname'])
66         else:
67             # if the ipv6 addr set on the slice does not exist yet, so, let's add it
68             if (ipv6addr is None) and len(value)>0:
69                 try:
70                     logger.log("update_ipv6addr_slivertag: slice name=%s" % (slice['name']) )
71                     slivertag_id=plc.AddSliceTag(slice['name'],tagname,value,node_id)
72                     logger.log("update_ipv6addr_slivertag: slice tag added to slice %s" % \
73                            (slice['name']) )
74                 except:
75                     logger.log("update_ipv6addr_slivertag: could not set ipv6 addr tag to sliver. "+
76                            "slice=%s tag=%s node_id=%d" % (slice['name'],tagname,node_id) )
77             # if the ipv6 addr set on the slice is different on the value provided, let's update it
78             if (ipv6addr is not None) and (len(value)>0) and (ipv6addr!=value):
79                 slivertag_id = getSliverTagId(slivertags)
80                 plc.UpdateSliceTag(slivertag_id,value)
81             # ipv6 entry on /etc/hosts of each slice
82             result = tools.search_ipv6addr_hosts(slice['name'], value)
83             if not result:
84                 tools.remove_all_ipv6addr_hosts(slice['name'], data['hostname'])
85                 tools.add_ipv6addr_hosts_line(slice['name'], data['hostname'], value)
86         logger.log("update_ipv6addr_slivertag: finishing the update process for " +
87                "slice=%s" % (slice['name']) )
88
89 def GetSlivers(data, config, plc):
90
91     SetSliverTag(plc, data, ipv6addrtag)
92
93     logger.log("update_ipv6addr_slivertag: all done!")