]>
Commit | Line | Data |
---|---|---|
1 | #!/usr/bin/python3 | |
2 | ||
3 | import guestfs | |
4 | ||
5 | import tempfile | |
6 | import tarfile | |
7 | import os | |
8 | ||
9 | import logging | |
10 | ||
11 | logger = logging.getLogger("nitsi.disk") | |
12 | ||
13 | ||
14 | class disk(): | |
15 | def __init__(self, disk): | |
16 | self.log = logger.getChild(os.path.basename(disk)) | |
17 | self.log.debug("Initiated a disk class for {}".format(disk)) | |
18 | self.con = guestfs.GuestFS(python_return_dict=True) | |
19 | self.con.add_drive_opts(disk, format="qcow2") | |
20 | ||
21 | def mount(self, uuid, path): | |
22 | self.log.debug("Trying to mount the partion with uuid: {} under {}".format(uuid, path)) | |
23 | self.con.launch() | |
24 | part = self.con.findfs_uuid(uuid) | |
25 | self.con.mount(part, path) | |
26 | ||
27 | def copy_in(self, fr, to): | |
28 | self.log.debug("Going to copy some files into the image.") | |
29 | tmp = tempfile.mkstemp() | |
30 | tmp = tmp[1] + ".tar" | |
31 | with tarfile.open(tmp, "w") as tar: | |
32 | for file in fr: | |
33 | self.log.debug("Adding {} to be copied into the image".format(file)) | |
34 | tar.add(file, arcname=os.path.basename(file)) | |
35 | ||
36 | self.log.debug("Going to copy the files into the image") | |
37 | self.con.tar_in_opts(tmp, to) | |
38 | ||
39 | def umount(self, path): | |
40 | self.log.debug("Unmounting the image") | |
41 | self.con.umount_opts(path) | |
42 | ||
43 | def close(self): | |
44 | self.log.debug("Flush the image and closing the connection") | |
45 | self.con.shutdown() | |
46 | self.con.close() | |
47 | ||
48 | # test = disk("/var/lib/libvirt/images/alice.qcow2") | |
49 | # test.mount("45598e92-3487-4a1b-961d-79aa3dd42a7d", "/") | |
50 | # test.copy_in("/home/jonatan/nitsi/libguestfs-test", "/root/") | |
51 | # test.umount("/") | |
52 | # test.close() |