Revision 279f2c7d

b/image_creator/disk.py
45 45 
from sendfile import sendfile
46 46 

  		





  47
  47
  
    

  		





  48
  
  
    
class DiskError(Exception):


  		





  49
  
  
    
    pass


  		





  50
  
  
    

  		





  51
  48
  
    
dd = get_command('dd')


  		





  52
  49
  
    
dmsetup = get_command('dmsetup')


  		





  53
  50
  
    
losetup = get_command('losetup')


  		





  ......




  80
  77
  
    
        return loop


  		





  81
  78
  
    

  		





  82
  79
  
    
    def _dir_to_disk(self):


  		





  83
  
  
    
        raise NotImplementedError


  		





  
  80
  
    
        raise FatalError("Using a directory as media source is not supported "


  		





  
  81
  
    
                         "yet!")


  		





  84
  82
  
    

  		





  85
  83
  
    
    def cleanup(self):


  		





  86
  84
  
    
        """Cleanup internal data. This needs to be called before the


  		





  ......




  103
  101
  
    
        sourcedev = self.source


  		





  104
  102
  
    
        mode = os.stat(self.source).st_mode


  		





  105
  103
  
    
        if stat.S_ISDIR(mode):


  		





  106
  
  
    
            success('looks like a directory')


  		





  
  104
  
    
            self.out.success('looks like a directory')


  		





  107
  105
  
    
            return self._losetup(self._dir_to_disk())


  		





  108
  106
  
    
        elif stat.S_ISREG(mode):


  		





  109
  
  
    
            success('looks like an image file')


  		





  
  107
  
    
            self.out.success('looks like an image file')


  		





  110
  108
  
    
            sourcedev = self._losetup(self.source)


  		





  111
  109
  
    
        elif not stat.S_ISBLK(mode):


  		





  112
  110
  
    
            raise ValueError("Invalid media source. Only block devices, "


  		












Also available in:
  Unified diff