Function find_tree_root

database_utils.py:221–247

(working_dir,entity)

            print(e)
            print("insert communities error")
def find_tree_root(working_dir,entity):
    db = pymysql.connect(host='localhost',port=4321, user='root',
                      passwd='123',  charset='utf8mb4')
    dbname=os.path.basename(working_dir)
    res=[entity]
    cursor = db.cursor()
    db_name=os.path.basename(working_dir)
    depth_sql=f"select max(level) from {db_name}.entities"
    cursor.execute(depth_sql)
    depth=cursor.fetchall()[0][0]
    i=0
    
    while i< depth:
        sql=f"select parent from {db_name}.entities where entity_name=%s "
        
        cursor.execute(sql,(entity))
        ret=cursor.fetchall()
        # print(ret)
        i+=1
        if len(ret)==0:
            break
        entity=ret[0][0]
        res.append(entity)
    # res=list(set(res))
    # res = list(dict.fromkeys(res))

    return res

def find_path(entity1,entity2,working_dir,level,depth=5):
    db = pymysql.connect(host='localhost',port=4321, user='root',