#!/usr/bin/env python3 # coding=utf-8 # # Copyright (c) 2020-2022 Huawei Device Co., Ltd. # 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. # import os import json from xdevice import platform_logger from core.utils import get_file_list_by_postfix from core.utils import get_build_output_path from core.config.parse_parts_config import ParsePartsConfig LOG = platform_logger("SelectTargets") ############################################################################## ############################################################################## class SelectTargets(object): def __init__(self, project_rootpath): self.project_rootpath = project_rootpath @classmethod def _get_mlf_data_from_file(cls, filepath): data_list = [] if os.path.exists(filepath): with open(filepath, 'r') as mlf_file: data_list = json.load(mlf_file) if not data_list: LOG.warning("The %s file load error." % filepath) data_list = [] return data_list @classmethod def _get_part_path_data(cls, productform): part_path_dic = {} parser = ParsePartsConfig(productform) # 获取infos_for_testfwk.json配置文件中“phone”节点下“part_infos”节点数据 part_infos = parser.get_part_infos() if part_infos is None: LOG.error("part_infos is None.") return part_path_dic for part_name in part_infos: part_info = part_infos.get(part_name, None) if part_info is None: continue origin_part_name = part_info.get("origin_part_name") build_out_dir = part_info.get("build_out_dir") part_path_list = [] # default_part_path：~/OpenHarmony/out/rk3568/module_list_files/部件名（以rk3568举例） default_part_path = os.path.join( get_build_output_path(productform), "module_list_files", origin_part_name) if os.path.exists(default_part_path): part_path_list.append(default_part_path) # 如果build_out_dir不是当前目录，将新目录加到part_path_list中 if build_out_dir != ".": product_part_path = os.path.join( get_build_output_path(productform), build_out_dir, "module_list_files", origin_part_name) if os.path.exists(product_part_path): part_path_list.append(product_part_path) part_path_dic[part_name] = part_path_list return part_path_dic def get_build_targets(self, productform, typelist, partlist, testmodule): target_list = [] if productform == "" or len(typelist) == 0: LOG.warning("Error: productform or typelist is empty.") return [] if len(partlist) == 0 and testmodule != "": LOG.warning( "The part cannot be empty When the module is not empty.") return [] # productform不为空，typelist（test type[UT,MST,ST,PERF,ALL]）不为空 # partlist和testmodule为空，通过testtype获取部件列表 if len(partlist) == 0 and testmodule == "": target_list = self._get_target_list_by_type(productform, typelist) return target_list # productform不为空，typelist（test type[UT,MST,ST,PERF,ALL]）不为空 # partlist不为空，testmodule为空，通过testtype、partlist一起获取部件列表 if len(partlist) != 0 and testmodule == "": target_list = self._get_target_list_by_part(productform, typelist, partlist) return target_list # productform不为空，typelist（test type[UT,MST,ST,PERF,ALL]）不为空 # partlist不为空，testmodule不为空，通过testtype、partlist、testmodule一起获取部件列表 if len(partlist) != 0 and testmodule != "": target_list = self._get_target_list_by_module(productform, typelist, partlist, testmodule) return target_list # 通过infos_for_testfwk.json文件获取所有子部件信息编译目录信息： # [{“部件名1”：[~/OpenHarmony/out/rk3568/module_list_files/部件名1]}] # 然后遍历这些目录中的mlf文件，获取其中定义的label，返回label集合 # 遍历时通过testmodule控制遍历的部件指定模块目录，如果不定义，则遍历子部件下面所有模块目录 # 遍历时通过partlist控制遍历指定部件目录，如果不定义，则遍历infos_for_testfwk.json文件中定义的所有子部件目录 def filter_build_targets(self, para): productform = para.productform typelist = para.testtype partlist = para.partname_list testmodule = para.testmodule print("partlist = %s" % str(partlist)) target_list = self.get_build_targets(productform, typelist, partlist, testmodule) return target_list def _get_target_list_from_path(self, typelist, check_path): target_list = [] if os.path.exists(check_path): # 获取部件编译输出目录（~/OpenHarmony/out/rk3568/module_list_files/部件名1）中.mlf文件列表 mlf_file_list = get_file_list_by_postfix( check_path, ".mlf") # 遍历mlf_file_list中所有目录下面mlf文件中的label列表 for filepath in mlf_file_list: # 获取mlf文件中的JSON数据信息列表 mlf_info_list = self._get_mlf_data_from_file(filepath) for data in mlf_info_list: # 举例："test_type": "moduletest" test_type = data.get("test_type") # 举例："label": "//base/accessibility/services/test:aams_accessibility_keyevent_filter_test # (//build/toolchain/ohos:ohos_clang_arm)" target_path = data.get("label") if "ALL" in typelist: target_list.append(target_path) continue if test_type in typelist: target_list.append(target_path) return target_list def _get_target_list_by_type(self, productform, typelist): target_list = [] # 获取所有部件编译输出信息列表：[{“部件名1”：[~/OpenHarmony/out/rk3568/module_list_files/部件名1]}] # 或者{“部件名1”：[~/OpenHarmony/out/rk3568/module_list_files/部件名1， # ~/OpenHarmony/out/rk3568/编译目录build_out_dir/module_list_files/部件名1]} part_path_dic = self._get_part_path_data(productform) for item in part_path_dic: part_path_list = part_path_dic.get(item) for part_path in part_path_list: print("part_path = %s" % part_path) temp_list = self._get_target_list_from_path(typelist, part_path) target_list.extend(temp_list) return target_list def _get_target_list_by_part(self, productform, typelist, partlist): target_list = [] part_path_dic = self._get_part_path_data(productform) for partname in partlist: part_path_list = part_path_dic.get(partname, []) for part_path in part_path_list: temp_list = self._get_target_list_from_path(typelist, part_path) target_list.extend(temp_list) return target_list def _get_target_list_by_module(self, productform, typelist, partlist, testmodule): target_list = [] part_path_dic = self._get_part_path_data(productform) for partname in partlist: part_path_list = part_path_dic.get(partname, []) for part_path in part_path_list: module_path = os.path.join(part_path, testmodule) LOG.info("module_path = %s." % module_path) if os.path.exists(module_path): temp_list = self._get_target_list_from_path(typelist, module_path) target_list.extend(temp_list) return target_list ############################################################################## ##############################################################################