summary refs log tree commit diff
path: root/graph
diff options
context:
space:
mode:
Diffstat (limited to 'graph')
-rw-r--r--graph/graph.py151
-rw-r--r--graph/graph2.py156
2 files changed, 0 insertions, 307 deletions
diff --git a/graph/graph.py b/graph/graph.py
deleted file mode 100644
index b2acadcf5e..0000000000
--- a/graph/graph.py
+++ /dev/null
@@ -1,151 +0,0 @@
-# Copyright 2014 OpenMarket Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-
-import sqlite3
-import pydot
-import cgi
-import json
-import datetime
-import argparse
-import urllib2
-
-
-def make_name(pdu_id, origin):
-    return "%s@%s" % (pdu_id, origin)
-
-
-def make_graph(pdus, room, filename_prefix):
-    pdu_map = {}
-    node_map = {}
-
-    origins = set()
-    colors = set(("red", "green", "blue", "yellow", "purple"))
-
-    for pdu in pdus:
-        origins.add(pdu.get("origin"))
-
-    color_map = {color: color for color in colors if color in origins}
-    colors -= set(color_map.values())
-
-    color_map[None] = "black"
-
-    for o in origins:
-        if o in color_map:
-            continue
-        try:
-            c = colors.pop()
-            color_map[o] = c
-        except:
-            print "Run out of colours!"
-            color_map[o] = "black"
-
-    graph = pydot.Dot(graph_name="Test")
-
-    for pdu in pdus:
-        name = make_name(pdu.get("pdu_id"), pdu.get("origin"))
-        pdu_map[name] = pdu
-
-        t = datetime.datetime.fromtimestamp(
-            float(pdu["ts"]) / 1000
-        ).strftime('%Y-%m-%d %H:%M:%S,%f')
-
-        label = (
-            "<"
-            "<b>%(name)s </b><br/>"
-            "Type: <b>%(type)s </b><br/>"
-            "State key: <b>%(state_key)s </b><br/>"
-            "Content: <b>%(content)s </b><br/>"
-            "Time: <b>%(time)s </b><br/>"
-            "Depth: <b>%(depth)s </b><br/>"
-            ">"
-        ) % {
-            "name": name,
-            "type": pdu.get("pdu_type"),
-            "state_key": pdu.get("state_key"),
-            "content": cgi.escape(json.dumps(pdu.get("content")), quote=True),
-            "time": t,
-            "depth": pdu.get("depth"),
-        }
-
-        node = pydot.Node(
-            name=name,
-            label=label,
-            color=color_map[pdu.get("origin")]
-        )
-        node_map[name] = node
-        graph.add_node(node)
-
-    for pdu in pdus:
-        start_name = make_name(pdu.get("pdu_id"), pdu.get("origin"))
-        for i, o in pdu.get("prev_pdus", []):
-            end_name = make_name(i, o)
-
-            if end_name not in node_map:
-                print "%s not in nodes" % end_name
-                continue
-
-            edge = pydot.Edge(node_map[start_name], node_map[end_name])
-            graph.add_edge(edge)
-
-        # Add prev_state edges, if they exist
-        if pdu.get("prev_state_id") and pdu.get("prev_state_origin"):
-            prev_state_name = make_name(
-                pdu.get("prev_state_id"), pdu.get("prev_state_origin")
-            )
-
-            if prev_state_name in node_map:
-                state_edge = pydot.Edge(
-                    node_map[start_name], node_map[prev_state_name],
-                    style='dotted'
-                )
-                graph.add_edge(state_edge)
-
-    graph.write('%s.dot' % filename_prefix, format='raw', prog='dot')
-#    graph.write_png("%s.png" % filename_prefix, prog='dot')
-    graph.write_svg("%s.svg" % filename_prefix, prog='dot')
-
-
-def get_pdus(host, room):
-    transaction = json.loads(
-        urllib2.urlopen(
-            "http://%s/_matrix/federation/v1/context/%s/" % (host, room)
-        ).read()
-    )
-
-    return transaction["pdus"]
-
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(
-        description="Generate a PDU graph for a given room by talking "
-                    "to the given homeserver to get the list of PDUs. \n"
-                    "Requires pydot."
-    )
-    parser.add_argument(
-        "-p", "--prefix", dest="prefix",
-        help="String to prefix output files with"
-    )
-    parser.add_argument('host')
-    parser.add_argument('room')
-
-    args = parser.parse_args()
-
-    host = args.host
-    room = args.room
-    prefix = args.prefix if args.prefix else "%s_graph" % (room)
-
-    pdus = get_pdus(host, room)
-
-    make_graph(pdus, room, prefix)
diff --git a/graph/graph2.py b/graph/graph2.py
deleted file mode 100644
index 6b551d42e5..0000000000
--- a/graph/graph2.py
+++ /dev/null
@@ -1,156 +0,0 @@
-# Copyright 2014 OpenMarket Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-
-import sqlite3
-import pydot
-import cgi
-import json
-import datetime
-import argparse
-
-from synapse.events import FrozenEvent
-
-
-def make_graph(db_name, room_id, file_prefix, limit):
-    conn = sqlite3.connect(db_name)
-
-    sql = (
-        "SELECT json FROM event_json as j "
-        "INNER JOIN events as e ON e.event_id = j.event_id "
-        "WHERE j.room_id = ?"
-    )
-
-    args = [room_id]
-
-    if limit:
-        sql += (
-            " ORDER BY topological_ordering DESC, stream_ordering DESC "
-            "LIMIT ?"
-        )
-
-        args.append(limit)
-
-    c = conn.execute(sql, args)
-
-    events = [FrozenEvent(json.loads(e[0])) for e in c.fetchall()]
-
-    events.sort(key=lambda e: e.depth)
-
-    node_map = {}
-    state_groups = {}
-
-    graph = pydot.Dot(graph_name="Test")
-
-    for event in events:
-        c = conn.execute(
-            "SELECT state_group FROM event_to_state_groups "
-            "WHERE event_id = ?",
-            (event.event_id,)
-        )
-
-        res = c.fetchone()
-        state_group = res[0] if res else None
-
-        if state_group is not None:
-            state_groups.setdefault(state_group, []).append(event.event_id)
-
-        t = datetime.datetime.fromtimestamp(
-            float(event.origin_server_ts) / 1000
-        ).strftime('%Y-%m-%d %H:%M:%S,%f')
-
-        content = json.dumps(event.get_dict()["content"])
-
-        label = (
-            "<"
-            "<b>%(name)s </b><br/>"
-            "Type: <b>%(type)s </b><br/>"
-            "State key: <b>%(state_key)s </b><br/>"
-            "Content: <b>%(content)s </b><br/>"
-            "Time: <b>%(time)s </b><br/>"
-            "Depth: <b>%(depth)s </b><br/>"
-            "State group: %(state_group)s<br/>"
-            ">"
-        ) % {
-            "name": event.event_id,
-            "type": event.type,
-            "state_key": event.get("state_key", None),
-            "content": cgi.escape(content, quote=True),
-            "time": t,
-            "depth": event.depth,
-            "state_group": state_group,
-        }
-
-        node = pydot.Node(
-            name=event.event_id,
-            label=label,
-        )
-
-        node_map[event.event_id] = node
-        graph.add_node(node)
-
-    for event in events:
-        for prev_id, _ in event.prev_events:
-            try:
-                end_node = node_map[prev_id]
-            except:
-                end_node = pydot.Node(
-                    name=prev_id,
-                    label="<<b>%s</b>>" % (prev_id,),
-                )
-
-                node_map[prev_id] = end_node
-                graph.add_node(end_node)
-
-            edge = pydot.Edge(node_map[event.event_id], end_node)
-            graph.add_edge(edge)
-
-    for group, event_ids in state_groups.items():
-        if len(event_ids) <= 1:
-            continue
-
-        cluster = pydot.Cluster(
-            str(group),
-            label="<State Group: %s>" % (str(group),)
-        )
-
-        for event_id in event_ids:
-            cluster.add_node(node_map[event_id])
-
-        graph.add_subgraph(cluster)
-
-    graph.write('%s.dot' % file_prefix, format='raw', prog='dot')
-    graph.write_svg("%s.svg" % file_prefix, prog='dot')
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(
-        description="Generate a PDU graph for a given room by talking "
-                    "to the given homeserver to get the list of PDUs. \n"
-                    "Requires pydot."
-    )
-    parser.add_argument(
-        "-p", "--prefix", dest="prefix",
-        help="String to prefix output files with",
-        default="graph_output"
-    )
-    parser.add_argument(
-        "-l", "--limit",
-        help="Only retrieve the last N events.",
-    )
-    parser.add_argument('db')
-    parser.add_argument('room')
-
-    args = parser.parse_args()
-
-    make_graph(args.db, args.room, args.prefix, args.limit)