metaforce/hecl/blender/blendershell.py

import bpy, sys, os

# Extract pipe file descriptors from arguments
print(sys.argv)
if '--' not in sys.argv:
    bpy.ops.wm.quit_blender()
args = sys.argv[sys.argv.index('--')+1:]
readfd = int(args[0])
writefd = int(args[1])

def readpipeline():
    retval = bytearray()
    while True:
        ch = os.read(readfd, 1)
        if ch == b'\n' or ch == b'':
            return retval
        retval += ch

def writepipeline(linebytes):
    os.write(writefd, linebytes + b'\n')

def writepipebuf(linebytes):
    writepipeline(b'BUF')
    os.write(writefd, struct.pack('I', len(linebytes)) + linebytes)

def quitblender():
    writepipeline(b'QUITTING')
    bpy.ops.wm.quit_blender()

# Check that HECL addon is installed/enabled
if 'hecl' not in bpy.context.user_preferences.addons:
    if 'FINISHED' not in bpy.ops.wm.addon_enable(module='hecl'):
        writepipeline(b'NOADDON')
        bpy.ops.wm.quit_blender()

# Make addon available to commands
import hecl

# Intro handshake
writepipeline(b'READY')
ackbytes = readpipeline()
if ackbytes != b'ACK':
    quitblender()

# Command loop
while True:
    cmdline = readpipeline().split(b' ')

    if cmdline[0] == b'QUIT':
        quitblender()

    elif cmdline[0] == b'OPEN':
        if 'FINISHED' in bpy.ops.wm.open_mainfile(filepath=cmdline[1].decode()):
            writepipeline(b'FINISHED')
        else:
            writepipeline(b'CANCELLED')

    elif cmdline[0] == b'CREATE':
        bpy.context.user_preferences.filepaths.save_version = 0
        if 'FINISHED' in bpy.ops.wm.save_as_mainfile(filepath=cmdline[1].decode()):
            writepipeline(b'FINISHED')
        else:
            writepipeline(b'CANCELLED')

    elif cmdline[0] == b'PYBEGIN':
        writepipeline(b'READY')
        globals = dict()
        locals = dict()
        while True:
            try:
                line = readpipeline()
                if line == b'PYEND':
                    writepipeline(b'DONE')
                    break
                co = compile(line+'\n', '<HECL>', 'single')
                exec(co, globals, locals)
            except Exception as e:
                writepipeline(b'EXCEPTION')
                break
            writepipeline(b'OK')

    elif cmdline[0] == b'PYEND':
        writepipeline(b'ERROR')

    else:
        hecl.command(cmdline, writepipeline, writepipebuf)