from .data import randomise_list

def parse_test_code(code):
    return code.replace('—', '').lower()

def generate_questions(dataset:list):
    output = []
    for block in randomise_list(dataset):
        if block['type'] == 'question':
            question = {
                'type': 'question',
                'q_no': block['q_no'],
                'question_header': '',
                'text': block['text']
            }
            if block['q_type'] == 'Multiple Choice': question['options'] = randomise_list([*enumerate(block['options'])])
            else: question['options'] = block['options'].copy()
            output.append(question)
        elif block['type'] == 'block':
            for key, _question in enumerate(randomise_list(block['questions'])):
                question = {
                    'type': 'block',
                    'q_no': _question['q_no'],
                    'question_header': block['question_header'] if 'question_header' in block else '',
                    'block_length': len(block['questions']),
                    'block_q_no': key,
                    'text': _question['text']
                }
                if _question['q_type'] == 'Multiple Choice': question['options'] = randomise_list([*enumerate(_question['options'])])
                else: question['options'] = _question['options'].copy()
                output.append(question)
    return output

def evaluate_answers(answers:dict, key:list):
    score = 0
    max = 0
    tags = {}
    for block in key:
        if block['type'] == 'question':
            max += 1
            q_no = block['q_no']
            if str(q_no) in answers:
                submitted_answer = int(answers[str(q_no)])
                if submitted_answer == block['correct']:
                    score += 1
                    for tag in block['tags']:
                        if tag not in tags:
                            tags[tag] = {
                                'scored': 1,
                                'max': 1
                            }
                        else:
                            tags[tag]['scored'] += 1
                            tags[tag]['max'] += 1
                else:
                    for tag in block['tags']:
                        if tag not in tags:
                            tags[tag] = {
                                'scored': 0,
                                'max': 1
                            }
                        else: tags[tag]['max'] += 1
        elif block['type'] == 'block':
            for question in block['questions']:
                max += 1
                q_no = question['q_no']
                if str(q_no) in answers:
                    submitted_answer = int(answers[str(q_no)])
                    if submitted_answer == question['correct']:
                        score += 1
                        for tag in question['tags']:
                            if tag not in tags:
                                tags[tag] = {
                                    'scored': 1,
                                    'max': 1
                                }
                            else:
                                tags[tag]['scored'] += 1
                                tags[tag]['max'] += 1
                    else:
                        for tag in question['tags']:
                            if tag not in tags:
                                tags[tag] = {
                                    'scored': 0,
                                    'max': 1
                                }
                            else: tags[tag]['max'] += 1
    grade = 'merit' if score/max >= .85 else 'pass' if score/max >= .70 else 'fail'
    return {
        'grade': grade,
        'tags': tags,
        'score': score,
        'max': max
    }

def get_correct_answers(dataset:list):
    output = {}
    for block in dataset:
        if block['type'] == 'question':
            output[str(block['q_no'])] = block['options'][block['correct']]
        if block['type'] == 'block':
            for question in block['questions']:
                output[str(question['q_no'])] = question['options'][question['correct']]
    return output