vnpy/binding/generator/generate_ctp.py

import logging
import os

from adaptor.ctpadaptor import CtpAdaptor
from autocxxpy.generator import GeneratorOptions, Generator

logger = logging.getLogger(__file__)


def clear_dir(path: str):
    for file in os.listdir(path):
        os.unlink(os.path.join(path, file))


def main():
    options: GeneratorOptions = CtpAdaptor(
        "../source/ctp/api/ThostFtdcMdApi.h",
        "../source/ctp/api/ThostFtdcTraderApi.h",
    ).parse()

    options.includes.append("api/ThostFtdcTraderApi.h")
    options.includes.append("api/ThostFtdcMdApi.h")
    options.split_in_files = True
    options.module_name = "vnctp"
    options.max_classes_in_one_file = 100

    saved_files = Generator(options=options).generate()
    output_dir = "./generated_files"
    # clear output dir
    if not os.path.exists(output_dir):
        os.mkdir(output_dir)
    clear_dir(output_dir)

    for name, data in saved_files.items():
        with open(f"{output_dir}/{name}", "wt") as f:
            f.write(data)


if __name__ == "__main__":
    main()