render.py

#!/usr/bin/env python

from __future__ import print_function
import os
import sys
import shutil
import subprocess
from time import *

def render(machine, parts = None):
    stl_dir = machine + os.sep + "stls"
    render_dir = machine + os.sep + "render"

    if os.path.isdir(render_dir):
        if not parts:
            shutil.rmtree(render_dir)   #if doing them all clear the directory first
            sleep(0.1)
            os.makedirs(render_dir)
    else:
        os.makedirs(render_dir)

    #
    # List of individual part files
    #
    if parts:
        stls = [i[:-4] for i in parts]
    else:
        stls = [i[:-4] for i in os.listdir(stl_dir) if i[-4:] == ".stl"]
        #
        # Add the multipart files
        #
        for i in  os.listdir(stl_dir + os.sep + "printed"):
            if i[-4:] == ".stl" and not i[:-4] in stls:
                stls.append("printed" + os.sep + i[:-4])

    for i in stls:
        command = 'blender -b  utils' + os.sep + 'render.blend -P utils' + os.sep + 'viz.py -- ' + \
            stl_dir + os.sep + i + '.stl ' + render_dir + os.sep + i + '.png'
        print(command)
        subprocess.check_output(command.split())

if __name__ == '__main__':
    if len(sys.argv) > 1:
        render(sys.argv[1], sys.argv[2:])
    else:
        print("usage: render dibond|mendel|sturdy|huxley|your_machine, [part.stl ...]")
        sys.exit(1)