traderx/config.py

import configargparse
import logging

from chive.config import config_parse

_logger = logging.getLogger(__name__)


def config_argparse():
    parser = configargparse.ArgumentParser(
        # default_config_files=['config.yaml', 'config.ini'],
        default_config_files=['config.yaml'],
        config_file_parser_class=configargparse.YAMLConfigFileParser,  # 使用YAML解析器
        auto_env_var_prefix='TRADERX_'  # 环境变量前缀
    )

    common_group = parser.add_argument_group('Common')
    common_group.add_argument('--app_name', default='traderx', help='TraderXApp name')
    common_group.add_argument('--log_level', choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'], default='INFO')
    common_group.add_argument('--zookeeper_url', default='192.168.3.3:2181', help='Zookeeper URL')
    common_group.add_argument('--pulsar_url', default='pulsar://192.168.3.3:6650', help='Pulsar URL')
    common_group.add_argument('--rpc_server', default='tcp://10.0.0.1:9483', help='RPC Server URL')

    tdx_group = parser.add_argument_group('TRADER')
    tdx_group.add_argument('--trader.account', help='Tdx Main Folder')

    tdx_group = parser.add_argument_group('TDX')
    tdx_group.add_argument('--tdx.main_folder', default='C:/new_tdx', help='Tdx Main Folder')
    tdx_group.add_argument('--tdx.export_concept_file', default=u'T0002/export/概念板块.txt', help='Concepts file.')
    tdx_group.add_argument('--tdx.watch_files', default=['tdx_block.yaml', 'pecat_block.yaml'], help='Files to watch.')
    tdx_group.add_argument('--tdx.local_host', default='10.0.0.1:50051', help='Service host')

    return config_parse(parser)


config = config_argparse()