File permissions, optimized db access

pull/8/head
Jake Funke 2018-03-11 15:44:10 -07:00
parent f716b4ab96
commit a74d257506
1 changed files with 15 additions and 13 deletions

View File

@ -225,6 +225,7 @@ class Plant(object):
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'])
@ -232,15 +233,16 @@ class Plant(object):
visitors_this_check.append(element['user'])
if element['timestamp'] > latest_timestamp:
latest_timestamp = element['timestamp']
with open(visitor_filepath, 'w') as visitor_file:
visitor_file.write('[]')
try:
self.update_visitor_db(visitors_this_check)
except:
pass
with open(visitor_filepath, 'w') as visitor_file:
visitor_file.write('[]')
else:
with open(visitor_filepath, mode='w') as f:
json.dump([], f)
os.chmod(visitor_filepath, 0666)
return latest_timestamp
def water_check(self):