Improve debug

1 files changed, 7 insertions, 1 deletions

diff --git a/esds/debug.py b/esds/debug.py
index 168033a..fe19745 100644
--- a/esds/debug.py
+++ b/esds/debug.py
@@ -11,6 +11,7 @@ class Debug:
         self.simulator=simulator
         self.file_path=file_path
         self.loop_count=0
+        self.logs=list()
         header={
             "python_version" : sys.version,
             "simulation_started_at": simulator.startat,
@@ -20,6 +21,9 @@ class Debug:
         }
         self.write(header,append=False)
 
+    def append_log(self,log):
+        self.logs.append(log)
+
     def write(self,data, append=True):
         mode="a" if append else "w"
         with open(self.file_path, mode) as f:
@@ -95,6 +99,8 @@ class Debug:
             "simulated_time_accurate": self.simulator.time,
             "network_interfaces": self.get_network_interfaces(),
             "events_list": self.get_events_list(),
-            "nodes_infos": self.get_nodes_infos()
+            "nodes_infos": self.get_nodes_infos(),
+            "logs": self.logs
         }
         self.write(loop_data)
+        self.logs.clear()