omf/reset_xmpp_pubsub_nodes.py

   1 #!/usr/bin/env /usr/bin/plcsh
   2
   3 import sys
   4 import xmlrpclib
   5 sys.path.append("/usr/bin/")
   6 from omf_slicemgr import *
   7
   8 xmlrpc = xmlrpclib.ServerProxy("http://localhost:5053")
   9
  10 xmppserver = config.PLC_OMF_XMPP_SERVER
  11 xmppuser = "@".join([config.PLC_OMF_XMPP_USER, xmppserver])
  12 xmpppass = config.PLC_OMF_XMPP_PASSWORD
  13 pubsub = PubSubClient(xmppuser, xmpppass, verbose=False)
  14
  15
  16 def delete_all_nodes(iq):
  17     global pubsub
  18     print "Deleting PubSub groups..."
  19     for i in iq.query.elements():
  20         node = i['node']
  21         reactor.callLater(1, pubsub.delete_node, node)
  22
  23 def is_local_node(node_id, slice_name):
  24     try:
  25         return GetNodes({'node_id': node_id}, ['peer_id'])[0]['peer_id'] == None
  26     except IndexError:
  27         print "WARNING: can not find the node with node_id %s" % node_id
  28         print "WARNING: node_id %s was referenced in slice %s" % (node_id, slice_name)
  29         return False
  30
  31 if __name__ == "__main__":
  32     pubsub.add_result_hook("discover", delete_all_nodes)
  33     reactor.callLater(1, pubsub.discover)
  34     reactor.callLater(2, pubsub.create_node, "/OMF")
  35     reactor.callLater(2, pubsub.create_node, "/SYSTEM")
  36
  37     reactor.callLater(4, reactor.stop)
  38     reactor.connectTCP(pubsub.id.host, 5222, pubsub.factory)
  39     reactor.run()
  40
  41     print "Re-creating PubSub groups..."
  42     slices = GetSlices()
  43     for slice in slices:
  44         xmlrpc.createSlice(slice['name'])
  45         for node_id in slice['node_ids']:
  46             try:
  47                 hrn = GetNodeTags({'tagname':'hrn', 'node_id': node_id})[0]['value']
  48                 xmlrpc.addResource(slice['name'], hrn)
  49             except IndexError:
  50                 if is_local_node(node_id, slice['name']):
  51                     print "WARNING: missing hrn tag for node_id: %s" % node_id