python使用xlsxwriter实现有向无环图到Excel的转换

 (py3) [root@7-o-1 py-dag]# cat test.py from dag import DAG dag = DAG() dag.from_dict({'a': ['b', 'c','e'], 'b': ['d','g'], 'c': ['d'], 'g':['i'], 'i':[], 'e':['gh','ox','wer'], 'gh':[], 'ox':[], 'wer':[], 'd': []}) def generate_excel(dag, row_start,col,excel_file):#参数分别为有向无环图图、开始写入的行,开始写入列，以及文件名 import xlsxwriter#导入Excel操作模块 workbook = xlsxwriter.Workbook(excel_file)#创建Excel文件 worksheet = workbook.add_worksheet('testsheet')#创建工作簿 #以下为部分格式的预定义（包含合并行格式，数字日期格式等） merge_format = workbook.add_format({ 'bold': 1, 'border': 1, 'align': 'left', 'valign': 'vcenter'}) cell_format = workbook.add_format({ 'bold': 1, 'border': 1, 'align': 'top', 'valign': 'left'}) date_format = workbook.add_format({ 'bold': 1, 'border': 1, 'align': 'top', 'valign': 'left', 'num_format': 'd mmmm yyyy'}) num_format = workbook.add_format({ 'bold': 1, 'border': 1, 'align': 'top', 'valign': 'left', 'num_format': '0.0000%'}) def get_node_leaves(node,ret = []):#用来返回一个节点的所有最终叶子结点 downs = dag.downstream(node)#先获取当前节点的下一级节点 if len(downs) 0: row_next = r_idx fo<b style="color:transparent">来源gao@!dai!ma.com搞$$代^@码!网</b>r downstream in downstreams: subnode_leaves = len(get_node_leaves(downstream,[]))#获取当前节点的所有叶子用来为下一个节点的填充确定位移 if downstream in all_leaves: subnode_leaves = 1 if col == merge_to:#对于子节点为叶子并且跨度为1应用write worksheet.write(col+str(row_next), downstream,cell_format) else:#对于子节点为叶子结点但是跨度大于1应用merge_range merge_rg = col+str(row_next) +':'+merge_to+str(row_next) worksheet.merge_range(merge_rg, downstream,merge_format) else:#子节点非叶子结点进行递归遍历 dag_visit(dag, downstream, row_next,col) #每访问一个子结点需要将row值增加方便下一个node正确填充，row的值是前一个结点的子节点个数 row_next += subnode_leaves else: print(node) head_node = dag.ind_nodes()[0] all_leaves = dag.all_leaves() dag_depth = dag.dag_depth(head_node,0) merge_to= chr(ord(col)+dag_depth)#对于合并列的终结列index dag_visit(dag, head_node, row_start,col)#传递根节点进行遍历 workbook.close() generate_excel(dag,1,'A','test.xlsx')