myned/modufur
1
0
Fork 0
mirror of https://github.com/myned/modufur.git synced 2024-12-25 06:37:29 +00:00
Code Issues Activity

Added json to setdefault helper method

Browse source
This commit is contained in:
Myned 2017-12-21 22:39:20 -05:00
parent a3d9ac81d9
commit 4d188e2a4a
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/utils/utils.py
View file

@ -36,7 +36,8 @@ except FileNotFoundError:
'FILE NOT FOUND : config.json created with abstract values. Restart run.py with correct values') 'FILE NOT FOUND : config.json created with abstract values. Restart run.py with correct values')
def setdefault(filename, default=None): def setdefault(filename, default=None, json=False):
if not json:
try: try:
with open(filename, 'rb') as infile: with open(filename, 'rb') as infile:
print('LOADED : {}'.format(filename)) print('LOADED : {}'.format(filename))
@ -47,6 +48,17 @@ def setdefault(filename, default=None):
pkl.dump(default, iofile) pkl.dump(default, iofile)
iofile.seek(0) iofile.seek(0)
return pkl.load(iofile) return pkl.load(iofile)
else:
try:
with open(filename, 'rb') as infile:
print(f'LOADED : {filename}')
return jsn.load(infile)
except FileNotFoundError:
with open(filename, 'wb+') as iofile:
print(f'FILE NOT FOUND : {filename} created and loaded with default values')
jsn.dump(default, iofile)
iofile.seek(0)
return jsn.load(iofile)
def load(filename, *, json=False): def load(filename, *, json=False):