Do not crash with malformed visitors.json files

If the visitors.json file does not contain a valid JSON file, reset
it, and act as if it was empty.
pull/35/head
Marcos Marado 2020-10-03 17:28:10 +01:00
parent f243391418
commit acea6018a6
1 changed files with 18 additions and 14 deletions

View File

@ -229,6 +229,7 @@ class Plant(object):
visitors_this_check = [] visitors_this_check = []
if os.path.isfile(visitor_filepath): if os.path.isfile(visitor_filepath):
with open(visitor_filepath, 'r') as visitor_file: with open(visitor_filepath, 'r') as visitor_file:
try:
data = json.load(visitor_file) data = json.load(visitor_file)
if data: if data:
for element in data: for element in data:
@ -245,6 +246,9 @@ class Plant(object):
pass pass
with open(visitor_filepath, 'w') as visitor_file: with open(visitor_filepath, 'w') as visitor_file:
visitor_file.write('[]') visitor_file.write('[]')
except:
with open(visitor_filepath, 'w') as visitor_file:
visitor_file.write('[]')
else: else:
with open(visitor_filepath, mode='w') as f: with open(visitor_filepath, mode='w') as f:
json.dump([], f) json.dump([], f)