import tempfile
+__version__ = '0.1.0'
+
+
class VmDebootstrap(cliapp.Application):
def add_settings(self):
'install package in DEB file onto system '
'(not from mirror)',
metavar='DEB')
+ self.settings.boolean(['no-kernel'], 'do not install a linux package')
self.settings.boolean(['enable-dhcp'], 'enable DHCP on eth0')
self.settings.string(['root-password'], 'set root password',
metavar='PASSWORD')
metavar='USER/PASSWORD')
self.settings.boolean(['serial-console'],
'configure image to use a serial console')
+ self.settings.string(['serial-console-command'],
+ 'command to manage the serial console, appended '
+ 'to /etc/inittab (%default)',
+ metavar='COMMAND',
+ default='/sbin/getty -L ttyS0 115200 vt100')
self.settings.boolean(['sudo'],
'install sudo, and if user is created, add them '
'to sudo group')
self.set_hostname(rootdir)
self.create_fstab(rootdir)
self.install_debs(rootdir)
+ self.cleanup_apt_cache(rootdir)
self.set_root_password(rootdir)
self.create_users(rootdir)
self.remove_udev_persistent_rules(rootdir)
self.setup_networking(rootdir)
+ self.customize(rootdir)
if self.settings['image']:
self.install_extlinux(rootdev, rootdir)
- self.customize(rootdir)
+ self.optimize_image(rootdir)
if self.settings['tarball']:
self.create_tarball(rootdir)
except BaseException, e:
def debootstrap(self, rootdir):
self.message('Debootstrapping')
- if self.settings['arch'] == 'i386':
- kernel_arch = '686'
- else:
- kernel_arch = self.settings['arch']
- kernel_image = 'linux-image-2.6-%s' % kernel_arch
+ necessary_packages = ['acpid']
+
+ include = self.settings['package']
+
+ if not self.settings['no-kernel']:
+ if self.settings['arch'] == 'i386':
+ kernel_arch = '486'
+ else:
+ kernel_arch = self.settings['arch']
+ kernel_image = 'linux-image-%s' % kernel_arch
+ include.append(kernel_image)
- include = [kernel_image] + self.settings['package']
if self.settings['sudo'] and 'sudo' not in include:
include.append('sudo')
- self.runcmd(['debootstrap',
- '--arch=%s' % self.settings['arch'],
- '--include=%s' % ','.join(include),
- self.settings['distribution'],
- rootdir,
- self.settings['mirror']])
+ args = ['debootstrap', '--arch=%s' % self.settings['arch']]
+ args.append(
+ '--include=%s' % ','.join(necessary_packages + include))
+ args += [self.settings['distribution'],
+ rootdir, self.settings['mirror']]
+ self.runcmd(args)
def set_hostname(self, rootdir):
hostname = self.settings['hostname']
logging.debug('stdout:\n%s' % out)
shutil.rmtree(tmp)
+ def cleanup_apt_cache(self, rootdir):
+ out = self.runcmd(['chroot', rootdir, 'apt-get', 'clean'])
+ logging.debug('stdout:\n%s' % out)
+
def set_root_password(self, rootdir):
if self.settings['root-password']:
self.message('Setting root password')
if self.settings['enable-dhcp']:
f.write('\n')
- f.write('allow-hotplug eth0\n')
+ f.write('auto eth0\n')
f.write('iface eth0 inet dhcp\n')
f.close()
label linux
kernel %(kernel)s
-append initrd=%(initrd)s root=UUID=%(uuid)s ro quiet %(kserial)s
+append initrd=%(initrd)s root=UUID=%(uuid)s ro %(kserial)s
%(extserial)s
''' % {
'kernel': kernel_image,
f.close()
if self.settings['serial-console']:
+ serial_command = self.settings['serial-console-command']
logging.debug('adding getty to serial console')
inittab = os.path.join(rootdir, 'etc/inittab')
with open(inittab, 'a') as f:
- f.write('\nS0:23:respawn:/sbin/getty -L ttyS0 115200 vt100\n')
+ f.write('\nS0:23:respawn:%s\n' % serial_command)
self.runcmd(['extlinux', '--install', rootdir])
self.runcmd(['sync'])
import time; time.sleep(2)
-
+
+ def optimize_image(self, rootdir):
+ """
+ Filing up the image with zeros will increase its compression rate
+ """
+ zeros = os.path.join(rootdir, 'ZEROS')
+ self.runcmd_unchecked(['dd', 'if=/dev/zero', 'of=' + zeros, 'bs=1M'])
+ self.runcmd(['rm', '-f', zeros])
+
+
def cleanup_system(self):
# Clean up after any errors.
if __name__ == '__main__':
- VmDebootstrap().run()
+ VmDebootstrap(version=__version__).run()