From 03d02a7c3f00171bc4ba441873fa3ec3ac3e98fc Mon Sep 17 00:00:00 2001 From: Marcin Kostrzewski Date: Thu, 16 Jun 2022 18:39:16 +0200 Subject: [PATCH] Initialized graph classes --- network_attack_propagation.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 network_attack_propagation.py diff --git a/network_attack_propagation.py b/network_attack_propagation.py new file mode 100644 index 0000000..bcabc56 --- /dev/null +++ b/network_attack_propagation.py @@ -0,0 +1,18 @@ +class Node: + def __init__(self): + self.is_infected = False + + +class Edge: + def __init__(self, node_a: Node, node_b: Node, weight: float): + self.node_a = node_a + self.node_b = node_b + self.weight = weight + + +class Graph: + def __init__(self): + self.edges = [] + + def add_edge(self, edge: Edge): + self.edges.append(edge) \ No newline at end of file