diff options
Diffstat (limited to 'lib/python')
| -rwxr-xr-x | lib/python/build_all.py | 76 |
1 files changed, 65 insertions, 11 deletions
diff --git a/lib/python/build_all.py b/lib/python/build_all.py index 652ff2994..df4592e2d 100755 --- a/lib/python/build_all.py +++ b/lib/python/build_all.py @@ -2,27 +2,81 @@ import subprocess import os import sys import re +import argparse + +# Parse arguments +parser = argparse.ArgumentParser(description="Build QMK for Sonix keyboards") +parser.add_argument("--whitelist", help="enables the whitelist (specify a filename)") +parser.add_argument("--blacklist", help="enables the blacklist (specify a filename)") +parser.add_argument("--debug", help="displays which keyboards are being excluded based on whitelist/blacklist", action="store_true") +args = parser.parse_args() + KEYBOARDS = [] # Search the repository for Sonix SN32F2 keyboard directories -command = "grep -rl 'MCU = SN32F2' | sed -e 's/keyboards\///g' -e 's/\/rules.mk//g'| sort" +command = "grep -rl 'MCU = SN32F2' | sed -e 's;keyboards/;;g' -e 's;/rules.mk;;g'| sort" ret = subprocess.run(command, capture_output=True, shell=True) BOARDS = ret.stdout.decode().split('\n') + +if args.debug: + print("using debug mode") + +if args.whitelist: + # Grab the list of whitelisted keyboards + whitelisted_kb_command = "cat " + args.whitelist + " | tr -d '\r'" + whitelisted_kb_ret = subprocess.run(whitelisted_kb_command, capture_output=True, shell=True) + WHITELISTED_BOARDS = whitelisted_kb_ret.stdout.decode().split('\n') + if args.debug: + print("using whitelist: ", args.whitelist) + print("whitelist contents: ", WHITELISTED_BOARDS) + +if args.blacklist: + # Grab the list of blacklisted keyboards + blacklisted_kb_command = "cat " + args.blacklist + " | tr -d '\r'" + blacklisted_kb_ret = subprocess.run(blacklisted_kb_command, capture_output=True, shell=True) + BLACKLISTED_BOARDS = blacklisted_kb_ret.stdout.decode().split('\n') + if args.debug: + print("using blacklist: ", args.blacklist) + print("blacklist contents: ", BLACKLISTED_BOARDS) + + def main(): for line in BOARDS: # We need to manipulate some non-standard directories - if line.strip() != "" and line.strip() != "lib/python/build_all.py": - if re.match("^(gmmk)",line.strip()): - KEYBOARDS.append(line.strip()+"/rev2") - KEYBOARDS.append(line.strip()+"/rev3") - if re.match("^(keychron/k)",line.strip()): + if should_include(line): + if re.match("^(gmmk)", line.strip()): + KEYBOARDS.append(line.strip() + "/rev2") + KEYBOARDS.append(line.strip() + "/rev3") + if re.match("^(keychron/k)", line.strip()): KEYBOARDS.append(line.strip()) # keychron K series white don't have yet via/optical support - if re.match("(?!.*white)",line.strip()): - KEYBOARDS.append(line.strip()+"/via") - KEYBOARDS.append(line.strip()+"/optical") - KEYBOARDS.append(line.strip()+"/optical_via") - else: KEYBOARDS.append(line.strip()) + if re.match("(?!.*white)", line.strip()): + KEYBOARDS.append(line.strip() + "/via") + KEYBOARDS.append(line.strip() + "/optical") + KEYBOARDS.append(line.strip() + "/optical_via") + else: + KEYBOARDS.append(line.strip()) + if args.debug: + print('Filtered and processed boards: ', KEYBOARDS) + + +def should_include(keyboard): + if keyboard.strip() == "": + return False + if keyboard.strip() == "lib/python/build_all.py": + return False + if args.blacklist: + if keyboard.strip() in BLACKLISTED_BOARDS: + if args.debug: + print("Skipping blacklisted keyboard: ", keyboard.strip()) + return False + if args.whitelist: + if keyboard.strip() not in WHITELISTED_BOARDS: + if args.debug: + print("Skipping non-whitelisted keyboard: ", keyboard.strip()) + return False + return True + if __name__ == '__main__': main() |