import json
from pprint import pprint
from random import shuffle, randint
from collections import Counter

from django.core.management.base import BaseCommand

from university.models import OnlineExamQuestion


class Command(BaseCommand):

    def handle(self, *args, **options):
        question_ids = []
        answers = {}
        with open('pctp_exam.json', 'r') as f:
            exam = json.loads(f.read())
            question_ids = [q['id'] for q in exam[0]['questions']] + \
                           [q['id'] for q in exam[1]['questions']] + \
                           [q['id'] for q in exam[2]['questions']]

        questions = OnlineExamQuestion.objects.filter(id__in=question_ids)
        for q in questions:
            if q.question_type == OnlineExamQuestion.TYPE_SINGLE_CHOICE:
                answers[str(q.id)] = q.answer[0]
            else:
                answers[str(q.id)] = q.answer

        with open('pctp_exam_answers.json', 'w') as f:
            f.write(json.dumps(answers))