viveksantayana/wanderhome-quiz
viveksantayana/wanderhome-quiz
1
0
Fork 0
Code Issues 2 Pull Requests Projects Releases Wiki Activity

Remove the playbook feature text from storage

Browse Source
This commit is contained in:
Vivek Santayana 2022-08-31 14:50:47 +01:00
parent c6b656cbee
commit afbb41f7bd
2 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server/app/models/entry.py
View File

@ -1,4 +1,5 @@
from ..extensions import db from ..extensions import db
from ..tools.data import declutter_results
from ..tools.models import JsonString from ..tools.models import JsonString
from sqlalchemy_json import MutableJson from sqlalchemy_json import MutableJson
@ -18,8 +19,8 @@ class Entry(db.Model):
def __init__(self, answers:list, results:dict): def __init__(self, answers:list, results:dict):
self.id = uuid4().hex self.id = uuid4().hex
self.timestamp = datetime.utcnow() self.timestamp = datetime.utcnow()
self.answers = answers self.answers = answers.copy()
self.results = results self.results = declutter_results(results)
def add(self): def add(self):
db.session.add(self) db.session.add(self)

10
server/app/tools/data.py
View File

@ -4,6 +4,8 @@ import json
from os.path import isfile from os.path import isfile
from pathlib import Path from pathlib import Path
from copy import deepcopy
def check_file(filename:str) -> bool: def check_file(filename:str) -> bool:
data_dir = Path(app.config.get('DATA')) data_dir = Path(app.config.get('DATA'))
if isfile(f'./{data_dir}/{filename}'): return True if isfile(f'./{data_dir}/{filename}'): return True
@ -12,3 +14,11 @@ def check_file(filename:str) -> bool:
def load(filename:str) -> dict: def load(filename:str) -> dict:
data_dir = Path(app.config.get('DATA')) data_dir = Path(app.config.get('DATA'))
with open(f'./{data_dir}/{filename}') as file: return json.load(file) with open(f'./{data_dir}/{filename}') as file: return json.load(file)
def declutter_results(raw_results:dict) -> dict:
data = deepcopy(raw_results)
playbook_lists = []
for item in data['playbooks']:
playbook_lists.append(list(item.keys())[0])
data['playbooks'] = playbook_lists
return data