import tarfile
import os
+import logging
+
+logger = logging.getLogger("nitsi.disk")
+
class disk():
def __init__(self, disk):
+ self.log = logger.getChild(os.path.basename(disk))
+ self.log.debug("Initiated a disk class for {}".format(disk))
self.con = guestfs.GuestFS(python_return_dict=True)
self.con.add_drive_opts(disk, format="qcow2")
def mount(self, uuid, path):
+ self.log.debug("Trying to mount the partion with uuid: {} under {}".format(uuid, path))
self.con.launch()
part = self.con.findfs_uuid(uuid)
self.con.mount(part, path)
def copy_in(self, fr, to):
+ self.log.debug("Going to copy some files into the image.")
tmp = tempfile.mkstemp()
tmp = tmp[1] + ".tar"
with tarfile.open(tmp, "w") as tar:
for file in fr:
+ self.log.debug("Adding {} to be copied into the image".format(file))
tar.add(file, arcname=os.path.basename(file))
+
+ self.log.debug("Going to copy the files into the image")
self.con.tar_in_opts(tmp, to)
def umount(self, path):
+ self.log.debug("Unmounting the image")
self.con.umount_opts(path)
def close(self):
+ self.log.debug("Flush the image and closing the connection")
self.con.shutdown()
self.con.close()