import networkx
from networkx.drawing.nx_agraph import write_dot
from graphviz import Source

nodes = ['malicious.org','sample_apt.exe','sample_malware.ps1',"fbi.org","wpp-bussiness.exe","china231msam1231jkas.onion"]

network = networkx.Graph()

for node in nodes:
    network.add_node(node,color="red",shape="diamond",label=node)

network.add_edge("malicious.org","sample_apt.exe",label="Malware executado chama o callback para baixar outro arquivo.")
network.add_edge("malicious.org","sample_malware.ps1",label="Malware executado chama o callback para baixar outro arquivo.")
network.add_edge("sample_malware.ps1","fbi.org",label="Chama o fbi onde ele se proprio denuncia pra ficar famoso.")
network.add_edge("fbi.org","wpp-bussiness.exe",penwidth=5,color="yellow",label="Onde baixa o wpp bussiness pra ver quem acessou seu perfil")
network.add_edge("wpp-bussiness.exe","china231msam1231jkas.onion",penwidth=10,color="orange",label="Onde vaza todas as suas conversas com a cremosa")


write_dot(network,"network.dot")

s = Source.from_file("network.dot",format="png")
s.view()