1#!/usr/bin/env python3 2# -*- coding: utf-8 -*- 3 4# Copyright (c) 2021-2024 Huawei Device Co., Ltd. 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16# 17 18from functools import cached_property 19from typing import Dict, Union 20 21from runner.options.decorator_value import value, _to_str, _to_bool, _to_int 22 23 24class ETSOptions: 25 __DEFAULT_COMPARE_FILES_ITERATIONS = 2 26 27 def __str__(self) -> str: 28 return _to_str(self, 1) 29 30 def to_dict(self) -> Dict[str, Union[bool, int]]: 31 return { 32 "force-generate": self.force_generate, 33 "compare-files": self.compare_files, 34 "compare-files-iterations": self.compare_files_iterations, 35 } 36 37 @cached_property 38 @value(yaml_path="ets.force-generate", cli_name="is_force_generate", cast_to_type=_to_bool) 39 def force_generate(self) -> bool: 40 return False 41 42 @cached_property 43 @value(yaml_path="ets.compare-files", cli_name="compare_files", cast_to_type=_to_bool) 44 def compare_files(self) -> bool: 45 return False 46 47 @cached_property 48 @value(yaml_path="ets.compare-files-iterations", cli_name="compare_files_iterations", cast_to_type=_to_int) 49 def compare_files_iterations(self) -> int: 50 return self.__DEFAULT_COMPARE_FILES_ITERATIONS 51 52 def get_command_line(self) -> str: 53 options = [ 54 '--force-generate' if self.force_generate else '', 55 '---compare-files' if self.compare_files else '', 56 f'--compare-files-iterations={self.compare_files_iterations}', 57 ] 58 return ' '.join(options) 59