mclu_build.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 os
  21 import re
  22 import subprocess
  23 import sys
  24 import libvirt
  25
  26 import config
  27 import lib
  28 import libvirt_xml
  29
  30 def cmdline (subparsers):
  31     p = subparsers.add_parser (
  32         'build',
  33         help='build and run a new virtual machine',
  34     )
  35     p.add_argument (
  36         '--memory', default=1024, type=int,
  37         help='RAM to give to guest (default unit is megabytes)'
  38     )
  39     p.add_argument (
  40         '--vcpus', default=4, type=int,
  41         help='virtual CPUs to give to guest'
  42     )
  43     p.add_argument (
  44         '--virtio', default=True,
  45         help='use virtio disks and network'
  46     )
  47     p.add_argument (
  48         '--size', required=True,
  49         help='size of disk'
  50     )
  51     p.add_argument (
  52         'name',
  53         help='name of the new VM (or use "vm:name")'
  54     )
  55     p.add_argument (
  56         'os_version',
  57         help='OS & version of guest (see virt-builder docs)'
  58     )
  59     p.add_argument (
  60         'vbargs', nargs='*',
  61         help='virt-builder arguments (after "--")'
  62     )
  63     p.set_defaults (run=run)
  64
  65 def run (c, args):
  66     # Did the user request a particular node?  If not, we'll run it
  67     # on any node which is up.
  68     m = re.match (r'^(.*):(.*)$', args.name)
  69     if m:
  70         node_name = m.group (1)
  71         vm_name = m.group (2)
  72     else:
  73         node_name = lib.pick_any_node_which_is_up (c)
  74         vm_name = args.name
  75
  76     # Get all the guests, so we can tell if the name is a duplicate.
  77     running, inactive = lib.get_all_guests (c)
  78
  79     if vm_name in running or vm_name in inactive:
  80         sys.exit ("error: VM name (%s) already exists" % vm_name)
  81
  82     output = '%s/%s.img' % (c['images_dir'], vm_name)
  83
  84     # Call out to virt-builder to build the disk image.
  85     vbargs = [
  86         config.VIRT_BUILDER,
  87         args.os_version,
  88         '--output', output,
  89         '--format', 'qcow2',
  90     ]
  91     if args.size:
  92         vbargs.extend (['--size', args.size])
  93     if args.vbargs:
  94         vbargs.extend (args.vbargs)
  95     subprocess.check_call (vbargs)
  96
  97     # XXX Unfortunately this is necessary so qemu can access the disk.
  98     os.chmod (output, 0666)
  99
 100     # Generate the XML.  Would be nice to use virt-install here, but
 101     # it doesn't work: RHBZ#1095789
 102     xml = libvirt_xml.generate_libvirt_xml (vm_name,
 103                                             args.memory,
 104                                             args.vcpus,
 105                                             args.virtio,
 106                                             output)
 107
 108     # Write the XML to the xmls_dir.
 109     fp = open ("%s/%s.xml" % (c['xmls_dir'], vm_name), "w")
 110     fp.write (xml)
 111     fp.close ()
 112
 113     # Start the guest.
 114     lib.start_guest (c, node_name, vm_name)
 115     print "guest built and started on node %s" % node_name