Revision 279f2c7d image_creator/disk.py
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