]>
git.ipfire.org Git - people/ms/nitsi.git/blob - src/nitsi/disk.py
9 logger
= logging
.getLogger("nitsi.disk")
13 def __init__(self
, disk
):
14 self
.log
= logger
.getChild(os
.path
.basename(disk
))
15 self
.log
.debug("Initiated a disk class for {}".format(disk
))
16 self
.con
= guestfs
.GuestFS(python_return_dict
=True)
17 self
.con
.add_drive_opts(disk
, format
="qcow2")
19 def mount(self
, uuid
, path
):
20 self
.log
.debug("Trying to mount the partion with uuid: {} under {}".format(uuid
, path
))
22 part
= self
.con
.findfs_uuid(uuid
)
23 self
.con
.mount(part
, path
)
25 def copy_in(self
, fr
, to
):
26 self
.log
.debug("Going to copy some files into the image.")
27 tmp
= tempfile
.mkstemp()
29 with tarfile
.open(tmp
, "w") as tar
:
31 self
.log
.debug("Adding {} to be copied into the image".format(file))
32 tar
.add(file, arcname
=os
.path
.basename(file))
34 self
.log
.debug("Going to copy the files into the image")
35 self
.con
.tar_in_opts(tmp
, to
)
37 def umount(self
, path
):
38 self
.log
.debug("Unmounting the image")
39 self
.con
.umount_opts(path
)
42 self
.log
.debug("Flush the image and closing the connection")
46 # test = disk("/var/lib/libvirt/images/alice.qcow2")
47 # test.mount("45598e92-3487-4a1b-961d-79aa3dd42a7d", "/")
48 # test.copy_in("/home/jonatan/nitsi/libguestfs-test", "/root/")