#!/usr/bin/python
# -*- coding: UTF-8 -*-
import matplotlib
import matplotlib.pyplot as plt
from xml import sax
import networkx as nx
import dfs_to_routes
#import pymysql

Edges = []
Value = {}
Labels = {}


class UmlHandler(sax.ContentHandler):
    # XML Content Handler 重写类
    def __init__(self):
        self.CurrentData = ""

    # 元素开始事件处理
    def startElement(self, tag, attributes):
        self.CurrentData = tag
        if tag == "mxCell":
            if 'value' in attributes:
                if attributes['value'] != '':
                    if "开始" == attributes['value']:
                        g.add_node('开始')
                        Value[attributes['id']] = '开始'
                    if "结束" == attributes['value']:
                        g.add_node('结束')
                        Value[attributes['id']] = '结束'
                if "vertex" in attributes:
                    if "style" in attributes:
                        if "swimlane" in attributes['style']:
                            pass
                        else:
                            if attributes['value'] != '':
                                if 'edgeLabel' not in attributes['style']:
                                    g.add_node(attributes['value'])
                                    Value[attributes['id']] = attributes['value']
                                else:
                                    Labels[attributes['parent']] = attributes['value']
            if "edge" in attributes:
                if "target" not in attributes or "source" not in attributes:
                    print(attributes['value'])
                else:
                    source = attributes['source']
                    target = attributes['target']
                    if 'value' in attributes:
                        if attributes['value'] != '':
                            Labels[attributes['id']] = attributes['value']
                    if source != target:
                        Edges.append((source, target, {'id': attributes['id']}))


def ProcessXML(fp):
    # 创建一个 XMLReader
    parser = sax.make_parser()
    parser.setFeature(sax.handler.feature_namespaces, 0)
    # 重写 ContextHandler
    Handler = UmlHandler()
    parser.setContentHandler(Handler)
    parser.parse(fp)


def ProcessEdges(g):
    # 处理边
    for i in Edges:
        g.add_edge(Value[i[0]], Value[i[1]], id=i[2]['id'])
    # 处理边的标签
    for i in Labels.keys():
        for edge in g.edges(data=True):
            if edge[2]['id'] == i:
                edge[2]['value'] = Labels[i]


def DrawGraph(g):
    nx.draw_circular(g, with_labels=True)

    # 圆形画出该有向图
    plt.show()


def ProcessDatabase(routes_lst):
    # 指定数据库
    con = pymysql.connect(host='localhost', user='root', passwd='root', port=3306, db='routes')
    cursor = con.cursor()
    # Insert 数据
    for route in routes_lst:
        sql = "INSERT INTO routes_input (Route) VALUE" + "(\'" + route + "\')"
        cursor.execute(sql)
    con.commit()


def CheckOutDegrees(g):
    for i in g.out_degree:
        if i[1] == 0:
            if i[0] != "结束":
                raise Exception("图存在格式问题，请检查，结点“"+i[0]+"”出度为0。")


if __name__ == "__main__":
    # 防止matplot的中文乱码 Mac 平台下
    matplotlib.rcParams['font.sans-serif'] = ['Arial Unicode MS']
    # Windows 平台下
    # matplotlib.rcParams['font.sans-serif'] = ['Microsoft Yahei']

    # 创建新的有向图
    g = nx.DiGraph()
    g.clear()

    # !!!!!!!!!!
    # 在此处更换 XML 文件路径
    # !!!!!!!!!!
    filepath = "bpmn.xml"
    # 处理 XML 数据
    ProcessXML(filepath)

    # 处理边的标签
    ProcessEdges(g)

    # 检查出度
    CheckOutDegrees(g)

    # 画图（虽然不能看几乎）
    DrawGraph(g)

    # DFS 搜索路径
    routes = dfs_to_routes.dfs(g, "开始", g.adj, g.nodes.__sizeof__())

    # 插入数据库
    # ProcessDatabase(routes)