Move recipe class into an own file
[nitsi.git] / machine.py
CommitLineData
a6a7fe20
JS
1#!/usr/bin/python3
2
3import xml.etree.ElementTree as ET
4
5from disk import disk
6
7from serial_connection import serial_connection
8
9from test import libvirt_con
10
11import os
12import libvirt
13
14
15class machine():
16 def __init__(self, vm_xml_file, snapshot_xml_file, image, root_uid, username, password):
17 self.log = log(4)
18 self.con = libvirt_con("qemu:///system")
19 try:
20 with open(vm_xml_file) as fobj:
21 self.vm_xml = fobj.read()
22 except FileNotFoundError as error:
23 self.log.error("No such file: {}".format(vm_xml_file))
24
25 try:
26 with open(snapshot_xml_file) as fobj:
27 self.snapshot_xml = fobj.read()
28 except FileNotFoundError as error:
29 self.log.error("No such file: {}".format(snapshot_xml_file))
30
31 self.image = image
32
33 if not os.path.isfile(self.image):
34 self.log.error("No such file: {}".format(self.image))
35
36 self.root_uid = root_uid
37 self.disk = disk(image)
38
39 self.username = username
40 self.password = password
41
42 def define(self):
43 self.dom = self.con.con.defineXML(self.vm_xml)
44 if self.dom == None:
45 self.log.error("Could not define VM")
46 raise BaseException
47
48 def start(self):
49 if self.dom.create() < 0:
50 self.log.error("Could not start VM")
51 raise BaseException
52
53 def shutdown(self):
54 if self.is_running():
55 if self.dom.shutdown() < 0:
56 self.log.error("Could not shutdown VM")
57 raise BaseException
58 else:
59 self.log.error("Domain is not running")
60
61 def undefine(self):
62 self.dom.undefine()
63
64 def create_snapshot(self):
65
66 self.snapshot = self.dom.snapshotCreateXML(self.snapshot_xml)
67
68 if not self.snapshot:
69 self.log.error("Could not create snapshot")
70 raise BaseException
71
72 def revert_snapshot(self):
73 self.dom.revertToSnapshot(self.snapshot)
74 self.snapshot.delete()
75
76 def is_running(self):
77
78 state, reason = self.dom.state()
79
80 if state == libvirt.VIR_DOMAIN_RUNNING:
81 return True
82 else:
83 return False
84
85 def get_serial_device(self):
86
87 if not self.is_running():
88 raise BaseException
89
90 xml_root = ET.fromstring(self.dom.XMLDesc(0))
91
92 elem = xml_root.find("./devices/serial/source")
93 return elem.get("path")
94
95 def check_is_booted_up(self):
96 serial_con = serial_connection(self.get_serial_device())
97
98 serial_con.write("\n")
99 # This will block till the domain is booted up
100 serial_con.read(1)
101
102 #serial_con.close()
103
104 def login(self):
105 try:
106 self.serial_con = serial_connection(self.get_serial_device(), username=self.username)
107 self.serial_con.login(self.password)
108 except BaseException as e:
109 self.log.error("Could not connect to the domain via serial console")
110
111 def cmd(self, cmd):
112 return self.serial_con.command(cmd)
113
114 def copy_in(self, fr, to):
115 try:
116 self.disk.mount(self.root_uid, "/")
117 self.disk.copy_in(fr, to)
118 except BaseException as e:
119 self.log.error(e)
120 finally:
121 self.disk.umount("/")
122 self.disk.close()