And...I should actually read examples before I copy them

[invirt/packages/invirt-web.git] / code / controls.py

diff --git a/code/controls.py b/code/controls.py
index 7134668..b2c60d5 100644 (file)
--- a/code/controls.py
+++ b/code/controls.py
@@ -68,6 +68,10 @@ def makeDisks(machine):
     for disk in machine.disks:
         lvcreate(machine, disk)
 
     for disk in machine.disks:
         lvcreate(machine, disk)
 

+def lvcopy(machine_orig_name, machine, rootpw):
+    """Copy a golden image onto a machine's disk"""
+    remctl('web', 'lvcopy', machine_orig_name, machine.name, rootpw)
+

 def bootMachine(machine, cdtype):
     """Boot a machine with a given boot CD.
 
 def bootMachine(machine, cdtype):
     """Boot a machine with a given boot CD.
 

@@ -95,7 +99,7 @@ def unregisterMachine(machine):
     """Unregister a machine to not be controlled by the web interface"""
     remctl('web', 'unregister', machine.name)
 
     """Unregister a machine to not be controlled by the web interface"""
     remctl('web', 'unregister', machine.name)
 

-def createVm(owner, contact, name, memory, disk_size, is_hvm, cdrom):
+def createVm(owner, contact, name, memory, disk_size, is_hvm, cdrom, clone_from):

     """Create a VM and put it in the database"""
     # put stuff in the table
     transaction = ctx.current.create_transaction()
     """Create a VM and put it in the database"""
     # put stuff in the table
     transaction = ctx.current.create_transaction()

@@ -136,6 +140,8 @@ def createVm(owner, contact, name, memory, disk_size, is_hvm, cdrom):
         raise
     registerMachine(machine)
     makeDisks(machine)
         raise
     registerMachine(machine)
     makeDisks(machine)

+    if clone_from:
+        lvcopy(clone_from, machine, 'password')

     # tell it to boot with cdrom
     bootMachine(machine, cdrom)
     return machine
     # tell it to boot with cdrom
     bootMachine(machine, cdrom)
     return machine