Disassembler/Spotter-VM
1
0
Fork 0
Code Issues 78 Pull Requests Actions Packages Projects Releases Wiki Activity

Add config file locking

Browse Source
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
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
basic/srv/vm/mgr/config.py
View File

@ -1,20 +1,27 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import fcntl
import json import json
CONF_FILE = '/srv/vm/config.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: class Config:
def __init__(self): def __init__(self):
self.load() self.load()
def load(self): def load(self):
with open(CONF_FILE, 'r') as f: with open(LOCK_FILE, 'w') as l:
self.data = json.load(f) fcntl.flock(l, fcntl.LOCK_EX)
with open(CONF_FILE, 'r') as f:
self.data = json.load(f)
def save(self): def save(self):
with open(CONF_FILE, 'w') as f: with open(LOCK_FILE, 'w') as l:
json.dump(self.data, f, sort_keys=True, indent=4) 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): def __getitem__(self, attr):
return self.data[attr] return self.data[attr]