Mini Shell

#!/opt/cloudlinux/venv/bin/python3
# coding:utf-8

# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
from __future__ import print_function
from __future__ import division
from __future__ import absolute_import
import copy
import argparse
import os
import json

BASE_DIR = os.path.join(os.path.dirname(__file__), "../")
USER_LOCALES_SOURCE = '/usr/share/l.v.e-locales/'
MAP = {
    'cPanel':       BASE_DIR + 'cpanel/spa/assets/i18n',
    'Plesk':        BASE_DIR + 'plesk/htdocs/modules/plesk-lvemanager/i18n/',
    'DirectAdmin':  BASE_DIR + 'directadmin/lvemanager_spa/images/assets/i18n/',
}

I18N_SOURCE = BASE_DIR + 'commons/spa-resources/i18n/'

def main():
    """
    Analyze arguments and merge locale files
    :return:
    """
    parser = argparse.ArgumentParser(
        description='Convert yaml locales to json')

    parser.add_argument(
        '--panel', '-p',
        type=str,
        help="The target control panel"
    )

    args = parser.parse_args()

    if not os.path.isdir(USER_LOCALES_SOURCE):
        return
    panel_path = MAP.get(args.panel, I18N_SOURCE)
    for locale in available_locales(USER_LOCALES_SOURCE):
        user_locale = read_locale_json(USER_LOCALES_SOURCE, locale)
        current_locale = read_locale_json(I18N_SOURCE, locale)
        if current_locale:
            current_locale = nested_merge(copy.deepcopy(current_locale), user_locale)
        write_locale_json(panel_path, locale, current_locale)

def available_locales(path):
    """
    Get list of locales in directory
    :param path: directory with locales
    :return: list of locales
    """
    for locale in os.listdir(path):
        filename = '{}/{}'.format(path, locale)
        if os.path.isfile(filename) and filename.endswith('.json'):
            yield locale.split('.')[0]

def read_locale_json(path, locale):
    """
    read locale from directory
    :param path: directory with locales
    :param locale: locale (en, etc)
    :return: locale data
    """
    filename = '{}/{}.json'.format(path, locale)
    if os.path.isfile(filename):
        with open(filename, 'r') as f:
            return json.load(f)


def write_locale_json(path, locale, data):
    """
    Save json into file
    :param path: directory with locales
    :param locale: locale (en, etc)
    :param data: locale dict
    :return:
    """
    with open('{}/{}.json'.format(path, locale), 'w') as f:
        return json.dump(data, f, indent=4)

def nested_merge(a, b, path=None):
    """
    Merger a into b
    :param a: dict
    :param b: dict
    :return: merger dict
    """
    path = path or []
    for key in b:
        if key in a:
            if isinstance(a[key], dict) and isinstance(b[key], dict):
                nested_merge(a[key], b[key], path + [str(key)])
            else:
                a[key] = b[key]
        else:
            a[key] = b[key]
    return a

if __name__ == '__main__':
    main()