import sys sys.path.append("scripts") from constructors_visitor import ConstructorsVisitor from cached_exception import CachedException import urllib import urllib2 import json timeout = 100 class ConstructorsObjectVisitor(ConstructorsVisitor): def __init__(self, args): ConstructorsVisitor.__init__(self, args) self.username = args[0] self.obj_file = args[1] self.real_file = args[2] self.address = args[3] self.object_symbols = {} with open(self.real_file, 'r') as f: import hashlib md5 = hashlib.md5() md5.update(f.read()) self.hash_file = md5.hexdigest() # Create user urllib2.urlopen(urllib2.Request(self.address, urllib.urlencode({"op": "set_input", "value": '"%s"' % self.username, "username": "user_manager"}))).read() simple_filename = self.real_file.rsplit("/")[-1] print("[COMPILE] %s" % simple_filename) def dump(self): v = ConstructorsVisitor.dump(self) def flush_data(data): if data: urllib2.urlopen(urllib2.Request(self.address, urllib.urlencode({"op": "set_input", "data": json.dumps(data), "username": self.username})), timeout=timeout).read() return [] # Set up interface flush_data([3, "upload", self.obj_file, self.hash_file, True]) flush_data(v) # Upload symbol table data = [] for e, v in self.object_symbols.iteritems(): data.extend([True, e, v]) # Finish the symbol table data.append(False) urllib2.urlopen(urllib2.Request(self.address, urllib.urlencode({"op": "set_input", "data": json.dumps(data), "username": self.username}))).read() # Wait for kernel to signal that it finished urllib2.urlopen(urllib2.Request(self.address, urllib.urlencode({"op": "set_input", "value": '2', "username": self.username}))).read() v = urllib2.urlopen(urllib2.Request(self.address, urllib.urlencode({"op": "get_output", "username": self.username}))).read() v = json.loads(v) if v == "DONE": return True else: return False def visit_definition(self, tree): for a in tree.get_children("ID"): name = a.get_tail()[0] self.object_symbols[name] = True return ConstructorsVisitor.visit_definition(self, tree) def visit_vardecl(self, tree): if len(tree.get_tail()) > 2: for a in tree.get_children("ID"): name = a.get_tail()[0] self.object_symbols.setdefault(name, False) return ConstructorsVisitor.visit_vardecl(self, tree) else: return ConstructorsVisitor.visit_vardecl(self, tree) def visit_funcdecl(self, tree): for a in tree.get_children("func_name"): for b in a.get_children("ID"): name = b.get_tail()[0] if tree.get_children("func_body") or tree.get_children("ASSIGN"): self.object_symbols[name] = True else: self.object_symbols.setdefault(name, False) return ConstructorsVisitor.visit_funcdecl(self, tree)