[Add] added format_check.py

2019-01-07 02:42:55 -04:00 · 2019-01-07 02:42:55 -04:00 · bf2565920f
commit bf2565920f
parent 665f515d86
1 changed files with 27 additions and 0 deletions
--- a/travis/format_check.py
+++ b/travis/format_check.py
@ -0,0 +1,27 @@
+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)