Read /etc/os-release to fetch release information

[people/ms/bricklayer.git] / src / python / __init__.py

###############################################################################
#                                                                             #
# Bricklayer - An Installer for IPFire                                        #
# Copyright (C) 2021 IPFire Development Team                                  #
#                                                                             #
# This program is free software; you can redistribute it and/or               #
# modify it under the terms of the GNU General Public License                 #
# as published by the Free Software Foundation; either version 2              #
# of the License, or (at your option) any later version.                      #
#                                                                             #
# This program is distributed in the hope that it will be useful,             #
# but WITHOUT ANY WARRANTY; without even the implied warranty of              #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
# GNU General Public License for more details.                                #
#                                                                             #
# You should have received a copy of the GNU General Public License           #
# along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
#                                                                             #
###############################################################################

import logging
import sys
import traceback

from . import disk
from . import lang
from . import logger
from . import step
from . import tui
from . import util
from .i18n import _
from .errors import *

# Setup logging
log = logging.getLogger("bricklayer")

class Bricklayer(object):
	"""
		Bricklayer's base class
	"""
	def __init__(self, test=False, debug=False, unattended=False):
		self.test = test
		self.unattended = unattended

		# Enable debug logging
		if debug:
			log.setLevel(logging.DEBUG)

		# Settings
		self.settings = {
			"language" : lang.default_language,
		}

		# Read OS information
		self.os = self._read_os_release()

		# Hardware
		self.disks = disk.Disks(self)

		# Initialise the text user interface
		self.tui = tui.Tui(self)

		# Log when we are ready
		if self.test:
			log.info("Bricklayer initialized in test mode")
		else:
			log.info("Bricklayer initialized")

	# An ordered list of all available steps
	steps = (
		step.UnattendedWarning,
		lang.SelectLanguage,
		step.Welcome,
		disk.SelectDisk,
	)

	def __call__(self):
		with self.tui:
			# Walk through all steps
			for step in self.steps:
				try:
					self._run_step(step)

				# End installer if aborted
				except InstallAbortedError:
					return 1

				# Catch all other exceptions and show an error
				except:
					type, value, tb = sys.exc_info()

					# Log the error
					log.error("An unexpected error occured:", exc_info=True)

					# Format the exception
					error = _("The installation cannot be continued due to an error:"
						"\n\n%s") % "".join(traceback.format_exception(type, value, tb))

					# Show an error message
					self.tui.error(_("An Unexpected Error Occured"),
						"".join(error), buttons=[_("Exit")], width=60)

					# Exit
					return 1

	def _run_step(self, stepcls):
		"""
			Runs a single step
		"""
		# Initialize the step
		step = stepcls(self)

		# Skip this step if it isn't enabled
		if not step.enabled:
			return

		# Run it
		return step.run(self.tui)

	def _read_os_release(self):
		"""
			Read /etc/os-release
		"""
		with open("/etc/os-release") as f:
			return util.config_read(f)