29 lines
940 B
Python
29 lines
940 B
Python
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)
|