下载
中文
注册

创建客户端

首先可以创建create_client函数,将函数所在文件命名为utils.py,后续可持续使用该方法;建议url和management_url的默认值为MindIE Server配置的IP和端口。

import argparse
from mindieclient.python.httpclient import MindIEHTTPClient
def create_client(request_count=1):
    # get argument
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-u",
        "--url",
        required=False,
        default="https://127.0.0.1:1025",
        help="MindIE-Server URL.",
    )
    parser.add_argument(
        "-mu",
        "--management_url",
        required=False,
        default="https://127.0.0.2:1026",
        help="MindIE-Server management URL.",
    )
    parser.add_argument(
        "-v",
        "--verbose",
        action="store_true",
        required=False,
        default=True,
        help="Enable detailed information output.",
    )
    parser.add_argument(
        "-s",
        "--ssl",
        action="store_true",
        required=False,
        default=False,
        help="Enable encrypted link with https",
    )
    parser.add_argument(
        "-ca",
        "--ca_certs",
        required=False,
        default="ca.pem",
        help="Provide https ca certificate.",
    )
    parser.add_argument(
        "-key",
        "--key_file",
        required=False,
        default="client.key.pem",
        help="Provide https client certificate.",
    )
    parser.add_argument(
        "-cert",
        "--cert_file",
        required=False,
        default="client.pem",
        help="Provide https client keyfile.",
    )
    parser.add_argument(
        "-crl",
        "--crl_file",
        required=False,
        default="crl.pem",
        help="Provide https crlfile.",
    )
    args = parser.parse_args()
    # create client
    try:
        if args.ssl:
            ssl_options = {}
            if args.ca_certs is not None:
                ssl_options["ca_certs"] = args.ca_certs
            if args.key_file is not None:
                ssl_options["keyfile"] = args.key_file
            if args.cert_file is not None:
                ssl_options["certfile"] = args.cert_file
            if args.crl_file is not None:
                ssl_options["crlfile"] = args.crl_file
            mindie_client = MindIEHTTPClient(
                url=args.url,
                verbose=args.verbose,
                enable_ssl=True,
                ssl_options=ssl_options,
                concurrency=request_count,
                management_url=args.management_url
            )
        else:
            mindie_client = MindIEHTTPClient(
                url=args.url, verbose=args.verbose,
                enable_ssl=False,concurrency=request_count,
                management_url=args.management_url
            )
    except Exception as e:
        raise e
    return mindie_client