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