glue.py

import re
import copy

from collections import defaultdict
from string import Template

# initialize the dictionary for the methods with checked exceptions such as {fake method: real method}
method_dict_checked = {'deleteRecord' : 'delete', \
		'editText' : 'setText_new', \
		'insertData' : 'insert_new', \
		'setLayout' : 'setContentView_new', \
		'findViewId' : 'findViewById_new', \
		'changeTextColor' : 'setTextColor_new', \
		'getCursorString' : 'getString', \
		'queryData' : 'query_new', \
		'updateRecord' : 'update', \
		'drawTxt' : 'drawText_new'}

# initialize the dictionary for the methods with unchecked exceptions such as {fake method: real method}
method_dict_unchecked = {'deleteRecord' : 'delete', \
		'editText' : 'setText', \
		'insertData' : 'insert', \
		'setLayout' : 'setContentView', \
		'findViewId' : 'findViewById', \
		'changeTextColor' : 'setTextColor', \
		'getCursorString' : 'getString', \
		'queryData' : 'query', \
		'updateRecord' : 'update', \
		'drawTxt' : 'drawText'}

# answer_block is a dict of user's answers,
# i.e. answer_block = {'answer_1' : fake_answer}
# survey type refers to the different surveys 
# (methods with checked exceptions Vs. methods with unchecked exceptions--documented and undocumented)
def glue_answer(filepath, answers, survey_type, email):
	method_dict = set_dict(survey_type)
	# open the file
	filein = open(filepath)
	# read it
	src = Template(filein.read())
	result = src.substitute(answers)
	with open('static/%s-NoteEditor.java' % (email), 'w') as f:
		f.write("%s" % result)
	# dictionary for answers with real Android's API methods
	real_answers = bind_method(answers, method_dict)
	#do the substitution
	result = src.substitute(real_answers)
	return result

# Bind the answers' methods to the real Android's API methods
# answers is a dict, i.e. answers = {'answer_1' : fake_answer}
# This function returns a dict of answers with real Android's
# API methods, i.e. real_answers = {'answer_1' : real_answer}
def bind_method(answers, method_dict):
	real_answers = {}
	a_keys = list(answers.keys())
	m_keys = list(method_dict.keys())
	# for each user answer
	for k, l in enumerate(a_keys):
		# get the value of the answer
		an = answers.get(a_keys[k])
		# for each fake method
		for m, n in enumerate(m_keys):
			# search for fake method in the answer
			fake = m_keys[m]
			if (re.search(fake, an)):
				#print ("find fake :" + fake)
				# get real method
				real = method_dict.get(fake)
				if (a_keys[k] not in list(real_answers.keys())):
					real_answers[a_keys[k]] = re.sub(fake+'\(', real+'(', an)
					break
	# check if finally there exists fake method in user's answer
	for d, f in enumerate(a_keys):
		if (a_keys[d] not in list(real_answers.keys())):
			real_answers[a_keys[d]] = answers.get(a_keys[d])
	return real_answers

def replace_methods(compiler_output, survey_type):
	method_dict = set_dict(survey_type)
	for fake, real in method_dict.items():
		#compiler_output = compiler_output.replace(fake, real)
		compiler_output = re.sub(real, fake, compiler_output)
	if re.search("\bsetTextColor\b\(\bcolors\b\)", compiler_output):
		compiler_output = re.sub("\bsetTextColor\b\(\bcolors\b\)", "changeTextColor(colors)", replace_output)
	# check for line numbers
	#comp_output = remove_line_numbers(compiler_output)
	return compiler_output

# dict depending on the survey type
def set_dict(survey_type):
	if (survey_type == 'unchecked'):
		return method_dict_unchecked
	elif (survey_type == 'checked'):
		return method_dict_checked

# replace line numbers with spaces
def remove_line_numbers(output):
	out = ''
	#.java:118
	print ("Here is the output.")
	print (output)
	#if re.seach('.java:/d+', output):
	#	print ("OKK")
	out = re.sub(':[0-9]+', '', output)
	return out

# vim: tabstop=8 noexpandtab shiftwidth=8 softtabstop=0