52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
|
import logging
|
||
|
|
import os
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
from tools.check.utils import check_and_warning
|
||
|
|
|
||
|
|
logger = logging.Logger(__file__)
|
||
|
|
|
||
|
|
|
||
|
|
def check_by_yapf():
|
||
|
|
from yapf.yapflib.yapf_api import FormatFile
|
||
|
|
|
||
|
|
passed = True
|
||
|
|
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:
|
||
|
|
passed = False
|
||
|
|
logger.warning("File {} not formatted!".format(path))
|
||
|
|
else:
|
||
|
|
logger.info("File {} is formatted!".format(path))
|
||
|
|
return passed
|
||
|
|
|
||
|
|
|
||
|
|
def check_black():
|
||
|
|
passed = True
|
||
|
|
try:
|
||
|
|
subprocess.check_call(["black", "--check", "./"])
|
||
|
|
except subprocess.SubprocessError:
|
||
|
|
passed = False
|
||
|
|
return passed
|
||
|
|
|
||
|
|
|
||
|
|
def check_format():
|
||
|
|
return check_and_warning(check_black)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if not check_format():
|
||
|
|
exit(1)
|
||
|
|
exit(0)
|