From 0b7e8576147bb3a1b5b6cf6a9c4afcd8ea8958a2 Mon Sep 17 00:00:00 2001
From: Scott Lahteine <github@thinkyhead.com>
Date: Wed, 3 Mar 2021 20:48:58 -0600
Subject: [PATCH] Update some py scripts

---
 .../PlatformIO/scripts/fysetc_cheetah_v20.py  |  4 ++
 .../PlatformIO/scripts/preflight-checks.py    | 68 +++++++++----------
 2 files changed, 38 insertions(+), 34 deletions(-)

diff --git a/buildroot/share/PlatformIO/scripts/fysetc_cheetah_v20.py b/buildroot/share/PlatformIO/scripts/fysetc_cheetah_v20.py
index 10471d375..f36d51f97 100644
--- a/buildroot/share/PlatformIO/scripts/fysetc_cheetah_v20.py
+++ b/buildroot/share/PlatformIO/scripts/fysetc_cheetah_v20.py
@@ -1,3 +1,7 @@
+#
+# STM32F401VE_STEVAL.py
+# Customizations for env:STM32F401VE_STEVAL
+#
 import os
 Import("env")
 
diff --git a/buildroot/share/PlatformIO/scripts/preflight-checks.py b/buildroot/share/PlatformIO/scripts/preflight-checks.py
index c9bbb42ab..4c8a00e74 100644
--- a/buildroot/share/PlatformIO/scripts/preflight-checks.py
+++ b/buildroot/share/PlatformIO/scripts/preflight-checks.py
@@ -6,41 +6,41 @@ import os,re
 Import("env")
 
 def get_envs_for_board(board):
-    if board.startswith("BOARD_"):
-        board = board[6:]
-    with open(os.path.join("Marlin", "src", "pins", "pins.h"),"r") as f:
-        board_found = ""
-        r=re.compile(r"if\s+MB\((.+)\)")
-        for line in f.readlines():
-            mbs = r.findall(line)
-            if mbs:
-                board_found = board if board in re.split(r",\s*", mbs[0]) else ""
-            if board_found and "#include " in line and "env:" in line:
-                return re.findall(r"env:\w+", line)
-    return []
+	if board.startswith("BOARD_"):
+		board = board[6:]
+	with open(os.path.join("Marlin", "src", "pins", "pins.h"),"r") as f:
+		board_found = ""
+		r=re.compile(r"if\s+MB\((.+)\)")
+		for line in f.readlines():
+			mbs = r.findall(line)
+			if mbs:
+				board_found = board if board in re.split(r",\s*", mbs[0]) else ""
+			if board_found and "#include " in line and "env:" in line:
+				return re.findall(r"env:\w+", line)
+	return []
 
 def check_envs(build_env, base_envs, config):
-    if build_env in base_envs:
-        return True
-    ext = config.get(build_env, 'extends', default=None)
-    if ext:
-        if isinstance(ext, str):
-            return check_envs(ext, base_envs, config)
-        elif isinstance(ext, list):
-            for ext_env in ext:
-                if check_envs(ext_env, base_envs, config):
-                    return True
-    return False
+	if build_env in base_envs:
+		return True
+	ext = config.get(build_env, 'extends', default=None)
+	if ext:
+		if isinstance(ext, str):
+			return check_envs(ext, base_envs, config)
+		elif isinstance(ext, list):
+			for ext_env in ext:
+				if check_envs(ext_env, base_envs, config):
+					return True
+	return False
 
 # Sanity checks:
 if 'PIOENV' not in env:
-    raise SystemExit("Error: PIOENV is not defined. This script is intended to be used with PlatformIO")
+	raise SystemExit("Error: PIOENV is not defined. This script is intended to be used with PlatformIO")
 
 if 'MARLIN_FEATURES' not in env:
-    raise SystemExit("Error: this script should be used after common Marlin scripts")
+	raise SystemExit("Error: this script should be used after common Marlin scripts")
 
 if 'MOTHERBOARD' not in env['MARLIN_FEATURES']:
-    raise SystemExit("Error: MOTHERBOARD is not defined in Configuration.h")
+	raise SystemExit("Error: MOTHERBOARD is not defined in Configuration.h")
 
 build_env = env['PIOENV']
 motherboard = env['MARLIN_FEATURES']['MOTHERBOARD']
@@ -49,16 +49,16 @@ config = env.GetProjectConfig()
 result = check_envs("env:"+build_env, base_envs, config)
 
 if not result:
-    err = "Error: your selected build environment '%s' is not compatible with MOTHERBOARD=%s in Configuration.h. " \
-          "Please use one of compatible build environments for this board: %s" % \
-          (build_env, motherboard, ",".join([e[4:] for e in base_envs if e.startswith("env:")]))
-    raise SystemExit(err)
+	err = "Error: your selected build environment '%s' is not compatible with MOTHERBOARD=%s in Configuration.h. " \
+		  "Please use one of compatible build environments for this board: %s" % \
+		  (build_env, motherboard, ",".join([e[4:] for e in base_envs if e.startswith("env:")]))
+	raise SystemExit(err)
 
 #
 # Check for Config files in two common incorrect places
 #
 for p in [ env['PROJECT_DIR'], os.path.join(env['PROJECT_DIR'], "config") ]:
-    for f in [ "Configuration.h", "Configuration_adv.h" ]:
-        if os.path.isfile(os.path.join(p, f)):
-            err = "ERROR: Config files found in directory %s. Please move them into the Marlin subfolder." % p
-            raise SystemExit(err)
+	for f in [ "Configuration.h", "Configuration_adv.h" ]:
+		if os.path.isfile(os.path.join(p, f)):
+			err = "ERROR: Config files found in directory %s. Please move them into the Marlin subfolder." % p
+			raise SystemExit(err)