PloneWorkflowDefinitionToGraphviz/wtg.py at main · mbarde/PloneWorkflowDefinitionToGraphviz · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
from copy import deepcopy
from graphviz import Digraph
from xml.etree.ElementTree import parse as parseXML

import sys


if len(sys.argv) < 2:
    sys.exit(0)

xmlfile = sys.argv[1]

hideGuards = '--hide-guards' in sys.argv
useIdsAsLabels = '--use-ids' in sys.argv
withPermissions = '--with-permissions' in sys.argv

outputFormat = 'pdf'
for arg in sys.argv[2:]:
    if arg.startswith('--format='):
        val = arg.split('=', 1)[1]
        if val in ['pdf', 'jpg', 'png']:
            outputFormat = val

root = parseXML(xmlfile).getroot()

title = root.get('title')

dot = Digraph(comment=title)

transitions = dict()
for trans in root.findall('transition'):
    newTrans = dict()
    newTrans['id'] = trans.get('transition_id')
    newTrans['to'] = trans.get('new_state')
    if useIdsAsLabels is True:
        newTrans['label'] = trans.get('transition_id')
    else:
        newTrans['label'] = trans.get('title')

    if not hideGuards:
        guard_roles = []
        guard_permissions = []

        for guard in trans.findall('guard'):
            for guard_role in guard.findall('guard-role'):
                guard_roles.append(guard_role.text)

            for guard_permission in guard.findall('guard-permission'):
                guard_permissions.append(guard_permission.text)

        if len(guard_roles) > 0:
            newTrans['label'] += '\n(' + ',\n'.join(guard_roles) + ')'

        if len(guard_permissions) > 0:
            newTrans['label'] += '\n(' + ',\n'.join(guard_permissions) + ')'

    transitions[newTrans['id']] = newTrans

edges = []

for state in root.findall('state'):
    node_id = state.get('state_id')
    if useIdsAsLabels is True:
        label = state.get('state_id')
    else:
        label = state.get('title')

    if withPermissions:
        for trans_map in state.findall('permission-map'):
            permissionName = trans_map.get('name')
            roles = []
            for role in trans_map.findall('permission-role'):
                roles.append(role.text)
            label += '\n\n[' + permissionName + ']\n' + ', ' \
                .join(roles) + ''

    dot.node(node_id, label, shape='box')

    for exTrans in state.findall('exit-transition'):
        newEdge = deepcopy(transitions[exTrans.get('transition_id')])
        newEdge['from'] = node_id
        edges.append(newEdge)

for edge in edges:
    dot.edge(edge['from'], edge['to'], label=edge['label'])

dot.render(xmlfile + '.gv', view=True, format=outputFormat)