format

format.sh

@@ -0,0 +1,20 @@
+#!/usr/bin/env bash
+# Usage: at the root dir >> bash scripts/format.sh
+
+# Check yapf version.
+ver=$(yapf --version)
+if ! echo $ver | grep -q 0.30.0; then
+  echo "Wrong YAPF version installed: 0.30.0 is required, not $ver. Please install via `pip install yapf==0.30.0`"
+  exit 1
+fi
+
+yapf --in-place --recursive -p --verbose --style .style.yapf scenarionet/
+
+if [[ "$1" == '--test' ]]; then # Only for CI usage, user should not use --test flag.
+  if ! git diff --quiet &>/dev/null; then
+    echo '*** You have not formatted your code! Please run [bash format.sh] at root directory before commit! Thanks! ***'
+    exit 1
+  else
+    echo "Code style test passed!"
+  fi
+fi