15f9996aaSopenharmony_ci#!/usr/bin/env python3
25f9996aaSopenharmony_ci# -*- coding: utf-8 -*-
35f9996aaSopenharmony_ci# Copyright (c) 2021 Huawei Device Co., Ltd.
45f9996aaSopenharmony_ci# Licensed under the Apache License, Version 2.0 (the "License");
55f9996aaSopenharmony_ci# you may not use this file except in compliance with the License.
65f9996aaSopenharmony_ci# You may obtain a copy of the License at
75f9996aaSopenharmony_ci#
85f9996aaSopenharmony_ci#     http://www.apache.org/licenses/LICENSE-2.0
95f9996aaSopenharmony_ci#
105f9996aaSopenharmony_ci# Unless required by applicable law or agreed to in writing, software
115f9996aaSopenharmony_ci# distributed under the License is distributed on an "AS IS" BASIS,
125f9996aaSopenharmony_ci# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
135f9996aaSopenharmony_ci# See the License for the specific language governing permissions and
145f9996aaSopenharmony_ci# limitations under the License.
155f9996aaSopenharmony_ci
165f9996aaSopenharmony_ciimport argparse
175f9996aaSopenharmony_ciimport os
185f9996aaSopenharmony_ciimport sys
195f9996aaSopenharmony_ciimport json
205f9996aaSopenharmony_ci
215f9996aaSopenharmony_ci
225f9996aaSopenharmony_cidef merge_files(args):
235f9996aaSopenharmony_ci    products_dir = "../../productdefine/common/products"
245f9996aaSopenharmony_ci    device_dir = "../../productdefine/common/device"
255f9996aaSopenharmony_ci    products_path = r'%s/%s' % (products_dir, args)
265f9996aaSopenharmony_ci    with open(products_path, 'r+', encoding='utf-8') as f:
275f9996aaSopenharmony_ci        data = json.load(f)
285f9996aaSopenharmony_ci        name = data["product_name"]
295f9996aaSopenharmony_ci        company = data["product_company"]
305f9996aaSopenharmony_ci        device = data["product_device"]
315f9996aaSopenharmony_ci    device_path = os.path.join(device_dir, f"{device}.json")
325f9996aaSopenharmony_ci    path_str = os.path.join("../../vendor/", company, name)
335f9996aaSopenharmony_ci    path = os.path.join(path_str)
345f9996aaSopenharmony_ci    new_file_name = os.path.join(path, "config.json")
355f9996aaSopenharmony_ci    if os.path.exists(path):
365f9996aaSopenharmony_ci        pass
375f9996aaSopenharmony_ci    else:
385f9996aaSopenharmony_ci        os.mkdir(path)
395f9996aaSopenharmony_ci
405f9996aaSopenharmony_ci    with open(device_path, "r", encoding='utf-8') as device_read:
415f9996aaSopenharmony_ci        data_device_read = json.load(device_read)
425f9996aaSopenharmony_ci
435f9996aaSopenharmony_ci    with open(products_path, "r", encoding='utf-8') as products_read:
445f9996aaSopenharmony_ci        data_products_read = json.load(products_read)
455f9996aaSopenharmony_ci
465f9996aaSopenharmony_ci    data_all = merge(data_device_read, data_products_read)
475f9996aaSopenharmony_ci    new_json = json.dumps(data_all, indent=4)
485f9996aaSopenharmony_ci
495f9996aaSopenharmony_ci    flags = os.O_RDWR | os.O_CREAT
505f9996aaSopenharmony_ci    modes = stat.S_IWUSR | stat.S_IRUSR
515f9996aaSopenharmony_ci    with os.fdopen(os.open(new_file_name, flags, modes), "w") as new_write:
525f9996aaSopenharmony_ci        new_write.write(new_json)
535f9996aaSopenharmony_ci    readjson(new_file_name, device)
545f9996aaSopenharmony_ci
555f9996aaSopenharmony_ci
565f9996aaSopenharmony_cidef readjson(path, device):
575f9996aaSopenharmony_ci    subsystems_list = list()
585f9996aaSopenharmony_ci    config_dic = {}
595f9996aaSopenharmony_ci    with open(path, 'r+', encoding='utf-8') as f:
605f9996aaSopenharmony_ci        data = json.load(f)
615f9996aaSopenharmony_ci        parts = data['parts']
625f9996aaSopenharmony_ci        subsystem_list = list()
635f9996aaSopenharmony_ci        for key in parts:
645f9996aaSopenharmony_ci            substr = str(key).split(":")
655f9996aaSopenharmony_ci            if substr[0] not in subsystem_list:
665f9996aaSopenharmony_ci                subsystem_list.append(substr[0])
675f9996aaSopenharmony_ci                components_list = list()
685f9996aaSopenharmony_ci                for key_sub, value_sub in parts.items():
695f9996aaSopenharmony_ci                    features = []
705f9996aaSopenharmony_ci                    for value_fea in value_sub.values():
715f9996aaSopenharmony_ci                        for k, v in value_fea.items():
725f9996aaSopenharmony_ci                            fea = "{} = {}".format(str(k), str(v).lower())
735f9996aaSopenharmony_ci                            features.append(fea)
745f9996aaSopenharmony_ci                    if substr[0] == str(key_sub).split(":")[0]:
755f9996aaSopenharmony_ci                        components_list.append({"component": str(key_sub).split(":")[1], "features": features})
765f9996aaSopenharmony_ci                subsystems_list.append({"subsystem": substr[0], "components": components_list})
775f9996aaSopenharmony_ci        config_dic["subsystems"] = subsystems_list
785f9996aaSopenharmony_ci        del data['parts']
795f9996aaSopenharmony_ci        data.update({"version": "3.0"})
805f9996aaSopenharmony_ci        data.update({"board": device})
815f9996aaSopenharmony_ci        data.update(config_dic)
825f9996aaSopenharmony_ci        for datakey in data:
835f9996aaSopenharmony_ci            if "enable_ramdisk" in datakey:
845f9996aaSopenharmony_ci                dict_keys = ["product_name", "device_company", "device_build_path", "target_cpu", "type", "version",
855f9996aaSopenharmony_ci                             "board", "enable_ramdisk", "enable_absystem", "subsystems"]
865f9996aaSopenharmony_ci                break
875f9996aaSopenharmony_ci            else:
885f9996aaSopenharmony_ci                dict_keys = ["product_name", "device_company", "device_build_path", "target_cpu", "type", "version",
895f9996aaSopenharmony_ci                             "board", "enable_absystem", "subsystems"]
905f9996aaSopenharmony_ci        dict_you_want = {new_key: data[new_key] for new_key in dict_keys}
915f9996aaSopenharmony_ci        json_data = json.dumps(dict_you_want, indent=2)
925f9996aaSopenharmony_ci        f.seek(0)
935f9996aaSopenharmony_ci        f.write(json_data)
945f9996aaSopenharmony_ci        f.truncate()
955f9996aaSopenharmony_ci
965f9996aaSopenharmony_ci
975f9996aaSopenharmony_cidef merge(dict1, dict2):
985f9996aaSopenharmony_ci    res = {**dict1, **dict2}
995f9996aaSopenharmony_ci    return res
1005f9996aaSopenharmony_ci
1015f9996aaSopenharmony_ci
1025f9996aaSopenharmony_cidef main(args):
1035f9996aaSopenharmony_ci    merge_files(args)
1045f9996aaSopenharmony_ci
1055f9996aaSopenharmony_ci
1065f9996aaSopenharmony_ciif __name__ == '__main__':
1075f9996aaSopenharmony_ci    main(sys.argv[1])
108