synapse-old/contrib/graph/graph2.py

145 lines
4.1 KiB
Python
Raw Normal View History

2016-01-06 21:26:29 -07:00
# Copyright 2014-2016 OpenMarket Ltd
2014-12-10 09:10:25 -07:00
#
# 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
from synapse.util.frozenutils import unfreeze
2014-12-10 09:10:25 -07:00
2015-01-08 03:53:03 -07:00
def make_graph(db_name, room_id, file_prefix, limit):
2014-12-10 09:10:25 -07:00
conn = sqlite3.connect(db_name)
2015-01-08 03:53:03 -07:00
sql = (
"SELECT json FROM event_json as j "
"INNER JOIN events as e ON e.event_id = j.event_id "
"WHERE j.room_id = ?"
2014-12-10 09:10:25 -07:00
)
2015-01-08 03:53:03 -07:00
args = [room_id]
if limit:
2019-06-20 03:32:02 -06:00
sql += " ORDER BY topological_ordering DESC, stream_ordering DESC " "LIMIT ?"
2015-01-08 03:53:03 -07:00
args.append(limit)
c = conn.execute(sql, args)
2014-12-10 09:10:25 -07:00
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(
2019-06-20 03:32:02 -06:00
"SELECT state_group FROM event_to_state_groups " "WHERE event_id = ?",
(event.event_id,),
2014-12-10 09:10:25 -07:00
)
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
2019-06-20 03:32:02 -06:00
).strftime("%Y-%m-%d %H:%M:%S,%f")
2014-12-10 09:10:25 -07:00
content = json.dumps(unfreeze(event.get_dict()["content"]))
2014-12-10 09:10:25 -07:00
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,
}
2019-06-20 03:32:02 -06:00
node = pydot.Node(name=event.event_id, label=label)
2014-12-10 09:10:25 -07:00
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:
2019-06-20 03:32:02 -06:00
end_node = pydot.Node(name=prev_id, label="<<b>%s</b>>" % (prev_id,))
2014-12-10 09:10:25 -07:00
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
2019-06-20 03:32:02 -06:00
cluster = pydot.Cluster(str(group), label="<State Group: %s>" % (str(group),))
2014-12-10 09:10:25 -07:00
for event_id in event_ids:
cluster.add_node(node_map[event_id])
graph.add_subgraph(cluster)
2019-06-20 03:32:02 -06:00
graph.write("%s.dot" % file_prefix, format="raw", prog="dot")
graph.write_svg("%s.svg" % file_prefix, prog="dot")
2014-12-10 09:10:25 -07:00
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description="Generate a PDU graph for a given room by talking "
2019-06-20 03:32:02 -06:00
"to the given homeserver to get the list of PDUs. \n"
"Requires pydot."
2014-12-10 09:10:25 -07:00
)
parser.add_argument(
2019-06-20 03:32:02 -06:00
"-p",
"--prefix",
dest="prefix",
2015-01-08 03:53:03 -07:00
help="String to prefix output files with",
2019-06-20 03:32:02 -06:00
default="graph_output",
2014-12-10 09:10:25 -07:00
)
2019-06-20 03:32:02 -06:00
parser.add_argument("-l", "--limit", help="Only retrieve the last N events.")
parser.add_argument("db")
parser.add_argument("room")
2014-12-10 09:10:25 -07:00
args = parser.parse_args()
2015-01-08 03:53:03 -07:00
make_graph(args.db, args.room, args.prefix, args.limit)