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)})