mirror of
https://github.com/virt-manager/virt-manager.git
synced 2025-03-08 04:58:29 +03:00
virtinst: move OSInstallScript to unattended
Signed-off-by: Cole Robinson <crobinso@redhat.com>
This commit is contained in:
Cole Robinson
parent
a8d0fb5a38
commit
0031d15098
@ -10,11 +10,7 @@ import datetime
|
||||
import logging
|
||||
import re
|
||||
|
||||
import gi
|
||||
gi.require_version('Libosinfo', '1.0')
|
||||
from gi.repository import Libosinfo
|
||||
from gi.repository import Gio
|
||||
from gi.repository import GLib
|
||||
|
||||
|
||||
###################
|
||||
@ -545,186 +541,3 @@ class _OsVariant(object):
|
||||
# scripts when its actually needed, though.
|
||||
installscript = filtered_script_list.get_nth(0)
|
||||
return installscript
|
||||
|
||||
|
||||
class OSInstallScript:
|
||||
"""
|
||||
Wrapper for Libosinfo.InstallScript interactions
|
||||
"""
|
||||
def __init__(self, script, osobj):
|
||||
self._script = script
|
||||
self._osobj = osobj
|
||||
|
||||
def get_expected_filename(self):
|
||||
return self._script.get_expected_filename()
|
||||
|
||||
def set_preferred_injection_method(self, method):
|
||||
def nick_to_value(method):
|
||||
injection_methods = [
|
||||
Libosinfo.InstallScriptInjectionMethod.CDROM,
|
||||
Libosinfo.InstallScriptInjectionMethod.DISK,
|
||||
Libosinfo.InstallScriptInjectionMethod.FLOPPY,
|
||||
Libosinfo.InstallScriptInjectionMethod.INITRD,
|
||||
Libosinfo.InstallScriptInjectionMethod.WEB]
|
||||
|
||||
for m in injection_methods:
|
||||
if method == m.value_nicks[0]:
|
||||
return m
|
||||
|
||||
raise RuntimeError(
|
||||
_("%s is a non-valid injection method in libosinfo."))
|
||||
|
||||
injection_method = nick_to_value(method)
|
||||
supported_injection_methods = self._script.get_injection_methods()
|
||||
if (injection_method & supported_injection_methods == 0):
|
||||
raise RuntimeError(
|
||||
_("OS '%s' unattended install is not supported") %
|
||||
self._osobj.name)
|
||||
|
||||
logging.debug("Using '%s' injection method", method)
|
||||
self._script.set_preferred_injection_method(injection_method)
|
||||
|
||||
def set_installation_source(self, source):
|
||||
def nick_to_value(source):
|
||||
installation_sources = [
|
||||
Libosinfo.InstallScriptInstallationSource.MEDIA,
|
||||
Libosinfo.InstallScriptInstallationSource.NETWORK]
|
||||
|
||||
for s in installation_sources:
|
||||
if source == s.value_nick:
|
||||
return s
|
||||
|
||||
raise RuntimeError(
|
||||
_("%s is a non-valid installation source in libosinfo.") %
|
||||
source)
|
||||
|
||||
installation_source = nick_to_value(source)
|
||||
|
||||
logging.debug("Using '%s' installation source", source)
|
||||
self._script.set_installation_source(installation_source)
|
||||
|
||||
def get_config(self, unattended_data, arch, hostname):
|
||||
def requires_param(config_param):
|
||||
param = self._script.get_config_param(config_param)
|
||||
|
||||
if not param or param.is_optional():
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
def requires_user_password():
|
||||
return requires_param(Libosinfo.INSTALL_CONFIG_PROP_USER_PASSWORD)
|
||||
|
||||
def requires_admin_password():
|
||||
return requires_param(Libosinfo.INSTALL_CONFIG_PROP_ADMIN_PASSWORD)
|
||||
|
||||
def get_timezone():
|
||||
TZ_FILE = "/etc/localtime"
|
||||
localtime = Gio.File.new_for_path(TZ_FILE)
|
||||
if not localtime.query_exists():
|
||||
return None
|
||||
info = localtime.query_info(
|
||||
Gio.FILE_ATTRIBUTE_STANDARD_SYMLINK_TARGET,
|
||||
Gio.FileQueryInfoFlags.NOFOLLOW_SYMLINKS)
|
||||
if not info:
|
||||
return None
|
||||
target = info.get_symlink_target()
|
||||
if not target:
|
||||
return None
|
||||
tokens = target.split("zoneinfo/")
|
||||
if not tokens or len(tokens) < 2:
|
||||
return None
|
||||
return tokens[1]
|
||||
|
||||
def get_language():
|
||||
names = GLib.get_language_names()
|
||||
if not names or len(names) < 2:
|
||||
return None
|
||||
return names[1]
|
||||
|
||||
config = Libosinfo.InstallConfig()
|
||||
|
||||
# Set user login and name based on the one from the system
|
||||
config.set_user_login(GLib.get_user_name())
|
||||
config.set_user_realname(GLib.get_real_name())
|
||||
|
||||
# Set user-password.
|
||||
# In case it's required and not passed, just raise a RuntimeError.
|
||||
if requires_user_password() and not unattended_data.user_password:
|
||||
raise RuntimeError(
|
||||
_("%s requires the user-password to be set.") %
|
||||
self._osobj.name)
|
||||
config.set_user_password(
|
||||
unattended_data.user_password if unattended_data.user_password
|
||||
else "")
|
||||
|
||||
# Set the admin-password:
|
||||
# In case it's required and not passed, just raise a RuntimeError.
|
||||
if requires_admin_password() and not unattended_data.admin_password:
|
||||
raise RuntimeError(
|
||||
_("%s requires the admin-password to be set.") %
|
||||
self._osobj.name)
|
||||
config.set_admin_password(
|
||||
unattended_data.admin_password if unattended_data.admin_password
|
||||
else "")
|
||||
|
||||
# Set the target disk.
|
||||
# virtiodisk is the preferred way, in case it's supported, otherwise
|
||||
# just fallback to scsi.
|
||||
#
|
||||
# Note: this is linux specific and will require some changes whenever
|
||||
# support for Windows will be added.
|
||||
tgt = "/dev/vda" if self._osobj.supports_virtiodisk() else "/dev/sda"
|
||||
config.set_target_disk(tgt)
|
||||
|
||||
# Set hardware architecture and hostname
|
||||
config.set_hardware_arch(arch)
|
||||
config.set_hostname(hostname)
|
||||
|
||||
# Try to guess the timezone from '/etc/localtime', in case it's not
|
||||
# possible 'America/New_York' will be used.
|
||||
timezone = get_timezone()
|
||||
if timezone:
|
||||
config.set_l10n_timezone(timezone)
|
||||
else:
|
||||
logging.warning(
|
||||
_("'America/New_York' timezone will be used for this "
|
||||
"unattended installation."))
|
||||
|
||||
# Try to guess to language and keyboard layout from the system's
|
||||
# language.
|
||||
#
|
||||
# This method has flows as it's quite common to have language and
|
||||
# keyboard layout not matching. Otherwise, there's no easy way to guess
|
||||
# the keyboard layout without relying on a set of APIs of an specific
|
||||
# Desktop Environment.
|
||||
language = get_language()
|
||||
if language:
|
||||
config.set_l10n_language(language)
|
||||
config.set_l10n_keyboard(language)
|
||||
else:
|
||||
logging.warning(
|
||||
_("'en_US' will be used as both language and keyboard layout "
|
||||
"for unattended installation."))
|
||||
|
||||
logging.debug("InstallScriptConfig created with the following params:")
|
||||
logging.debug("username: %s", config.get_user_login())
|
||||
logging.debug("realname: %s", config.get_user_realname())
|
||||
logging.debug("user password: %s", config.get_user_password())
|
||||
logging.debug("admin password: %s", config.get_admin_password())
|
||||
logging.debug("target disk: %s", config.get_target_disk())
|
||||
logging.debug("hardware arch: %s", config.get_hardware_arch())
|
||||
logging.debug("hostname: %s", config.get_hostname())
|
||||
logging.debug("timezone: %s", config.get_l10n_timezone())
|
||||
logging.debug("language: %s", config.get_l10n_language())
|
||||
logging.debug("keyboard: %s", config.get_l10n_keyboard())
|
||||
|
||||
return config
|
||||
|
||||
def generate_output(self, config, output_dir):
|
||||
self._script.generate_output(
|
||||
self._osobj.get_handle(), config, output_dir)
|
||||
|
||||
def generate_cmdline(self, config):
|
||||
return self._script.generate_command_line(
|
||||
self._osobj.get_handle(), config)
|
||||
|
||||
@ -6,10 +6,199 @@
|
||||
# This work is licensed under the GNU GPLv2 or later.
|
||||
# See the COPYING file in the top-level directory.
|
||||
|
||||
import logging
|
||||
import os
|
||||
|
||||
import gi
|
||||
gi.require_version('Libosinfo', '1.0')
|
||||
from gi.repository import Libosinfo
|
||||
from gi.repository import Gio
|
||||
from gi.repository import GLib
|
||||
|
||||
from . import util
|
||||
from .osdict import OSInstallScript
|
||||
|
||||
|
||||
class OSInstallScript:
|
||||
"""
|
||||
Wrapper for Libosinfo.InstallScript interactions
|
||||
"""
|
||||
def __init__(self, script, osobj):
|
||||
self._script = script
|
||||
self._osobj = osobj
|
||||
|
||||
def get_expected_filename(self):
|
||||
return self._script.get_expected_filename()
|
||||
|
||||
def set_preferred_injection_method(self, method):
|
||||
def nick_to_value(method):
|
||||
injection_methods = [
|
||||
Libosinfo.InstallScriptInjectionMethod.CDROM,
|
||||
Libosinfo.InstallScriptInjectionMethod.DISK,
|
||||
Libosinfo.InstallScriptInjectionMethod.FLOPPY,
|
||||
Libosinfo.InstallScriptInjectionMethod.INITRD,
|
||||
Libosinfo.InstallScriptInjectionMethod.WEB]
|
||||
|
||||
for m in injection_methods:
|
||||
if method == m.value_nicks[0]:
|
||||
return m
|
||||
|
||||
raise RuntimeError(
|
||||
_("%s is a non-valid injection method in libosinfo."))
|
||||
|
||||
injection_method = nick_to_value(method)
|
||||
supported_injection_methods = self._script.get_injection_methods()
|
||||
if (injection_method & supported_injection_methods == 0):
|
||||
raise RuntimeError(
|
||||
_("OS '%s' unattended install is not supported") %
|
||||
self._osobj.name)
|
||||
|
||||
logging.debug("Using '%s' injection method", method)
|
||||
self._script.set_preferred_injection_method(injection_method)
|
||||
|
||||
def set_installation_source(self, source):
|
||||
def nick_to_value(source):
|
||||
installation_sources = [
|
||||
Libosinfo.InstallScriptInstallationSource.MEDIA,
|
||||
Libosinfo.InstallScriptInstallationSource.NETWORK]
|
||||
|
||||
for s in installation_sources:
|
||||
if source == s.value_nick:
|
||||
return s
|
||||
|
||||
raise RuntimeError(
|
||||
_("%s is a non-valid installation source in libosinfo.") %
|
||||
source)
|
||||
|
||||
installation_source = nick_to_value(source)
|
||||
|
||||
logging.debug("Using '%s' installation source", source)
|
||||
self._script.set_installation_source(installation_source)
|
||||
|
||||
def get_config(self, unattended_data, arch, hostname):
|
||||
def requires_param(config_param):
|
||||
param = self._script.get_config_param(config_param)
|
||||
|
||||
if not param or param.is_optional():
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
def requires_user_password():
|
||||
return requires_param(Libosinfo.INSTALL_CONFIG_PROP_USER_PASSWORD)
|
||||
|
||||
def requires_admin_password():
|
||||
return requires_param(Libosinfo.INSTALL_CONFIG_PROP_ADMIN_PASSWORD)
|
||||
|
||||
def get_timezone():
|
||||
TZ_FILE = "/etc/localtime"
|
||||
localtime = Gio.File.new_for_path(TZ_FILE)
|
||||
if not localtime.query_exists():
|
||||
return None
|
||||
info = localtime.query_info(
|
||||
Gio.FILE_ATTRIBUTE_STANDARD_SYMLINK_TARGET,
|
||||
Gio.FileQueryInfoFlags.NOFOLLOW_SYMLINKS)
|
||||
if not info:
|
||||
return None
|
||||
target = info.get_symlink_target()
|
||||
if not target:
|
||||
return None
|
||||
tokens = target.split("zoneinfo/")
|
||||
if not tokens or len(tokens) < 2:
|
||||
return None
|
||||
return tokens[1]
|
||||
|
||||
def get_language():
|
||||
names = GLib.get_language_names()
|
||||
if not names or len(names) < 2:
|
||||
return None
|
||||
return names[1]
|
||||
|
||||
config = Libosinfo.InstallConfig()
|
||||
|
||||
# Set user login and name based on the one from the system
|
||||
config.set_user_login(GLib.get_user_name())
|
||||
config.set_user_realname(GLib.get_real_name())
|
||||
|
||||
# Set user-password.
|
||||
# In case it's required and not passed, just raise a RuntimeError.
|
||||
if requires_user_password() and not unattended_data.user_password:
|
||||
raise RuntimeError(
|
||||
_("%s requires the user-password to be set.") %
|
||||
self._osobj.name)
|
||||
config.set_user_password(
|
||||
unattended_data.user_password if unattended_data.user_password
|
||||
else "")
|
||||
|
||||
# Set the admin-password:
|
||||
# In case it's required and not passed, just raise a RuntimeError.
|
||||
if requires_admin_password() and not unattended_data.admin_password:
|
||||
raise RuntimeError(
|
||||
_("%s requires the admin-password to be set.") %
|
||||
self._osobj.name)
|
||||
config.set_admin_password(
|
||||
unattended_data.admin_password if unattended_data.admin_password
|
||||
else "")
|
||||
|
||||
# Set the target disk.
|
||||
# virtiodisk is the preferred way, in case it's supported, otherwise
|
||||
# just fallback to scsi.
|
||||
#
|
||||
# Note: this is linux specific and will require some changes whenever
|
||||
# support for Windows will be added.
|
||||
tgt = "/dev/vda" if self._osobj.supports_virtiodisk() else "/dev/sda"
|
||||
config.set_target_disk(tgt)
|
||||
|
||||
# Set hardware architecture and hostname
|
||||
config.set_hardware_arch(arch)
|
||||
config.set_hostname(hostname)
|
||||
|
||||
# Try to guess the timezone from '/etc/localtime', in case it's not
|
||||
# possible 'America/New_York' will be used.
|
||||
timezone = get_timezone()
|
||||
if timezone:
|
||||
config.set_l10n_timezone(timezone)
|
||||
else:
|
||||
logging.warning(
|
||||
_("'America/New_York' timezone will be used for this "
|
||||
"unattended installation."))
|
||||
|
||||
# Try to guess to language and keyboard layout from the system's
|
||||
# language.
|
||||
#
|
||||
# This method has flows as it's quite common to have language and
|
||||
# keyboard layout not matching. Otherwise, there's no easy way to guess
|
||||
# the keyboard layout without relying on a set of APIs of an specific
|
||||
# Desktop Environment.
|
||||
language = get_language()
|
||||
if language:
|
||||
config.set_l10n_language(language)
|
||||
config.set_l10n_keyboard(language)
|
||||
else:
|
||||
logging.warning(
|
||||
_("'en_US' will be used as both language and keyboard layout "
|
||||
"for unattended installation."))
|
||||
|
||||
logging.debug("InstallScriptConfig created with the following params:")
|
||||
logging.debug("username: %s", config.get_user_login())
|
||||
logging.debug("realname: %s", config.get_user_realname())
|
||||
logging.debug("user password: %s", config.get_user_password())
|
||||
logging.debug("admin password: %s", config.get_admin_password())
|
||||
logging.debug("target disk: %s", config.get_target_disk())
|
||||
logging.debug("hardware arch: %s", config.get_hardware_arch())
|
||||
logging.debug("hostname: %s", config.get_hostname())
|
||||
logging.debug("timezone: %s", config.get_l10n_timezone())
|
||||
logging.debug("language: %s", config.get_l10n_language())
|
||||
logging.debug("keyboard: %s", config.get_l10n_keyboard())
|
||||
|
||||
return config
|
||||
|
||||
def generate_output(self, config, output_dir):
|
||||
self._script.generate_output(
|
||||
self._osobj.get_handle(), config, output_dir)
|
||||
|
||||
def generate_cmdline(self, config):
|
||||
return self._script.generate_command_line(
|
||||
self._osobj.get_handle(), config)
|
||||
|
||||
|
||||
class UnattendedData():
|
||||
@ -19,8 +208,6 @@ class UnattendedData():
|
||||
|
||||
|
||||
def generate_install_script(guest, unattended_data):
|
||||
from gi.repository import Gio as gio
|
||||
|
||||
rawscript = guest.osinfo.get_install_script(unattended_data.profile)
|
||||
script = OSInstallScript(rawscript, guest.osinfo)
|
||||
|
||||
@ -35,7 +222,7 @@ def generate_install_script(guest, unattended_data):
|
||||
if not os.path.exists(scratch):
|
||||
os.makedirs(scratch, 0o751)
|
||||
|
||||
script.generate_output(config, gio.File.new_for_path(scratch))
|
||||
script.generate_output(config, Gio.File.new_for_path(scratch))
|
||||
path = os.path.join(scratch, script.get_expected_filename())
|
||||
cmdline = script.generate_cmdline(config)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user