vnpy/tools/ci/format_check.py

import logging
import os

from yapf.yapflib.yapf_api import FormatFile

logger = logging.Logger(__file__)

if __name__ == "__main__":
    has_changed = False
    for root, dir, filenames in os.walk("vnpy"):
        for filename in filenames:
            basename, ext = os.path.splitext(filename)
            if ext == ".py":
                path = os.path.join(root, filename)
                reformatted_code, encoding, changed = FormatFile(
                    filename=path,
                    style_config=".style.yapf",
                    print_diff=True,
                    verify=False,
                    in_place=False,
                    logger=None,
                )
                if changed:
                    has_changed = True
                    logger.warning("File {} not formatted!".format(path))
                else:
                    logger.info("File {} is formatted!".format(path))
    exit(has_changed)