BIDI_ML_INFO_EXTRACTION/BiddingKG/dl_dev/labelServer/Entity2DB.py

'''
Created on 2019年3月7日

@author: User
'''
import re


def persistArticle(conn,articles,table):
    
    cursor = conn.cursor()
    for article in articles:
        sql = " insert into "+table+"(id,content) values('"+article.id+"',E'"+re.sub("'","\'",article.content)+"')"
        cursor.execute(sql)
        
def persistSentence(conn,sentences,table):
    
    cursor = conn.cursor()
    for sentence in sentences:
        sql = " insert into "+table+"(doc_id,sentence_index,sentence_text,tokens)"\
                +"values('"+sentence.doc_id+"',"+str(sentence.sentence_index)+",E'"+re.sub("'","\'",sentence.sentence_text)+"',array"+str(sentence.tokens)\
                +")"
        cursor.execute(sql)
        
def persistEntity(conn,entitys,table):
    
    cursor = conn.cursor()
    for entity in entitys:
        sql = " insert into "+table+"(entity_id,entity_text,entity_type,doc_id,sentence_index,begin_index,end_index,label,values)"\
                +" values('"+str(entity.entity_id)+"',E'"+re.sub("'","\'",entity.entity_text)+"','"+entity.entity_type+"','"+entity.doc_id+"',"+str(entity.sentence_index)+","+str(entity.begin_index)+","\
                + str(entity.end_index)+",'"+str(entity.label)+"','"+str(entity.values)+"')"
        cursor.execute(sql)
        
if __name__=="__main__":
    print(str(True))