216 lines
8.7 KiB
Python
216 lines
8.7 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
# Copyright (c) 2021 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 sys
|
||
|
|
import os
|
||
|
|
import argparse
|
||
|
|
import shutil
|
||
|
|
import xml.etree.ElementTree as ET
|
||
|
|
sys.path.append(
|
||
|
|
os.path.dirname(os.path.dirname(os.path.dirname(
|
||
|
|
os.path.abspath(__file__)))))
|
||
|
|
from scripts.util.file_utils import write_json_file # noqa: E402
|
||
|
|
from scripts.util import build_utils # noqa: E402
|
||
|
|
|
||
|
|
|
||
|
|
def copy_dir(src, dest):
|
||
|
|
if not os.path.exists(src):
|
||
|
|
raise Exception("src dir '{}' doesn't exist.".format(src))
|
||
|
|
if not os.path.exists(dest):
|
||
|
|
os.makedirs(dest, exist_ok=True)
|
||
|
|
src_files = []
|
||
|
|
for root, _, files in os.walk(src):
|
||
|
|
for _file in files:
|
||
|
|
file_path = os.path.join(root, _file)
|
||
|
|
src_files.append(file_path)
|
||
|
|
for src_path in src_files:
|
||
|
|
if os.path.islink(src_path):
|
||
|
|
continue
|
||
|
|
file_relpath = os.path.relpath(src_path, src)
|
||
|
|
dest_path = os.path.join(dest, file_relpath)
|
||
|
|
dest_dir = os.path.dirname(dest_path)
|
||
|
|
if not os.path.exists(dest_dir):
|
||
|
|
os.makedirs(dest_dir, exist_ok=True)
|
||
|
|
shutil.copy2(src_path, dest_path)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
def _resources_with_xml_v1(root, testcase_target_name, test_resource_path,
|
||
|
|
part_build_out_path, resource_output_path):
|
||
|
|
_out_resources_list = []
|
||
|
|
for target in root:
|
||
|
|
if target.attrib.get('name') != testcase_target_name:
|
||
|
|
continue
|
||
|
|
for _depend in target:
|
||
|
|
_findpath = _depend.attrib.get('findpath')
|
||
|
|
_resource_file = _depend.attrib.get('resource')
|
||
|
|
if _findpath == 'res':
|
||
|
|
_resource_src = os.path.join(test_resource_path,
|
||
|
|
_resource_file)
|
||
|
|
_res_dest = os.path.join(resource_output_path, _resource_file)
|
||
|
|
elif _findpath == 'out':
|
||
|
|
if not os.path.exists(_resource_file):
|
||
|
|
__dir_name = _resource_file.split('/')[0]
|
||
|
|
_resource_file_new = os.path.join(__dir_name,
|
||
|
|
_resource_file)
|
||
|
|
_resource_src_new = os.path.join(part_build_out_path,
|
||
|
|
_resource_file_new)
|
||
|
|
if os.path.exists(_resource_src_new):
|
||
|
|
_resource_src = _resource_src_new
|
||
|
|
_res_dest = os.path.join(resource_output_path,
|
||
|
|
_resource_file)
|
||
|
|
else:
|
||
|
|
_resource_src = ''
|
||
|
|
_res_dest = ''
|
||
|
|
else:
|
||
|
|
_resource_src = os.path.join(part_build_out_path,
|
||
|
|
_resource_file)
|
||
|
|
_res_dest = os.path.join(resource_output_path,
|
||
|
|
_resource_file)
|
||
|
|
else:
|
||
|
|
raise Exception(
|
||
|
|
"resource findpath type '{}' not support.".format(
|
||
|
|
_findpath))
|
||
|
|
if _resource_src:
|
||
|
|
_out_resources_list.append({
|
||
|
|
"src":
|
||
|
|
os.path.relpath(_resource_src),
|
||
|
|
"dest":
|
||
|
|
os.path.relpath(_res_dest)
|
||
|
|
})
|
||
|
|
return _out_resources_list
|
||
|
|
|
||
|
|
|
||
|
|
def _parse_res_value(value):
|
||
|
|
res_file = value.split('->')[0].strip()
|
||
|
|
return res_file
|
||
|
|
|
||
|
|
|
||
|
|
def _resources_with_xml_v2(root, testcase_target_name, test_resource_path,
|
||
|
|
part_build_out_path, resource_output_path):
|
||
|
|
_out_resources_list = []
|
||
|
|
for target in root:
|
||
|
|
if target.attrib.get('name') != testcase_target_name:
|
||
|
|
continue
|
||
|
|
for child in target:
|
||
|
|
if child.tag != 'preparer':
|
||
|
|
continue
|
||
|
|
for _option in child:
|
||
|
|
if _option.attrib.get('name') != 'push':
|
||
|
|
continue
|
||
|
|
_src_type = _option.attrib.get('src')
|
||
|
|
_resource_file_val = _option.attrib.get('value')
|
||
|
|
_resource_file = _parse_res_value(_resource_file_val)
|
||
|
|
if _src_type == 'res':
|
||
|
|
_resource_src = os.path.join(test_resource_path,
|
||
|
|
_resource_file)
|
||
|
|
_res_dest = os.path.join(resource_output_path,
|
||
|
|
_resource_file)
|
||
|
|
elif _src_type == 'out':
|
||
|
|
_resource_src = os.path.join(part_build_out_path,
|
||
|
|
_resource_file)
|
||
|
|
_res_dest = os.path.join(resource_output_path,
|
||
|
|
_resource_file)
|
||
|
|
else:
|
||
|
|
raise Exception(
|
||
|
|
"resource src type '{}' not support.".format(
|
||
|
|
_src_type))
|
||
|
|
if _resource_src:
|
||
|
|
_out_resources_list.append({
|
||
|
|
"src":
|
||
|
|
os.path.relpath(_resource_src),
|
||
|
|
"dest":
|
||
|
|
os.path.relpath(_res_dest)
|
||
|
|
})
|
||
|
|
return _out_resources_list
|
||
|
|
|
||
|
|
|
||
|
|
def find_testcase_resources(resource_config_file, testcase_target_name,
|
||
|
|
test_resource_path, part_build_out_path,
|
||
|
|
resource_output_path):
|
||
|
|
if not os.path.exists(resource_config_file):
|
||
|
|
return []
|
||
|
|
tree = ET.parse(resource_config_file)
|
||
|
|
root = tree.getroot()
|
||
|
|
if root.attrib.get('ver') == '2.0':
|
||
|
|
_resources_list = _resources_with_xml_v2(root, testcase_target_name,
|
||
|
|
test_resource_path,
|
||
|
|
part_build_out_path,
|
||
|
|
resource_output_path)
|
||
|
|
else:
|
||
|
|
_resources_list = _resources_with_xml_v1(root, testcase_target_name,
|
||
|
|
test_resource_path,
|
||
|
|
part_build_out_path,
|
||
|
|
resource_output_path)
|
||
|
|
# copy ohos_test.xml
|
||
|
|
_resources_list.append({
|
||
|
|
"src":
|
||
|
|
resource_config_file,
|
||
|
|
"dest":
|
||
|
|
os.path.join(resource_output_path,
|
||
|
|
os.path.basename(resource_config_file))
|
||
|
|
})
|
||
|
|
return _resources_list
|
||
|
|
|
||
|
|
|
||
|
|
def copy_testcase_resources(resource_infos):
|
||
|
|
for resource_info in resource_infos:
|
||
|
|
src_file = resource_info.get('src')
|
||
|
|
if not os.path.exists(src_file):
|
||
|
|
print("warning: testcase resource {} doesn't exist.".format(
|
||
|
|
src_file))
|
||
|
|
return 0
|
||
|
|
dest_file = resource_info.get('dest')
|
||
|
|
dest_dir = os.path.dirname(dest_file)
|
||
|
|
if os.path.isdir(src_file):
|
||
|
|
copy_dir(src_file, dest_file)
|
||
|
|
else:
|
||
|
|
if not os.path.exists(dest_dir):
|
||
|
|
os.makedirs(dest_dir, exist_ok=True)
|
||
|
|
shutil.copy2(src_file, dest_file)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument('--resource-config-file', required=False)
|
||
|
|
parser.add_argument('--testcase-target-name', required=True)
|
||
|
|
parser.add_argument('--part-build-out-path', required=True)
|
||
|
|
parser.add_argument('--resource-output-path', required=True)
|
||
|
|
parser.add_argument('--output-file', required=True)
|
||
|
|
args = parser.parse_args()
|
||
|
|
if not args.resource_config_file:
|
||
|
|
return 0
|
||
|
|
else:
|
||
|
|
resource_config_file = args.resource_config_file
|
||
|
|
if not os.path.exists(resource_config_file):
|
||
|
|
raise Exception(
|
||
|
|
"testcase '{}' resource_config_file config incorrect.".format(
|
||
|
|
args.testcase_target_name))
|
||
|
|
|
||
|
|
test_resource_path = os.path.dirname(resource_config_file)
|
||
|
|
resources_list = find_testcase_resources(resource_config_file,
|
||
|
|
args.testcase_target_name,
|
||
|
|
test_resource_path,
|
||
|
|
args.part_build_out_path,
|
||
|
|
args.resource_output_path)
|
||
|
|
write_json_file(args.output_file, resources_list)
|
||
|
|
copy_testcase_resources(resources_list)
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
sys.exit(main())
|