calamares/src/modules/localecfg/main.py

#!/usr/bin/env python3
# encoding: utf-8
# === This file is part of Calamares - <http://github.com/calamares> ===
#
#   Copyright 2014, Anke Boersma <demm@kaosx.us>
#
#   Calamares 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 3 of the License, or
#   (at your option) any later version.
#
#   Calamares 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 Calamares. If not, see <http://www.gnu.org/licenses/>.

import os
import shutil

import libcalamares


def run():
    """ Create locale """

    us = '#en_US'
    locale = libcalamares.globalstorage.value("lcLocale")
    if not locale:
        locale = 'en_US.UTF-8 UTF-8'

    install_path = libcalamares.globalstorage.value("rootMountPoint")

    # restore backup if available
    if os.path.exists('/etc/locale.gen.bak'):
        shutil.copy2('%s/etc/locale.gen.bak' %
                     (install_path), '%s/etc/locale.gen' % (install_path))

    # run locale-gen if detected
    if os.path.exists('/etc/locale.gen'):
        text = []
        with open("%s/etc/locale.gen" % install_path, "r") as gen:
            text = gen.readlines()

        # always enable en_US
        with open("%s/etc/locale.gen" % install_path, "w") as gen:
            for line in text:
                if us in line and line[0] == "#":
                    # uncomment line
                    line = line[1:]
                if locale in line and line[0] == "#":
                    # uncomment line
                    line = line[1:]
                gen.write(line)

        libcalamares.utils.chroot_call(['locale-gen'])
        print('locale.gen done')

    locale_conf_path = os.path.join(install_path, "etc/locale.conf")
    with open(locale_conf_path, "w") as locale_conf:
        locale_conf.write('LANG=%s\n' % locale.split(' ')[0])

    return None