source/steps/GetAndUpdateNodeDetails.py

   1 #!/usr/bin/python
   2 #
   3 # Copyright (c) 2003 Intel Corporation
   4 # All rights reserved.
   5 #
   6 # Copyright (c) 2004-2006 The Trustees of Princeton University
   7 # All rights reserved.
   8
   9 import sys, traceback
  10 import string
  11
  12 from Exceptions import *
  13 import BootAPI
  14 import ModelOptions
  15
  16 def Run( vars, log ):
  17     """
  18
  19     Contact PLC and get the attributes for this node. Also, parse in
  20     options from the node model strong.
  21
  22     Also, update any node network settings at PLC, minus the ip address,
  23     so, upload the mac (if node_id was in conf file), gateway, network,
  24     broadcast, netmask, dns1/2, and the hostname/domainname.
  25
  26     Expect the following keys to be set:
  27     SKIP_HARDWARE_REQUIREMENT_CHECK     Whether or not we should skip hardware
  28                                         requirement checks
  29
  30     The following keys are set/updated:
  31     WAS_NODE_ID_IN_CONF                 Set to 1 if the node id was in the conf file
  32     WAS_NODE_KEY_IN_CONF                Set to 1 if the node key was in the conf file
  33     BOOT_STATE                          The current node boot state
  34     NODE_MODEL                          The user specified model of this node
  35     NODE_MODEL_OPTIONS                  The options extracted from the user specified
  36                                                 model of this node
  37     SKIP_HARDWARE_REQUIREMENT_CHECK     Whether or not we should skip hardware
  38                                                 requirement checks
  39     NODE_SESSION                        The session value returned from BootGetNodeDetails
  40     INTERFACES                          The network interfaces associated with this node
  41     INTERFACE_SETTINGS                  A dictionary of the values of the interface settings
  42
  43     Return 1 if able to contact PLC and get node info.
  44     Raise a BootManagerException if anything fails.
  45     """
  46
  47     log.write( "\n\nStep: Retrieving details of node from PLC.\n" )
  48
  49     # make sure we have the variables we need
  50     try:
  51         SKIP_HARDWARE_REQUIREMENT_CHECK= vars["SKIP_HARDWARE_REQUIREMENT_CHECK"]
  52         if SKIP_HARDWARE_REQUIREMENT_CHECK == "":
  53             raise ValueError, "SKIP_HARDWARE_REQUIREMENT_CHECK"
  54
  55         INTERFACE_SETTINGS= vars["INTERFACE_SETTINGS"]
  56         if INTERFACE_SETTINGS == "":
  57             raise ValueError, "INTERFACE_SETTINGS"
  58
  59         WAS_NODE_ID_IN_CONF= vars["WAS_NODE_ID_IN_CONF"]
  60         if WAS_NODE_ID_IN_CONF == "":
  61             raise ValueError, "WAS_NODE_ID_IN_CONF"
  62
  63         WAS_NODE_KEY_IN_CONF= vars["WAS_NODE_KEY_IN_CONF"]
  64         if WAS_NODE_KEY_IN_CONF == "":
  65             raise ValueError, "WAS_NODE_KEY_IN_CONF"
  66
  67     except KeyError, var:
  68         raise BootManagerException, "Missing variable in vars: %s\n" % var
  69     except ValueError, var:
  70         raise BootManagerException, "Variable in vars, shouldn't be: %s\n" % var
  71
  72     node_details= BootAPI.call_api_function( vars, "GetNodes",
  73                                              (vars['NODE_ID'],
  74                                               ['boot_state', 'nodegroup_ids', 'interface_ids', 'model', 'site_id']))[0]
  75
  76     vars['BOOT_STATE']= node_details['boot_state']
  77     vars['RUN_LEVEL']= node_details['boot_state']
  78     vars['NODE_MODEL']= string.strip(node_details['model'])
  79     vars['SITE_ID'] = node_details['site_id']
  80     log.write( "Successfully retrieved node record.\n" )
  81     log.write( "Current boot state: %s\n" % vars['BOOT_STATE'] )
  82     log.write( "Node make/model: %s\n" % vars['NODE_MODEL'] )
  83
  84     # parse in the model options from the node_model string
  85     model= vars['NODE_MODEL']
  86     options= ModelOptions.Get(model)
  87     vars['NODE_MODEL_OPTIONS']=options
  88
  89     # Check if we should skip hardware requirement check
  90     if options & ModelOptions.MINHW:
  91         vars['SKIP_HARDWARE_REQUIREMENT_CHECK']=1
  92         log.write( "node model indicates override to hardware requirements.\n" )
  93
  94     # this contains all the node networks, for now, we are only concerned
  95     # in the primary network
  96     interfaces= BootAPI.call_api_function( vars, "GetInterfaces", (node_details['interface_ids'],))
  97     got_primary= 0
  98     for network in interfaces:
  99         if network['is_primary'] == 1:
 100             log.write( "Primary network as returned from PLC: %s\n" % str(network) )
 101             got_primary= 1
 102             break
 103
 104     if not got_primary:
 105         raise BootManagerException, "Node did not have a primary network."
 106
 107     vars['INTERFACES']= interfaces
 108
 109     # call getNodeFlavour and store in VARS['node_flavour']
 110     try:
 111         node_flavour = BootAPI.call_api_function(vars, "GetNodeFlavour", (vars['NODE_ID'], ) )
 112     except:
 113         log.write("GetNodeFlavour failed, not fatal if the node flavor is available in ``configuration''\n")
 114         pass
 115
 116     flavour_keys = [
 117             'virt',# 'vs' or 'lxc'
 118             'nodefamily',# the basename for downloading nodeimage
 119             'extensions',# extensions to be applied on top of the base nodeimage
 120             'plain'# false if compressed image, true if not
 121             ]
 122
 123     # MyPLC 5.0 workaround
 124     # make sure to define 'extensions' even if not yet set
 125     if ('extensions' not in vars or vars['extensions']==''):
 126         vars['extensions']=[]
 127
 128     for k in flavour_keys:
 129         # Support MyPLC <5.2
 130         if (not vars.has_key(k)):
 131             try:
 132                 vars[k] = node_flavour[k]
 133             except:
 134                 exc_type, exc_value, exc_traceback = sys.exc_info()
 135                 lines=traceback.format_exception(exc_type,exc_value,exc_traceback)
 136                 for line in lines: log.write(line)
 137                 raise BootManagerException ("Could not call GetNodeFlavour - need PLCAPI-5.2")
 138
 139     log.write ("NodeFlavour as returned from PLC: %s\n"%node_flavour)
 140
 141     return 1