Cleaning up - Update python script and readme

rbuild.py

@@ -1,13 +1,139 @@
+#!/usr/bin/env python3
+
 import argparse
+import subprocess
+import os
+import sys
 
-parser = argparse.ArgumentParser(
-    prog='rbuild',
-    description='CLI-based helper utility to build project items',
-    epilog='That\'s all folks!')
 
-parser.add_argument('build')
-#parser.add_argument('-c', '--count')      # option that takes a value
-#parser.add_argument('-v', '--verbose', action='store_true')  # on/off flag
+# For actual dimensions, please see profiles.scad.
+class BuildSizeConfig:
+    MINI = 'mini'
+    MICRO = 'micro'
+    DEFAULT = 'default'
 
-args = parser.parse_args()
-print(args.build)
+
+RACK_BUILD_DIR = './rack/print'
+RACK_MOUNT_BUILD_DIR = './rack-mount/print'
+
+RACK_BUILD_TARGET_DIR = './stl/rack'
+RACK_MOUNT_BUILD_TARGET_DIR = './stl/rack-mount'
+
+
+def main():
+    if not assert_os():
+        print("Linux Required for OpenSCAD CLI")
+        return
+
+    parser = argparse.ArgumentParser(
+        prog='rbuild',
+        description='CLI-based helper utility to build project items. '
+                    'This includes both the rack and also rack-mount items',
+        epilog='That\'s all folks!'
+    )
+
+    parser.add_argument(
+        '-b',
+        nargs='?',
+        const='all',
+        help='Build command. Optionally pass in a specific part name to only build that item. '
+             'Defaults to building everything'
+    )
+
+    parser.add_argument(
+        '-c',
+        default='default',
+        choices=[BuildSizeConfig.MINI, BuildSizeConfig.MICRO, BuildSizeConfig.DEFAULT],
+        help='Build size config profile. This will determine the size of the rack you wish to generate. '
+             'For actual dimensions, please see profiles.scad.'
+    )
+
+    args = parser.parse_args()
+
+    build_var = args.b
+    config_var = args.c
+
+    run_build(build_var, config_var)
+
+
+def run_build(build_var, config_var):
+    if build_var == 'all':
+        for dir_file in os.listdir(RACK_BUILD_DIR):
+            build_single(RACK_BUILD_DIR, RACK_BUILD_TARGET_DIR, dir_file, config_var)
+
+        for dir_file in os.listdir(RACK_MOUNT_BUILD_DIR):
+            build_single(RACK_MOUNT_BUILD_DIR, RACK_MOUNT_BUILD_TARGET_DIR, dir_file, config_var)
+
+        return
+
+    filename_rack = find_rack(build_var)
+    filename_rack_mount = find_rack_mount(build_var)
+
+    if not (filename_rack or filename_rack_mount):
+        print('File:', build_var, 'not found!')
+        return
+
+    if filename_rack:
+        build_single(RACK_BUILD_DIR, RACK_BUILD_TARGET_DIR, filename_rack, config_var)
+
+    if filename_rack_mount:
+        build_single(RACK_MOUNT_BUILD_DIR, RACK_MOUNT_BUILD_TARGET_DIR, filename_rack, config_var)
+
+
+def build_single(build_dir, target_dir, filename, config):
+    print('Building:', filename, 'from', build_dir, 'to', target_dir)
+    run_openscad(construct_openscad_args(build_dir, target_dir, filename, config))
+
+
+def construct_openscad_args(build_dir, target_dir, filename, config):
+    source = os.path.join(build_dir, filename)
+    target = os.path.join(target_dir, os.path.splitext(filename)[0] + '.stl')
+
+    return ['-o', target, source]
+
+
+def find_rack(filename):
+    return find_scad_file(RACK_BUILD_DIR, filename)
+
+
+def find_rack_mount(filename):
+    return find_scad_file(RACK_MOUNT_BUILD_DIR, filename)
+
+
+def find_scad_file(directory, filename):
+    for dir_file in os.listdir(directory):
+        dir_file_normalized = dir_file.lower()
+        filename_normalized = filename.lower()
+
+        if dir_file_normalized.endswith("_p.scad"):
+            dir_file_normalized = dir_file_normalized[:-7]
+
+        if filename_normalized.endswith("_p.scad"):
+            filename_normalized = filename_normalized[:-7]
+
+        if dir_file_normalized == filename_normalized \
+                or os.path.splitext(dir_file_normalized)[0] == filename_normalized:
+            return dir_file
+    return None
+
+
+def run_openscad(options=['-h']):
+    command = ['openscad', '-q'] + options
+    try:
+        subprocess.check_output(command, universal_newlines=True, stderr=subprocess.DEVNULL)
+
+    except FileNotFoundError:
+        print('OpenSCAD command not found! '
+              'Please make sure that you have OpenSCAD installed and can run OpenSCAD CLI commands. '
+              '(Currently need Linux for this)')
+
+
+def assert_os():
+    if sys.platform == 'linux':
+        return True
+    else:
+        return False
+
+
+if __name__ == '__main__':
+    main()