From 9452ce9080c51dc0a6f1ad13d9393cf179f0f2b2 Mon Sep 17 00:00:00 2001 From: Wapiti Date: Mon, 13 Dec 2021 10:33:22 +0000 Subject: [PATCH] Update app.py avoid conflict on non-generated requests_status.log file --- app.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/app.py b/app.py index d0a6580..be0d3b3 100644 --- a/app.py +++ b/app.py @@ -14,6 +14,7 @@ from collections import OrderedDict import json import config +from pathlib import Path # 实例化flask app = Flask(__name__) @@ -103,10 +104,13 @@ def index(): with open('./output/attack_status.log', 'r') as fp: attacks = json.load(fp) # 读取请求记录 - with open('./output/requests_status.log', 'r') as fp: + if not Path('./output/requests_status.log').is_file(): global ip_from_sort - requests = json.load(fp) - ip_from_sort = requests + requests = ip_from_sort + else: + with open('./output/requests_status.log', 'r') as fp: + requests = json.load(fp) + ip_from_sort = requests # 图表绘制 world_chart = WorldMapChart.world_status() world_id = world_chart._chart_id