Add config file locking

This commit is contained in:
Disassembler 2018-10-28 18:19:45 +01:00
parent cb0d0012c9
commit f675996e60
Signed by: Disassembler
GPG Key ID: 524BD33A0EE29499

View File

@ -1,20 +1,27 @@
# -*- coding: utf-8 -*-
import fcntl
import json
CONF_FILE = '/srv/vm/config.json'
# Locking is needed in order to prevent race conditions in WSGI threads
LOCK_FILE = '/srv/vm/config.lock'
class Config:
def __init__(self):
self.load()
def load(self):
with open(CONF_FILE, 'r') as f:
self.data = json.load(f)
with open(LOCK_FILE, 'w') as l:
fcntl.flock(l, fcntl.LOCK_EX)
with open(CONF_FILE, 'r') as f:
self.data = json.load(f)
def save(self):
with open(CONF_FILE, 'w') as f:
json.dump(self.data, f, sort_keys=True, indent=4)
with open(LOCK_FILE, 'w') as l:
fcntl.flock(l, fcntl.LOCK_EX)
with open(CONF_FILE, 'w') as f:
json.dump(self.data, f, sort_keys=True, indent=4)
def __getitem__(self, attr):
return self.data[attr]