diff --git a/creator/livecd-creator b/creator/livecd-creator index bcef946..a591eac 100755 --- a/creator/livecd-creator +++ b/creator/livecd-creator @@ -133,8 +133,17 @@ class InstallationTarget: return True + def check_free_space(self, scratch_space_dir): + """Check for space on defined build directory.""" + self.scratch_space_dir = scratch_space_dir + varspace = os.statvfs(scratch_space_dir) + + """Based on the default block size, calculate free space, and hand it back to the caller.""" + calcfreevar = varspace[0]*varspace[4] + return calcfreevar + + def setup(self, image_size, fs_label, base_on, scratch_space_dir): - def setup(self, image_size, fs_label, base_on): """setup target ext3 file system in preparation for an install""" # global variables needed @@ -143,9 +152,10 @@ class InstallationTarget: self.loop_device = "" # setup temporary build dirs - self.build_dir = "/var/tmp/livecd-creator/build-" + os.path.basename(tempfile.mktemp()) + appended_dir = scratch_space_dir + "build-" + os.path.basename(tempfile.mktemp()) + self.build_dir = appended_dir try: - os.makedirs(self.build_dir) + os.mkdir(self.build_dir) except OSError: print "Cannot create build directory at %s" return False @@ -315,7 +325,8 @@ class InstallationTarget: except OSError: pass os.system("/bin/rm -rf %s/out"%(self.build_dir)) - if "/var/tmp/livecd-creator/build-" in self.build_dir: + #if "/var/tmp/livecd-creator/build-" in self.build_dir: + if self.build_dir + "build-" in self.build_dir: os.system("/bin/rm -rf %s"%self.build_dir) def addRepository(self, name, url): @@ -459,6 +470,7 @@ def usage(): print " [--exclude-package=] --exclude-package= ...]" print " [--base-on=]" print " [--fslabel=