mclu.py

   1 #!/usr/bin/python
   2 # mclu (mini cluster)
   3 # Copyright (C) 2014 Red Hat Inc.
   4 #
   5 # This program is free software; you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation; either version 2 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program; if not, write to the Free Software
  17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18
  19 import argparse
  20 import ConfigParser
  21 import os
  22 import re
  23 import sys
  24
  25 import config
  26 from node import Node
  27
  28 parser = argparse.ArgumentParser (
  29     prog='mclu',
  30     description='Mini virtualization cluster management tool',
  31 )
  32 parser.add_argument (
  33     '-f',
  34     type=file,
  35     help='specify location of the configuration file',
  36     metavar='MCLU.CONF',
  37 )
  38
  39 # Add subcommands.
  40 subparsers = parser.add_subparsers ()
  41 import mclu_build
  42 mclu_build.cmdline (subparsers)
  43 import mclu_console
  44 mclu_console.cmdline (subparsers)
  45 import mclu_info
  46 mclu_info.cmdline (subparsers)
  47 import mclu_list
  48 mclu_list.cmdline (subparsers)
  49 import mclu_migrate
  50 mclu_migrate.cmdline (subparsers)
  51 import mclu_shutdown
  52 mclu_shutdown.cmdline (subparsers)
  53 import mclu_start
  54 mclu_start.cmdline (subparsers)
  55 import mclu_status
  56 mclu_status.cmdline (subparsers)
  57 import mclu_stop
  58 mclu_stop.cmdline (subparsers)
  59 import mclu_viewer
  60 mclu_viewer.cmdline (subparsers)
  61 import mclu_wake
  62 mclu_wake.cmdline (subparsers)
  63
  64 args = parser.parse_args()
  65
  66 # Default location of the config file if not defined on the command line:
  67 if not args.f:
  68     config_dir = config.sysconfdir + "/" + config.PACKAGE_NAME
  69     args.f = open (config_dir + "/" + "mclu.conf")
  70 else:
  71     # config_dir is the directory containing mclu.conf
  72     config_dir = os.path.dirname (args.f.name)
  73
  74 # Configuration file default settings.  You cannot set defaults per
  75 # section, so we have to rely on setting names not overlapping.
  76 conf_defaults = {
  77     "home" : os.getenv ("HOME"),
  78     "host" : "SET.THIS.IN.MCLU.CONF",
  79     "config_dir" : config_dir,
  80     "uri" : "qemu+ssh://root@%(host)s/system",
  81 }
  82
  83 # Read the configuration file.
  84 conf = ConfigParser.SafeConfigParser (conf_defaults)
  85 conf.readfp (args.f)
  86
  87 # Global configuration.
  88 images_dir = conf.get ("global", "images_dir")
  89 if not os.path.isdir (images_dir):
  90     sys.exit ("configuration error: [globals] 'images_dir' (%s) directory does not exist" % images_dir)
  91 xmls_dir = conf.get ("global", "xmls_dir")
  92 if not os.path.isdir (xmls_dir):
  93     sys.exit ("configuration error: [globals] 'xmls_dir' (%s) directory does not exist", xmls_dir)
  94
  95 # Get the list of node names.
  96 node_names = conf.items ("nodes")
  97 node_names = filter (lambda (x, _) : re.search (r'^node', x), node_names)
  98 node_names = [ value for _, value in node_names ]
  99 if not node_names:
 100     sys.exit ("configuration error: [nodes] section in configuration file is empty")
 101
 102 # Get information about each node.
 103 nodes = {}
 104 for node_name in node_names:
 105     host = conf.get (node_name, "host")
 106     if not host:
 107         host = node_name
 108     mac = conf.get (node_name, "mac")
 109     uri = conf.get (node_name, "uri")
 110     node = Node (node_name, host, mac, uri)
 111     nodes[node_name] = node
 112
 113 # A config dict with less-used configuration settings.
 114 c = {
 115     "config_file" : args.f.name,
 116     "config_dir" : config_dir,
 117     "images_dir" : images_dir,
 118     "node_names" : node_names,
 119     "xmls_dir" : xmls_dir,
 120
 121     "conf" : conf,
 122 }
 123
 124 # Run the subcommand.
 125 args.run (c, args, nodes)