calculate-utils-2.2-install/pym/cl_install.py

#-*- coding: utf-8 -*-

# Copyright 2010 Mir Calculate Ltd. http://www.calculate-linux.org
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

__version__ = "2.2.0"
__app__ = "calculate-install"

import os
import re
import sys

from cl_lang import lang
from cl_template import template
from cl_fill import clLocale
from cl_datavars import DataVars
from cl_print import color_print
from cl_distr import PartitionDistributive,DistributiveRepository,\
                DistributiveError, ScratchDistributive
from time import sleep

tr = lang()
tr.setGlobalDomain('cl_install')
tr.setLanguage(sys.modules[__name__])

class InstallError(Exception):
    """Installation Error"""
    pass

class printNoColor:
    def colorPrint(self,attr,fg,bg,string):
        sys.stdout.write(string)

class DataVarsInstall(DataVars):
    """Variable class for installation"""

    def flInstall(self, **args):
        '''Заполнить конфигурацию переменных, для десктопа'''
        # Имя секции в calculate.env
        envSection = "calculate-install"
        # заполнить переменные окружения алгоритмом по умолнанию
        self.importData(envSection, ('cl_vars_install','cl_fill_install'))

class cl_install(color_print):
    """Primary class for templates appling and system installation"""

    def __init__(self):
        self.clVars = None
        self.clTemp = None
        self.color = True

    def colorPrint(self,*argv,**kwarg):
        if self.color:
            color_print.colorPrint(self,*argv,**kwarg)
        else:
            sys.stdout.write(argv[-1])
            sys.stdout.flush()

    def setNoColor(self):
        self.color = False

    def initVars(self):
        """Primary initialization of variables"""
        self.clVars = DataVarsInstall()
        self.clVars.flInstall()
        self.clVars.flIniFile()

    def applyTemplatesForSystem(self):
        """Apply templates for root of system."""
        self.clVars.Set("cl_root_path","/", True)
        self.clTemp = template(self.clVars)
        dirsFiles = self.clTemp.applyTemplates()
        if self.clTemp.getError():
            self.printERROR(self.clTemp.getError())
            return False
        else:
            return dirsFiles

    def printInfo(self,sourceDistr,targetDistr):
        self.printSUCCESS(_("Machine hardware name: %s")%
            self.clVars.Get('os_arch_machine'))
        self.printSUCCESS("Found update: %s"%self.clVars.Get('os_linux_name'))

    def wait(self,sec=10):
        for i in xrange(sec,0,-1):
            self.printSUCCESS(_("Press %s to cancel")%"Ctrl+C"+"...  %d"%i)
            sleep(1)

    def prepareBoot(self):
        pass

    def getTargetDistributive(self,disk,buildermode):
        if buildermode:
            return ScratchDistributive(disk,mdirectory="/mnt/install",
                                    check=True)
        else:
            return PartitionDistributive(disk,mdirectory="/mnt/install",
                                    check=True)

    def joinTemplates(self,directory):
        pass

    def installSystem(self,disk="",buildermode=False):
        """install System by current variable enviroment"""
        sourceDistr = None
        targetDistr = None
        error = None
        try:
            targetDistr = self.getTargetDistributive(disk,buildermode)
            distRep = DistributiveRepository('/usr/calculate/share/linux')
            sourceDistr = distRep.getLastDistributive(
                march=self.clVars.Get('os_arch_machine'),
                shortname=self.clVars.Get('os_linux_shortname').lower())
            if sourceDistr:
                # print info
                self.printInfo(sourceDistr,targetDistr)
                # wait 10 sec
                self.wait(3)
                # install distributive
                self.printSUCCESS(_("Unpacking system image into target"))
                targetDistr.installFrom(sourceDistr)
                self.printOK("Unpacking complete")
                # join templates
                self.printSUCCESS(_("Update config"))
                self.joinTemplates(targetDistr.getDirectory())
                # change boot config
                self.printSUCCESS(_("Prepare system for reboot"))
                self.prepareBoot()
            else:
                self.printWARNING("No update available.")

        except DistributiveError,e:
            error = e
        except KeyboardInterrupt,e:
            error = _("Installation manually interrupt")
        if sourceDistr:
            sourceDistr.close()
        if targetDistr:
            targetDistr.close()
        if error:
            self.printERROR(str(error))
            return False
        return False

    def setAllLocaleByLang(self,lang):
        """Set all locale variable by specified lang"""
        locale = clLocale()
        if not locale.isLangExists(lang):
            return False
        self.clVars.Set('os_locale_lang',lang, True)
        self.clVars.Set('os_locale_locale',
            locale.getFieldByLang('locale',lang), True)
        self.clVars.Set('os_locale_language',
            locale.getFieldByLang('language',lang), True)
        self.clVars.Set('os_locale_keymap',
            locale.getFieldByLang('keymap',lang), True)
        self.clVars.Set('os_locale_dumpkeys',
            locale.getFieldByLang('dumpkeys_charset',lang), True)
        self.clVars.Set('os_locale_xkb',
            locale.getFieldByLang('xkblayout',lang), True)
        return True