将python字典输出转换为流程图

Question

我有一个文件（请参阅下面代码中的状态变量），我想将其转换为流程图（附件）。我的 python 脚本将“状态”转换为字典。如何将该字典转换为流程图或图形？ 我的代码：

status = """
Object car {
Name honda;
From Richland;
To Seattle;
Distance 160;
Configuration road_travel;
}

Object bus {
Name greyhound;
From pasco;
To richland;
Distance 15;
Configuration road_travel;
}

Object aeroplane {
Name united;
From miami_airport;
To pasco;
Distance 1000;
Configuration air_travel;
}

Object train {
Name gas_train;
From beach;
To miami_airport;
Distance 30;
Configuration train_travel;
}
"""
sale_number = ''

sales = collections.defaultdict(list)

for line in status.split('\n'):
    line = line.strip()
    if line.startswith("set"):
         continue
    elif (line.startswith("Object") or line.startswith("object")):
         sale_number = line.split(' ')[1].strip()
    elif not line or line.isspace() :
         continue
    else:
         # you can also use a regular expression here
         sales[sale_number].append(line.split())

for sale in sales:
    print sale+str(dict(sales[sale][:-1]))

这会产生：

car{'To': 'Seattle;', 'Configuration': 'road_travel;', 'From': 'Richland;', 'Name': 'honda;', 'Distance': '160;'}
train{'To': 'miami_airport;', 'Configuration': 'train_travel;', 'From': 'beach;', 'Name': 'gas_train;', 'Distance': '30;'}
aeroplane{'To': 'pasco;', 'Configuration': 'air_travel;', 'From': 'miami_airport;', 'Name': 'united;', 'Distance': '1000;'}
bus{'To': 'richland;', 'Configuration': 'road_travel;', 'From': 'pasco;', 'Name': 'greyhound;', 'Distance': '15;'}

我想将上面的 python 输出转换成如下图所示的图片。我不想使用 Giffy 或 MS-Visio 手动执行此操作，因为我的实际案例有大约 1000 个对象（此示例在“状态”中有 4 个对象）

Answer 1

只专注于将格式怪异的status 字符串转换为dict 已经够难了，难道你不能将它转换成像 JSON 这样更合理、更流行的格式吗？

import re
def Status2dict(status):
    result = {}
    current = {}
    lines = status.splitlines()
    for line in lines:
        line = line.strip()
        if not line: 
            continue
        mo = re.match(r'Object (\w+) {', line)
        if mo:
            curk = mo.group(1)
            current = {curk: {}}
        elif re.match('}', line):
            result.update(current)
            current = {}
        else:
            mo = re.match(r'(\w+)\s+([\w\s]+);', line)
            if not mo:
                raise ValueError('cannot match {!r}'.format(line))
            current[curk][mo.group(1)] = mo.group(2)
    if current:
        result.update(current)
    return result

import pprint    
pprint.pprint(Status2dict(status))

此代码试图在推断语法的小变化上稍微健壮，您可能希望根据情况将其调高或。不过，总比没有好。

Answer 2

如果您已经将代码转换为点文件，则可以使用 graphviz 的 tk 导出在 tkinter 画布上可视化您的图形。您可以查看this other question 了解其他库。

dot -Ttk这样输出tcl代码

# a
$c create oval 5.33 53.33 77.33 5.33 -fill white -width 1 -outline black -tags {1node1}
$c create text 41.33 30.3 -text {a} -fill black -font {"Times" 14} -tags {0node1}
# c
$c create oval 53.33 149.33 125.33 101.33 -fill white -width 1 -outline black -tags {1node2}
$c create text 89.33 126.3 -text {c} -fill black -font {"Times" 14} -tags {0node2}
#(...)

您可以通过调用 tcl 解释器将其显示在 tkinter 画布上，如下例所示：

from Tkinter import *
import subprocess

graph = "digraph g { a-> c ; b -> c ; c -> d }"

ps = subprocess.Popen(('dot', '-Ttk'),stdin=subprocess.PIPE, stdout=subprocess.PIPE)
out, err = ps.communicate(input=graph)

canvas = Canvas()
canvas.pack(expand=YES, fill=BOTH)

#the output of graphiz assume thazt the canvas in variable $c
out = "set c .{}\n".format(canvas._name) + out
#tk.eval expect command one by one
map(canvas.tk.eval, out.split("\n"))

canvas.mainloop()