28 lines
1.2 KiB
Python
28 lines
1.2 KiB
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)
|