def gen_tmpfile(kb_size, query_size, query_class): from sh import cp from experiments_pl import convertToQuery import re queryfile = "./{0}-{1}-queries.txt".format(query_class, query_size) tmpfile = make_tmpfile_name(kb_size, query_size, query_class) cp( "{0}.pl".format(kb_size), tmpfile ) r = [] with open(tmpfile, "a") as tmp: with open(queryfile) as queries: for i, query in enumerate(queries.readlines()): rule = convertToQuery(query.strip()) args = ",".join([chr(65+n) for n,_ in enumerate(re.finditer('tweet',rule))]) pred_sig = make_pred_sig(args, i) tmp.write("{0} :- {1}.\n".format(pred_sig, rule)) r.append({ 'args': args, 'i': i, 'kb_size': kb_size, 'query_size': query_size, 'query_class': query_class, 'orig_query': query.strip() }) return r
from experiments_pl import convertToQuery from sys import stdin for query in stdin: print convertToQuery(query)