Python使用graphviz画流程图过程解析


Posted in Python onMarch 31, 2020

问题描述

项目中需要用到流程图,如果用js的echarts处理,不同层级建动态计算位置比较复杂,考虑用python来实现

测试demo

实现效果如下

Python使用graphviz画流程图过程解析

完整代码

import yaml
import os
import ibm_db
from graphviz import Digraph
from datetime import datetime


# db连接
def db2_query(sql):
  conn = ibm_db.connect(
    "DATABASE=%s;HOSTNAME=%s;PORT=%s;PROTOCOL=TCPIP;UID=%s;PWD=%s" % (
      config['db2_databse'],
      config['db2_host'],
      config['db2_port'],
      config['db2_user'],
      config['db2_password']
    ), "", ""
  )

  if conn:
    # 执行
    stmt = ibm_db.exec_immediate(conn, sql)
    result = ibm_db.fetch_both(stmt)

  return result, stmt, conn


# 添加所有的点
def add_nodes():
  sql = "SELECT * FROM AML.DEP_OFTEN_TRAD WHERE ACCOUNT_NO=" + ACCOUNT_NO + " ORDER BY NUM DESC fetch first 10 rows ONLY;"
  result, stmt, conn = db2_query(sql)
  nodes = []
  nodes_dict = {}
  nodes_dict[ACCOUNT_NO] = 'A'

  count = 0
  while (result):
    count = count + 1
    print(chr(ord('A') + count) + "," + result[3])
    dot.node(chr(ord('A') + count), result[3], shape="rectangle", color="orange")

    nodes.append(result[2])
    nodes_dict[result[2]] = chr(ord('A') + count)
    print('-----------------')
    result = ibm_db.fetch_both(stmt)

  # 关闭数据库连接
  ibm_db.close(conn)
  nodes.append(ACCOUNT_NO)

  return dot, nodes, nodes_dict


# 添加所有的边
def add_edges(nodes):
  nodes_back = nodes
  print(nodes_back)

  for p in range(len(nodes_back)):
    nodes_temp = nodes.copy()
    from_nodes = nodes_temp.pop(p)
    print("len of nodes_temp:" + str(len(nodes_temp)) + ",from_nodes:" + from_nodes)

    nodes_temp_in = ",".join(nodes_temp)
    nodes_temp_in = '(' + nodes_temp_in + ')'

    sql = "SELECT * FROM AML.DEP_OFTEN_TRAD WHERE ACCOUNT_NO=" + from_nodes + " and CNTPRT_ACCOUNT_NO IN" + \
       nodes_temp_in
    result, stmt, conn = db2_query(sql)
    if (result):
      while (result):
        print('-----------------')
        print(result)
        print(result['CNTPRT_ACCOUNT_NO'] + nodes_dict[result['CNTPRT_ACCOUNT_NO']])
        print("add edges," + result['TRAD_VAL'] + " from:" + from_nodes + ",to:" + result['CNTPRT_ACCOUNT_NO'])
        dot.edge(nodes_dict[from_nodes], nodes_dict[result['CNTPRT_ACCOUNT_NO']], result['TRAD_VAL'])
        result = ibm_db.fetch_both(stmt)

      # 关闭数据库连接
      ibm_db.close(conn)
  return dot


if __name__ == "__main__":
  yaml_path = os.path.join('../', 'config.yaml')
  with open(yaml_path, 'r') as f:
    config = yaml.load(f)

  dot = Digraph(
    engine="circo",
    comment='The Test Table',
    format="png"
  )

  ACCOUNT_NO = '10100002181'
  # 添加圆点A,A的标签是Dot A
  dot.node('A', '中心客户', shape="rectangle", color="blue")

  nodes_begin = datetime.now()
  print(str(nodes_begin) + " nodes_begin")

  dot, nodes, nodes_dict = add_nodes()

  print(nodes_dict)
  edges_begin = datetime.now()
  print(str(edges_begin) + " edges_begin")

  dot = add_edges(nodes)

  print(dot.source)
  render_begin = datetime.now()
  print(str(render_begin) + " render_begin")
  dot.render('./file/db2-table.gv', view=True)

测试流程图

用不同的参数,生成的2个图如下Python使用graphviz画流程图过程解析

linux环境搭建

线上是不连外网的私有服务器,开始在离线环境安装,需要依赖的包太多,后来就在虚拟机里redhat6.6在线装好了。####redhat安装rpm依赖

#清除原有RHEL的YUM及相关软件包。
rpm -qa | grep yum | xargs rpm -e --nodeps
rpm -qa |grep python-urlgrabber|xargs rpm -e --nodeps

mv CentOS6-Base-163.repo /etc/yum.repos.d/

#安装rpm依赖包

rpm -ivh python-iniparse-0.3.1-2.1.el6.noarch.rpm
rpm -ivh python-urlgrabber-3.9.1-11.el6.noarch.rpm
rpm -ivh yum-metadata-parser-1.1.2-16.el6.x86_64.rpm
rpm -ivh yum-plugin-fastestmirror-1.1.30-41.el6.noarch.rpm yum-3.2.29-81.el6.centos.noarch.rpm

替换yum163源

cd /etc/yum.repos.d/mv rhel-source.repo rhel-source.repo.bak

清理yum缓存

yum clean allyum makecache#测试yum repolist

安装graphviz

yum install 'graphviz*'#测试which dot #/usr/bin/dotdot -V#dot - graphviz version 2.26.0 (20091210.2329)

查看yum的安装日志,发现需要依赖2,30个包

解决依赖关系
--> 执行事务检查
---> Package graphviz.x86_64 0:2.26.0-10.el6 will be 安装
--> 处理依赖关系 libgd.so.2()(64bit),它被软件包 graphviz-2.26.0-10.el6.x86_64 需要
--> 处理依赖关系 libXpm.so.4()(64bit),它被软件包 graphviz-2.26.0-10.el6.x86_64 需要
--> 处理依赖关系 libXaw.so.7()(64bit),它被软件包 graphviz-2.26.0-10.el6.x86_64 需要
---> Package graphviz-devel.x86_64 0:2.26.0-10.el6 will be 安装
---> Package graphviz-doc.x86_64 0:2.26.0-10.el6 will be 安装
---> Package graphviz-gd.x86_64 0:2.26.0-10.el6 will be 安装
---> Package graphviz-graphs.x86_64 0:2.26.0-10.el6 will be 安装
---> Package graphviz-guile.x86_64 0:2.26.0-10.el6 will be 安装
--> 处理依赖关系 guile,它被软件包 graphviz-guile-2.26.0-10.el6.x86_64 需要
---> Package graphviz-java.x86_64 0:2.26.0-10.el6 will be 安装
---> Package graphviz-lua.x86_64 0:2.26.0-10.el6 will be 安装
---> Package graphviz-perl.x86_64 0:2.26.0-10.el6 will be 安装
---> Package graphviz-php.x86_64 0:2.26.0-10.el6 will be 安装
--> 处理依赖关系 php(zend-abi) = 20090626,它被软件包 graphviz-php-2.26.0-10.el6.x86_64 需要
--> 处理依赖关系 php(api) = 20090626,它被软件包 graphviz-php-2.26.0-10.el6.x86_64 需要
---> Package graphviz-python.x86_64 0:2.26.0-10.el6 will be 安装
---> Package graphviz-ruby.x86_64 0:2.26.0-10.el6 will be 安装
--> 处理依赖关系 ruby,它被软件包 graphviz-ruby-2.26.0-10.el6.x86_64 需要
--> 处理依赖关系 libruby.so.1.8()(64bit),它被软件包 graphviz-ruby-2.26.0-10.el6.x86_64 需要
---> Package graphviz-tcl.x86_64 0:2.26.0-10.el6 will be 安装
--> 处理依赖关系 tcl >= 8.3,它被软件包 graphviz-tcl-2.26.0-10.el6.x86_64 需要
--> 处理依赖关系 tk,它被软件包 graphviz-tcl-2.26.0-10.el6.x86_64 需要
--> 处理依赖关系 libtk8.5.so()(64bit),它被软件包 graphviz-tcl-2.26.0-10.el6.x86_64 需要
--> 执行事务检查
---> Package gd.x86_64 0:2.0.35-11.el6 will be 安装
---> Package guile.x86_64 5:1.8.7-5.el6 will be 安装
---> Package libXaw.x86_64 0:1.0.11-2.el6 will be 安装
---> Package libXpm.x86_64 0:3.5.10-2.el6 will be 安装
---> Package php-common.x86_64 0:5.3.3-49.el6 will be 安装
---> Package ruby.x86_64 0:1.8.7.374-5.el6 will be 安装
---> Package ruby-libs.x86_64 0:1.8.7.374-5.el6 will be 安装
--> 处理依赖关系 libreadline.so.5()(64bit),它被软件包 ruby-libs-1.8.7.374-5.el6.x86_64 需要
---> Package tcl.x86_64 1:8.5.7-6.el6 will be 安装
---> Package tk.x86_64 1:8.5.7-5.el6 will be 安装
--> 执行事务检查
---> Package compat-readline5.x86_64 0:5.2-17.1.el6 will be 安装
--> 完成依赖关系计算

依赖关系解决

yum安装openssl-devel

#查看openssl安装包rpm -aq|grep opensslyum install openssl-devel -y

查看yum依赖

正在升级 : libcom_err-1.41.12-24.el6.x86_64 1/26
正在升级 : libselinux-2.0.94-7.el6.x86_64 2/26
正在升级 : krb5-libs-1.10.3-65.el6.x86_64 3/26
****************
Verifying : e2fsprogs-libs-1.41.12-21.el6.x86_64 23/26
Verifying : libselinux-2.0.94-5.8.el6.x86_64 24/26
Verifying : krb5-libs-1.10.3-33.el6.x86_64 25/26
Verifying : libss-1.41.12-21.el6.x86_64 26/26

python安装pip依赖

pip install -r requirement.txt

#Installing collected packages: PyYAML, ibm-db, tornado, graphviz
# Running setup.py install for PyYAML ... done
# Running setup.py install for ibm-db ... done
# Running setup.py install for tornado ... done
#Successfully installed PyYAML-5.1.1 graphviz-0.11.1 ibm-db-3.0.1 tornado-6.0.3 

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
从零学python系列之新版本导入httplib模块报ImportError解决方案
May 23 Python
使用Python的Bottle框架写一个简单的服务接口的示例
Aug 25 Python
Python实现周期性抓取网页内容的方法
Nov 04 Python
bat和python批量重命名文件的实现代码
May 19 Python
Python 序列的方法总结
Oct 18 Python
Python基于numpy灵活定义神经网络结构的方法
Aug 19 Python
pygame游戏之旅 添加碰撞效果的方法
Nov 20 Python
python 运用Django 开发后台接口的实例
Dec 11 Python
强悍的Python读取大文件的解决方案
Feb 16 Python
python 实现字符串下标的输出功能
Feb 13 Python
Python统计学一数据的概括性度量详解
Mar 03 Python
Flask response响应的具体使用
Jul 15 Python
Django模板之基本的 for 循环 和 List内容的显示方式
Mar 31 #Python
基于python实现计算且附带进度条代码实例
Mar 31 #Python
Django values()和value_list()的使用
Mar 31 #Python
利用python实现凯撒密码加解密功能
Mar 31 #Python
python手机号前7位归属地爬虫代码实例
Mar 31 #Python
django修改models重建数据库的操作
Mar 31 #Python
Python写捕鱼达人的游戏实现
Mar 31 #Python
You might like
PHP中__get()和__set()的用法实例详解
2013/06/04 PHP
常用的php图片处理类(水印、等比缩放、固定高宽)分享
2015/06/19 PHP
PHP生成随机密码方法汇总
2015/08/27 PHP
php实现学生管理系统
2020/03/21 PHP
PHP身份证校验码计算方法
2016/08/10 PHP
php中类和对象:静态属性、静态方法
2017/04/09 PHP
jQuery学习3:操作元素属性和特性
2010/02/07 Javascript
idTabs基于JQuery的根据URL参数选择Tab插件
2012/04/11 Javascript
js之事件冒泡和事件捕获详细介绍
2013/10/28 Javascript
基于jquery实现无限级树形菜单
2016/03/22 Javascript
jquery实现多次上传同一张图片
2017/01/09 Javascript
jQuery插件zTree实现单独选中根节点中第一个节点示例
2017/03/08 Javascript
jquery 手势密码插件
2017/03/17 Javascript
js实现QQ面板拖拽效果(慕课网DOM事件探秘)(全)
2017/09/19 Javascript
vue2.0s中eventBus实现兄弟组件通信的示例代码
2017/10/25 Javascript
JS实现关键词高亮显示正则匹配
2018/06/22 Javascript
React事件处理的机制及原理
2018/12/03 Javascript
微信小程序自定义可滑动日历界面
2018/12/28 Javascript
vue不操作dom实现图片轮播的示例代码
2019/12/18 Javascript
JS+css3实现幻灯片轮播图
2020/08/14 Javascript
使用python统计文件行数示例分享
2014/02/21 Python
对python 自定义协议的方法详解
2019/02/13 Python
11个Python3字典内置方法大全与示例汇总
2019/05/13 Python
六行python代码的爱心曲线详解
2019/05/17 Python
python NumPy ndarray二维数组 按照行列求平均实例
2019/11/26 Python
CSS3结构性伪类选择器九种写法
2012/04/18 HTML / CSS
女子锻炼服装和瑜伽服装:Splits59
2019/03/04 全球购物
几道Java和数据库的面试题
2013/05/30 面试题
校本教研工作制度
2014/01/22 职场文书
应用英语专业自荐信
2014/01/26 职场文书
大学班级学风建设方案
2014/05/01 职场文书
我的教育故事演讲稿
2014/05/04 职场文书
经典禁毒标语
2014/06/16 职场文书
万能检讨书2000字
2014/10/17 职场文书
房屋所有权证明
2015/06/19 职场文书
小学感恩主题班会
2015/08/12 职场文书