def get_document(self, doctag):
# return document entry with doctag
cur = self.db_conn.cursor()
query = """SELECT distinct id, doctag, title, doctext
FROM document
WHERE doctag =%s;"""
# print "QUERY", query
cur.execute(query, (doctag, ))
res = cur.fetchone()
if res is not None:
result = {
'docID': res[1],
'title': res[2],
'docText': res[3],
'abstract': {
'sentences': []
}
}
sentences = self.get_sentences(doctag)
for s in sentences:
sentence = Sentence(s[2], offset=s[3], sid=s[1], did=doctag)
sentence.process_corenlp_output(ast.literal_eval(s[4]))
sentence = self.get_entities(sentence)
result['abstract']['sentences'].append(sentence.get_dic("all"))
output = json.dumps(result)
return output
else:
return json.dumps(
{'error': 'could not find document {}'.format(doctag)})
def get_document(self, doctag):
# return document entry with doctag
cur = self.db_conn.cursor()
query = """SELECT distinct id, doctag, title, doctext
FROM document
WHERE doctag =%s;"""
# print "QUERY", query
cur.execute(query, (doctag,))
res = cur.fetchone()
if res is not None:
result = {'docID': res[1], 'title': res[2], 'docText': res[3], 'abstract':{'sentences':[]}}
sentences = self.get_sentences(doctag)
for s in sentences:
sentence = Sentence(s[2], offset=s[3], sid=s[1], did=doctag)
sentence.process_corenlp_output(ast.literal_eval(s[4]))
sentence = self.get_entities(sentence)
result['abstract']['sentences'].append(sentence.get_dic("all"))
output = json.dumps(result)
return output
else:
return json.dumps({'error': 'could not find document {}'.format(doctag)})
