add judgement for active_flow parameter

1 files changed, 4 insertions, 1 deletions

diff --git a/app.py b/app.py
index c545d9b..cbe8cc4 100644
--- a/app.py
+++ b/app.py
@@ -84,7 +84,10 @@ def run_traffic_env():
             yaml_name = json_data.get("yaml_name") + ".yaml"
             m = json_data.get("m")
             d = json_data.get("d")
-            active_flows = json_data.get("active_flows")
+            if "active_flows" in json_data:
+                active_flows = json_data.get("active_flows")
+            else:
+                active_flows = 0
             clients_start_ip = json_data.get("clients_start_ip")
             clients_end_ip = json_data.get("clients_end_ip")
             servers_start_ip = json_data.get("servers_start_ip")