modelverse/interface/HUTN/hutn_compiler/model_object_visitor.py

from model_visitor import ModelVisitor
from compiler import main as do_compile
import os
import urllib2
import urllib
import json

def jsonstr(s):
    return '"%s"' % s

def empty(s):
    return None

class ModelObjectVisitor(ModelVisitor):
    def __init__(self, args):
        ModelVisitor.__init__(self, args)
        self.username = args[0]
        self.obj_file = args[1]
        self.real_file = args[2]
        self.address = args[3]

    def dump(self):
        v = ModelVisitor.dump(self)
        data = []
        data.append(["V", '4'])
        for i in v:
            data.append(["V", i])
        data.append(["V", '"exit"'])

        # Wait for kernel to signal that it finished
        data.append(["V", '2'])

        urllib2.urlopen(urllib2.Request(self.address, urllib.urlencode({"op": "set_input", "data": json.dumps(data), "username": self.username}))).read()
        v = urllib2.urlopen(urllib2.Request(self.address, urllib.urlencode({"op": "get_output", "username": self.username}))).read()
        v = v.split("=", 2)[2]
        if v == "DONE":
            return True
        else:
            return False