BIDI_ML_INFO_EXTRACTION/BiddingKG/dl/role/Entity2DB.py

'''
Created on 2019年3月7日

@author: User
'''


def persistArticle(conn,articles,codeName):
    
    cursor = conn.cursor()
    for article,item in zip(articles,codeName):
        code = ";".join(item[1]["code"])
        name = item[1]["name"]
        sql = " insert into articles_processed(id,content,code,name) values('"+article.id+"','"+article.content+"','"+code+"','"+name+"')"
        cursor.execute(sql)
        
def persistSentence(conn,sentences):
    
    cursor = conn.cursor()
    for sentence in sentences:
        sql = " insert into sentences(doc_id,sentence_index,sentence_text,tokens)"\
                +"values('"+sentence.doc_id+"',"+str(sentence.sentence_index)+",'"+sentence.sentence_text+"',array"+str(sentence.tokens)\
                +")"
        cursor.execute(sql)
        
def persistEntity(conn,entitys):
    
    cursor = conn.cursor()
    for entity in entitys:
        sql = " insert into entity_mention(entity_id,entity_text,entity_type,doc_id,sentence_index,begin_index,end_index,label,values)"\
                +" values('"+str(entity.entity_id)+"','"+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))