#!/usr/bin/env python
-import losetup
+from image_creator.util import get_command
+from clint.textui import progress
+
import stat
import os
import tempfile
import sys
import guestfs
-import pbs
-from pbs import dd
-from clint.textui import progress
-
class DiskError(Exception):
pass
-
-def find_sbin_command(command, exception):
- search_paths = ['/usr/local/sbin', '/usr/sbin', '/sbin']
- for fullpath in map(lambda x: "%s/%s" % (x, command), search_paths):
- if os.path.exists(fullpath) and os.access(fullpath, os.X_OK):
- return pbs.Command(fullpath)
- continue
- raise exception
-
-
-try:
- from pbs import dmsetup
-except pbs.CommandNotFound as e:
- dmsetup = find_sbin_command('dmsetup', e)
-
-try:
- from pbs import blockdev
-except pbs.CommandNotFound as e:
- blockdev = find_sbin_command('blockdev', e)
+dd = get_command('dd')
+dmsetup = get_command('dmsetup')
+losetup = get_command('losetup')
+blockdev = get_command('blockdev')
class Disk(object):
self._cleanup_jobs.append((job, args))
def _losetup(self, fname):
- loop = losetup.find_unused_loop_device()
- loop.mount(fname)
- self._add_cleanup(loop.unmount)
- return loop.device
+ loop = losetup('-f', '--show', fname)
+ loop = loop.strip() # remove the new-line char
+ self._add_cleanup(losetup, '-d', loop)
+ return loop
def _dir_to_disk(self):
raise NotImplementedError
from image_creator import get_os_class
from image_creator import __version__ as version
from image_creator.disk import Disk
+from image_creator.util import get_command
+
import sys
import os
import optparse
-from pbs import dd
+
+dd = get_command('dd')
class FatalError(Exception):
size = options.shrink and dev.shrink() or dev.size()
metadata['size'] = str(size // 2 ** 20)
-
dd('if=%s' % dev.device,
'of=%s/%s.%s' % (options.outdir, options.name, 'diskdump'),
'bs=4M', 'count=%d' % ((size + 1) // 2 ** 22))
--- /dev/null
+#!/usr/bin/env python
+
+import pbs
+
+def get_command(command):
+ def find_sbin_command(command, exception):
+ search_paths = ['/usr/local/sbin', '/usr/sbin', '/sbin']
+ for fullpath in map(lambda x: "%s/%s" % (x, command), search_paths):
+ if os.path.exists(fullpath) and os.access(fullpath, os.X_OK):
+ return pbs.Command(fullpath)
+ raise exception
+
+ try:
+ return pbs.__getattr__(command)
+ except pbs.CommadNotFount as e:
+ return find_sbin_command(command, e)
license='BSD',
packages=['image_creator'],
include_package_data=True,
- install_requires=['losetup', 'pbs', 'clint'],
+ install_requires=['pbs', 'clint'],
entry_points={
'console_scripts': ['snf-image-creator = image_creator.main:main']
}