xref: /interface/sdk-js/build-tools/api_label_detection/src/bin/config.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (c) 2024 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 enum
from coreImpl import detection_label


class ToolNameType(enum.Enum):
    DETECTION = 'detection'


tool_name_type_set = [
    member.value for name_tool,
    member in ToolNameType.__members__.items()
]


class FormatType(enum.Enum):
    JSON = 'json'
    EXCEL = 'excel'


format_set = [
    member.value for name_format,
    member in FormatType.__members__.items()
]


def run_tools(options):
    tool_name = options.tool_name
    if tool_name == ToolNameType["DETECTION"].value:
        detection_label.detection_label(options.check_labels, options.result_json_path, options.output_path)


class Config(object):
    name = 'parser'
    version = '0.1.0'
    description = 'Compare the parser the NDKS'
    commands = [
        {
            "name": "--tool-name",
            "abbr": "-N",
            "required": True,
            "choices": tool_name_type_set,
            "type": str,
            "default": ToolNameType["DETECTION"],
            "help": "工具名称"
        },
        {
            "name": "--check-labels",
            "abbr": "-L",
            "required": True,
            "type": str,
            "help": "需要校验的标签"
        },
        {
            "name": "--result-json-path",
            "abbr": "-P",
            "required": True,
            "type": str,
            "help": "解析结果json文件路径"
        },
        {
            "name": "--output-path",
            "abbr": "-O",
            "required": False,
            "type": str,
            "help": "输出路径"
        }
    ]