Mercurial > repos > shellac > sam_consensus_v3
view env/share/doc/networkx-2.5/examples/drawing/plot_unix_email.py @ 0:4f3585e2f14b draft default tip
"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"
author | shellac |
---|---|
date | Mon, 22 Mar 2021 18:12:50 +0000 |
parents | |
children |
line wrap: on
line source
""" ========== Unix Email ========== Create a directed graph, allowing multiple edges and self loops, from a unix mailbox. The nodes are email addresses with links that point from the sender to the receivers. The edge data is a Python email.Message object which contains all of the email message data. This example shows the power of `DiGraph` to hold edge data of arbitrary Python objects (in this case a list of email messages). The sample unix email mailbox called "unix_email.mbox" may be found here: - https://github.com/networkx/networkx/blob/master/examples/drawing/unix_email.mbox """ from email.utils import getaddresses, parseaddr import mailbox import matplotlib.pyplot as plt import networkx as nx # unix mailbox recipe # see https://docs.python.org/3/library/mailbox.html def mbox_graph(): mbox = mailbox.mbox("unix_email.mbox") # parse unix mailbox G = nx.MultiDiGraph() # create empty graph # parse each messages and build graph for msg in mbox: # msg is python email.Message.Message object (source_name, source_addr) = parseaddr(msg["From"]) # sender # get all recipients # see https://docs.python.org/3/library/email.html tos = msg.get_all("to", []) ccs = msg.get_all("cc", []) resent_tos = msg.get_all("resent-to", []) resent_ccs = msg.get_all("resent-cc", []) all_recipients = getaddresses(tos + ccs + resent_tos + resent_ccs) # now add the edges for this mail message for (target_name, target_addr) in all_recipients: G.add_edge(source_addr, target_addr, message=msg) return G G = mbox_graph() # print edges with message subject for (u, v, d) in G.edges(data=True): print(f"From: {u} To: {v} Subject: {d['message']['Subject']}") pos = nx.spring_layout(G, iterations=10) nx.draw(G, pos, node_size=0, alpha=0.4, edge_color="r", font_size=16, with_labels=True) plt.show()