87 lines
4.5 KiB
Python
87 lines
4.5 KiB
Python
|
from data.sources import sources
|
||
|
|
from data.playbooks import playbooks
|
||
|
|
from data.labels import labels
|
||
|
|
from data.questions import questions
|
||
|
|
from random import randrange
|
||
|
|
|
||
|
|
def evaluate_quiz(submission):
|
||
|
|
# Set up dictionaries to count scores
|
||
|
|
results = {
|
||
|
|
'playbooks' : {},
|
||
|
|
'selected_playbooks': {},
|
||
|
|
'display_labels': [],
|
||
|
|
'max_score': 0
|
||
|
|
}
|
||
|
|
for playbook in playbooks:
|
||
|
|
if playbooks[playbook]['source'] in set.intersection(set(sources), set(submission)):
|
||
|
|
results['playbooks'][playbook] = 0
|
||
|
|
|
||
|
|
for answer in submission:
|
||
|
|
if answer.startswith('q'):
|
||
|
|
qno = int(answer[1:]) - 1
|
||
|
|
ano = int(submission[answer]) - 1
|
||
|
|
match_list = questions[qno]['answers'][ano]['matches']
|
||
|
|
for match in match_list:
|
||
|
|
if match in results['playbooks']: results['playbooks'][match] += match_list[match]
|
||
|
|
results['max_score'] += max(match_list.values())
|
||
|
|
high_score = max(results['playbooks'].values())
|
||
|
|
for playbook, score in results['playbooks'].items():
|
||
|
|
if score == high_score:
|
||
|
|
results['selected_playbooks'][playbook] = {}
|
||
|
|
for label in labels:
|
||
|
|
if 'custom' not in labels[label] or not labels[label]['custom']:
|
||
|
|
results['selected_playbooks'][playbook][label] = playbooks[playbook]['labels'][label]
|
||
|
|
if label not in results['display_labels']: results['display_labels'].append(label)
|
||
|
|
else:
|
||
|
|
if playbook in labels[label]['playbooks']:
|
||
|
|
if label in playbooks[playbook]['labels']:
|
||
|
|
results['selected_playbooks'][playbook][label] = playbooks[playbook]['labels'][label]
|
||
|
|
else:
|
||
|
|
results['selected_playbooks'][playbook][label] = labels[label]['default_value']
|
||
|
|
if label not in results['display_labels']: results['display_labels'].append(label)
|
||
|
|
|
||
|
|
if 'joined' in results['selected_playbooks']:
|
||
|
|
if len(results['selected_playbooks']) > 1:
|
||
|
|
l = list(results['selected_playbooks'])
|
||
|
|
l.remove('joined')
|
||
|
|
i = randrange(len(l)-1)
|
||
|
|
results['joined_cloned'] = l[i]
|
||
|
|
results['selected_playbooks']['joined'] = results['selected_playbooks'][l[i]].copy()
|
||
|
|
else:
|
||
|
|
d = results['playbooks'].copy()
|
||
|
|
del d['joined']
|
||
|
|
high_score = max(d.values())
|
||
|
|
p = []
|
||
|
|
for playbook, score in d.items():
|
||
|
|
if score == high_score:
|
||
|
|
p.append(playbook)
|
||
|
|
p_sel = p[randrange(len(p)-1)] if len(p) > 1 else p[0]
|
||
|
|
results['selected_playbooks'][p_sel] = {}
|
||
|
|
results['joined_cloned'] = p_sel
|
||
|
|
for label in labels:
|
||
|
|
if 'custom' not in labels[label] or not labels[label]['custom']:
|
||
|
|
results['selected_playbooks'][p_sel][label] = playbooks[playbook]['labels'][label]
|
||
|
|
if label not in results['display_labels']: results['display_labels'].append(label)
|
||
|
|
else:
|
||
|
|
if playbook in labels[label]['playbooks']:
|
||
|
|
if label in playbooks[playbook]['labels']:
|
||
|
|
results['selected_playbooks'][p_sel][label] = playbooks[playbook]['labels'][label]
|
||
|
|
else:
|
||
|
|
results['selected_playbooks'][p_sel][label] = labels[label]['default_value']
|
||
|
|
if label not in results['display_labels']: results['display_labels'].append(label)
|
||
|
|
results['selected_playbooks']['joined'] = results['selected_playbooks'][p_sel].copy()
|
||
|
|
|
||
|
|
for answer in submission:
|
||
|
|
if answer.startswith('q'):
|
||
|
|
qno = int(answer[1:]) - 1
|
||
|
|
ano = int(submission[answer]) - 1
|
||
|
|
for increase in questions[qno]['answers'][ano]['increase']:
|
||
|
|
for playbook in results['selected_playbooks']:
|
||
|
|
if increase in results['selected_playbooks'][playbook]:
|
||
|
|
if results['selected_playbooks'][playbook][increase] < 3: results['selected_playbooks'][playbook][increase] += 1
|
||
|
|
for decrease in questions[qno]['answers'][ano]['decrease']:
|
||
|
|
for playbook in results['selected_playbooks']:
|
||
|
|
if decrease in results['selected_playbooks'][playbook]:
|
||
|
|
if results['selected_playbooks'][playbook][decrease] > -2: results['selected_playbooks'][playbook][decrease] -= 1
|
||
|
|
return results
|