class ToolkitDatabase(object):
def __init__(self, conn):
self.conn = conn
self.profile = Profile(self.conn)
self.family = Family(self.conn)
self.suite = None
self.trait = None
self.machine = MachineHandler(self.conn)
def set_profile(self, profile):
self.profile.set_profile(profile)
self.suite = self.profile.current.suite
self.trait = Trait(self.conn, self.suite)
def set_trait(self, trait):
self.trait.set_trait(trait)
def set_machine(self, machine):
self.machine.set_machine(machine)
profile = self.machine.get_profile()
self.set_profile(profile)
if os.environ.has_key('PAELLA_TRAIT'):
self.set_trait(os.environ['PAELLA_TRAIT'])
def env(self):
env = TemplatedEnvironment()
if self.trait.current_trait is not None:
env.update(self.trait._parents.Environment())
env.update(self.profile.get_family_data())
env.update(self.profile.get_profile_data())
if self.machine.current_machine is not None:
env.update(self.machine.get_machine_data())
return env
class MachineInstaller(BaseMachineInstaller):
def __init__(self, conn):
BaseMachineInstaller.__init__(self, conn)
# the processes are mostly the same as in the
# ChrootInstaller
self._processes = list(DEFAULT_PROCESSES)
pmap = dict(setup_disks=self.setup_disks,
mount_target=self.mount_target,
install_fstab=self.install_fstab,
install_modules=self.install_modules,
install_kernel=self.install_kernel,
prepare_bootloader=self.prepare_bootloader
)
self._process_map.update(pmap)
self.machine = MachineHandler(self.conn)
self.helper = None
self._target_mounted = False
self._disks_setup = False
def set_machine(self, machine):
self.check_target_set()
self.machine.set_machine(machine)
logdir = path(self.defenv.get('installer', 'base_log_directory'))
if not logdir.isdir():
logdir.mkdir()
logfile = logdir / 'paella-install-%s.log' % machine
os.environ['PAELLA_MACHINE'] = machine
disklogpath = path(self.defenv.get('installer', 'disk_log_directory'))
self.disklogpath = disklogpath / ('disklog-%s' % machine)
if not self.disklogpath.isdir():
self.disklogpath.makedirs()
self.set_logfile(logfile)
self.log.info('machine set to %s' % machine)
# we need to set machine_data before setting the profile
# so that the machine_data is passed to the profile and trait installers
self.machine_data = self.machine.get_machine_data()
profile = self.machine.get_profile()
self.set_profile(profile)
self.curenv = CurrentEnvironment(self.conn, machine)
self.helper = MachineInstallerHelper(self)
self.helper.curenv = self.curenv
def make_script(self, procname):
self.check_machine_set()
script = self.machine.relation.get_script(procname, inherited=True)
if script is not None:
return make_script(procname, script, '/')
else:
return None
def ready_base_for_install(self):
self.check_target_mounted()
# run ready_base_for_install from chroot installer first
ChrootInstaller.ready_base_for_install(self)
def install_modules(self):
self.check_install_complete()
self.log.info("install_modules isn't being used anymore.")
msg = "This step is still here, in case you need to use a script"
msg += " to add modules to /etc/modules ."
self.log.info(msg)
if False:
self.helper.install_modules()
def install_kernel(self):
self.check_install_complete()
self.helper.install_kernel()
def prepare_bootloader(self):
self.helper.prepare_bootloader()
def install_fstab(self):
self.check_install_complete()
self.helper.install_fstab()
def setup_disks(self):
self.helper.setup_disks()
self._disks_setup = True
def mount_target(self):
self.check_target_exists()
self.check_disks_setup()
self.helper.mount_target()
self._target_mounted = True
class MachineInstaller(BaseMachineInstaller):
def __init__(self, conn):
BaseMachineInstaller.__init__(self, conn)
# the processes are mostly the same as in the
# ChrootInstaller
self._processes = list(DEFAULT_PROCESSES)
pmap = dict(setup_disks=self.setup_disks,
mount_target=self.mount_target,
install_fstab=self.install_fstab,
install_modules=self.install_modules,
install_kernel=self.install_kernel,
prepare_bootloader=self.prepare_bootloader
)
self._process_map.update(pmap)
self.machine = MachineHandler(self.conn)
self.helper = None
self._target_mounted = False
self._disks_setup = False
def set_machine(self, machine):
self.check_target_set()
self.machine.set_machine(machine)
logdir = path(self.defenv.get('installer', 'base_log_directory'))
if not logdir.isdir():
logdir.mkdir()
logfile = logdir / 'paella-install-%s.log' % machine
os.environ['PAELLA_MACHINE'] = machine
disklogpath = path(self.defenv.get('installer', 'disk_log_directory'))
self.disklogpath = disklogpath / ('disklog-%s' % machine)
if not self.disklogpath.isdir():
self.disklogpath.makedirs()
self.set_logfile(logfile)
self.log.info('machine set to %s' % machine)
# we need to set machine_data before setting the profile
# so that the machine_data is passed to the profile and trait installers
self.machine_data = self.machine.get_machine_data()
profile = self.machine.get_profile()
self.set_profile(profile)
self.curenv = CurrentEnvironment(self.conn, machine)
self.helper = MachineInstallerHelper(self)
self.helper.curenv = self.curenv
def make_script(self, procname):
self.check_machine_set()
script = self.machine.relation.get_script(procname, inherited=True)
if script is not None:
return make_script(procname, script, '/')
else:
return None
def ready_base_for_install(self):
self.check_target_mounted()
# run ready_base_for_install from chroot installer first
ChrootInstaller.ready_base_for_install(self)
def install_modules(self):
self.check_install_complete()
self.log.info("install_modules isn't being used anymore.")
msg = "This step is still here, in case you need to use a script"
msg += " to add modules to /etc/modules ."
self.log.info(msg)
if False:
self.helper.install_modules()
def install_kernel(self):
self.check_install_complete()
self.helper.install_kernel()
def prepare_bootloader(self):
self.helper.prepare_bootloader()
def install_fstab(self):
self.check_install_complete()
self.helper.install_fstab()
def setup_disks(self):
self.helper.setup_disks()
self._disks_setup = True
def mount_target(self):
self.check_target_exists()
self.check_disks_setup()
self.helper.mount_target()
# this is now done in post_process
#self._target_mounted = True
def log_all_processes_started(self):
machine = self.machine.current_machine
installer = self.__class__.__name__
self.log.info('Starting all processes for %s(%s)' % (installer, machine))
def log_all_processes_finished(self):
machine = self.machine.current_machine
installer = self.__class__.__name__
self.log.info('Finished all processes for %s(%s)' % (installer, machine))
def post_process(self, procname):
# be sure to run parent's post_process first
BaseMachineInstaller.post_process(self, procname)
name = self.__class__.__name__
# now check for processes specific to the machine installer
if procname == 'mount_target':
self.log.info('%s marking %s finished' % (name, procname))
self._target_mounted = True
elif procname == 'setup_disks':
self.log.info('%s marking %s finished' % (name, procname))
self._disks_setup = True