punky/botany
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Do not crash with malformed visitors.json files

Browse Source 
If the visitors.json file does not contain a valid JSON file, reset
it, and act as if it was empty.
This commit is contained in:
Marcos Marado 2020-10-03 17:28:10 +01:00
parent f243391418
commit acea6018a6
1 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
botany.py
View File

@ -229,20 +229,24 @@ class Plant(object):
visitors_this_check = []
if os.path.isfile(visitor_filepath):
with open(visitor_filepath, 'r') as visitor_file:
data = json.load(visitor_file)
if data:
for element in data:
if element['user'] not in self.visitors:
self.visitors.append(element['user'])
if element['user'] not in visitors_this_check:
visitors_this_check.append(element['user'])
# prevent users from manually setting watered_time in the future
if element['timestamp'] <= int(time.time()):
guest_timestamps.append(element['timestamp'])
try:
self.update_visitor_db(visitors_this_check)
except:
pass
try:
data = json.load(visitor_file)
if data:
for element in data:
if element['user'] not in self.visitors:
self.visitors.append(element['user'])
if element['user'] not in visitors_this_check:
visitors_this_check.append(element['user'])
# prevent users from manually setting watered_time in the future
if element['timestamp'] <= int(time.time()):
guest_timestamps.append(element['timestamp'])
try:
self.update_visitor_db(visitors_this_check)
except:
pass
with open(visitor_filepath, 'w') as visitor_file:
visitor_file.write('[]')
except:
with open(visitor_filepath, 'w') as visitor_file:
visitor_file.write('[]')
else: