qaapi/qa/Config.py

   1 import xmlrpclib
   2 import os
   3 import sys
   4 import re
   5 import socket
   6 import utils
   7 import copy
   8 from logger import logfile, Logfile
   9 from Table import Table
  10 from PLCs import PLC, PLCs
  11 from Sites import Site, Sites
  12 from Nodes import Node, Nodes
  13 from Slices import Slice, Slices
  14 from Persons import Person, Persons
  15 from TestScripts import TestScript
  16
  17 path = os.path.dirname(os.path.abspath(__file__))
  18
  19 class Config:
  20
  21     path = os.path.dirname(os.path.abspath(__file__))
  22     tests_path = path + os.sep + 'tests' + os.sep
  23     node_tests_path = tests_path + os.sep + 'node' + os.sep
  24     slice_tests_path = tests_path + os.sep + 'slice' + os.sep
  25     vserver_scripts_path = path + os.sep + 'vserver' + os.sep
  26
  27     def update_api(self, plc = None):
  28         # Set up API acccess
  29         # If plc is specified, find its configuration
  30         # and use its API
  31         if plc is not None:
  32             protocol, host, path, port  = 'http', plc['host'], plc['api_path'], plc['port']
  33             if port in ['443']:
  34                 protocol = 'https'
  35             api_server = "%(protocol)s://%(host)s:%(port)s/%(path)s" % locals()
  36             self.api = xmlrpclib.Server(api_server, allow_none = 1)
  37             self.api_type = 'xmlrpc'
  38         else:
  39
  40             # Try importing the API shell for direct api access.
  41             # If that fails fall back to using xmlrpm
  42             try:
  43                 sys.path.append('/usr/share/plc_api')
  44                 from PLC.Shell import Shell
  45                 shell = Shell(globals = globals())
  46
  47                 # test it
  48                 shell.GetRoles()
  49                 self.api = shell
  50                 self.api_type = 'direct'
  51             except:
  52                 self.api = xmlrpclib.Server('https://%s/PLCAPI/' % self.PLC_API_HOST, allow_none = 1)
  53                 self.api_type = 'xmlrpc'
  54
  55     def __init__(self, config_file = path+os.sep+'qa_config', logdir = "/var/log"):
  56         # Load config file
  57         try:
  58             execfile(config_file, self.__dict__)
  59         except:
  60             raise "Could not find system config in %s" % config_file
  61
  62         self.logdir = logdir
  63         log_filename = logdir + os.sep + "qaapi.log"
  64         self.update_logfile(log_filename)
  65         self.auth = {}
  66         self.auth['Username'] = self.PLC_ROOT_USER
  67         self.auth['AuthString'] = self.PLC_ROOT_PASSWORD
  68         self.auth['AuthMethod'] = 'password'
  69         self.verbose = self.VERBOSE
  70
  71         attributes = ['plcs', 'sites', 'slices', 'nodes', 'persons', 'nodegroups']
  72         for attribute in attributes:
  73             setattr(self, attribute, [])
  74
  75         # try setting hostname and ip
  76         self.hostname = socket.gethostname()
  77         try:
  78             command = "/sbin/ifconfig eth0 | grep -v inet6 | grep inet | awk '{print$2;}'"
  79             (status, output) = utils.commands(command, logfile = self.logfile)
  80             self.ip = re.findall(r'[0-9\.]+', output)[0]
  81         except:
  82             self.ip = '127.0.0.1'
  83
  84         api_host = self.__dict__.get("PLC_API_HOST",self.ip)
  85         self.PLC_API_HOST=api_host
  86
  87         self.update_api()
  88
  89         # Load list of node tests
  90         valid_node_test_files = lambda name: not name.startswith('__init__') \
  91                                              and not name.endswith('pyc')
  92         node_test_files = os.listdir(self.node_tests_path)
  93         self.node_test_files = filter(valid_node_test_files, node_test_files)
  94
  95     def update_logfile(self, filename):
  96         self.logfile = Logfile(filename)
  97         filename_parts = self.logfile.filename.split(os.sep)
  98         self.logdir = os.sep + os.sep.join(filename_parts[:-1]) + os.sep
  99
 100     def get_plc(self, plc_name):
 101         plc = PLC(self)
 102         if hasattr(self, 'plcs')  and plc_name in self.plcs.keys():
 103             plc.update(self.plcs[plc_name])
 104             plc.update_api()
 105         return plc
 106
 107     def get_node(self, hostname):
 108         node = Node(self)
 109         if hasattr(self, 'nodes') and hostname in self.nodes.keys():
 110             node.update(self.nodes[hostname])
 111             node.__init_logfile__()
 112         return node
 113
 114     def get_node_test(self, testscript):
 115         script = TestScript({'name': testscript})
 116         if hasattr(self, 'node_tests') and testscript in self.node_tests.keys():
 117             script.update(self.node_tests[testscript])
 118         return script
 119
 120     def get_slice_test(self, testscript):
 121         script = TestScript()
 122         if hasattr(self, 'slice_tests') and testscript in self.slice_tests.keys():
 123             script.update(self.slice_tests[testscript])
 124         return script
 125
 126
 127     def load(self, conffile):
 128
 129         confdata = {}
 130         try: execfile(conffile, confdata)
 131         except: raise
 132
 133         from Nodes import Nodes
 134         from PLCs import PLCs
 135         loadables = ['plcs', 'sites', 'nodes', 'nodegroups', 'slices', 'persons']
 136         config = Config(logdir = self.logdir)
 137         for element in confdata.keys():
 138             if element in ['plcs'] and confdata.has_key(element):
 139                 setattr(self, element, PLCs(config, confdata[element]).dict('name'))
 140                 setattr(config, element, PLCs(config, confdata[element]).dict('name'))
 141             elif element in ['nodes'] and confdata.has_key(element):
 142                 setattr(self, element, Nodes(config, confdata[element]).dict('hostname'))
 143                 setattr(config, element, Nodes(config, confdata[element]).dict('hostname'))
 144             elif element in ['nodegroups'] and confdata.has_key(element):
 145                 setattr(self, element, Table(confdata[element]).dict('name'))
 146                 setattr(config, element, Table(confdata[element]).dict('name'))
 147             elif element in ['sites'] and confdata.has_key(element):
 148                 setattr(self, element, Sites(confdata[element]).dict('login_base'))
 149                 setattr(config, element, Sites(confdata[element]).dict('login_base'))
 150             elif element in ['slices'] and confdata.has_key(element):
 151                 setattr(self, element, Slices(config, confdata[element]).dict('name'))
 152                 setattr(config, element, Slices(config, confdata[element]).dict('name'))
 153             elif element in ['persons'] and confdata.has_key(element):
 154                 setattr(self, element, Persons(confdata[element]).dict('email'))
 155                 setattr(config, element, Persons(confdata[element]).dict('email'))
 156             elif element in ['node_tests'] and confdata.has_key(element):
 157                 setattr(self, element, TestScripts(confdata[element]).dict('name'))
 158                 setattr(config, element, TestScripts(confdata[element]).dict('name'))
 159             elif element in ['slice_tests'] and confdata.has_key(element):
 160                 setattr(self, element, TestScripts(confdata[element]).dict('name'))
 161                 setattr(config, element, TestScripts(confdata[element]).dict('name'))
 162
 163     def archive_scripts(self, prefix):
 164         valid_prefix = ['slice', 'node']
 165         if prefix not in valid_prefix:
 166             raise "Error. Invalid prefix %s. Must be in %s" %  (prefix, valid_prefix)
 167
 168         scripts_dir = self.path + os.sep + 'tests' +os.sep + prefix + os.sep
 169         workdir = '/tmp' + os.sep
 170         archive_path = workdir + os.sep + prefix + os.sep
 171         archive_filename = prefix + ".tar.gz"
 172
 173         if self.verbose:
 174             utils.header("Creating/Updating %s archive %s" % (prefix, archive_path + archive_filename), logfile = self.logfile)
 175         utils.commands("mkdir -p %(archive_path)s" % locals(), logfile = self.logfile)
 176         utils.commands("cp -Rf %(scripts_dir)s* %(archive_path)s" % locals(), logfile = self.logfile)
 177         tar_cmd = "cd %(workdir)s && tar -czf %(workdir)s/%(archive_filename)s %(prefix)s" % locals()
 178         utils.commands(tar_cmd, logfile = self.logfile)
 179         return (archive_filename, workdir+archive_filename)
 180
 181     def archive_slice_tests(self):
 182         return self.archive_scripts('slice')
 183
 184     def archive_node_tests(self):
 185         return self.archive_scripts('node')