knowledge.py 4.26 KB

# python standard library
import random
from  datetime import datetime
import logging

# libraries
import networkx as nx

# this project
import questions

# setup logger for this module
logger = logging.getLogger(__name__)

# ----------------------------------------------------------------------------
# contains the kowledge state of each student.
class Knowledge(object):
    def __init__(self, depgraph, state={}, student=''):
        self.student = student
        self.depgraph = depgraph
        self.topic_sequence = nx.topological_sort(self.depgraph) # FIXME

        # state = {'topic_id': {'level':0.5, 'date': datetime}, ...}
        self.state = state

        # select a topic to do
        self.new_topic()

    # ------------------------------------------------------------------------
    def new_topic(self, topic=None):
        if topic is None:
            # select the first topic that has level < 0.9
            for topic in self.topic_sequence:
                if topic not in self.state or self.state[topic]['level'] < 0.8:
                    break
        logger.debug(f'Student {self.student}: new_topic({topic})')

        # FIXME if all are > 0.9, will stay in the last one forever...
        self.current_topic = topic
        self.current_topic_idx = self.topic_sequence.index(topic)
        self.questions = self.generate_questions_for_topic(topic)
        self.current_question = None
        self.finished_questions = []
        self.correct_answers = 1
        self.wrong_answers = 0

    # ------------------------------------------------------------------------
    def generate_questions_for_topic(self, topic):
        logger.debug(f'Student {self.student}: generate_questions_for_topic "{topic}"')

        factory_list = self.depgraph.node[topic]['factory']
        return [q.generate() for q in factory_list]

    # ------------------------------------------------------------------------
    def get_current_question(self):
        return self.current_question

    # ------------------------------------------------------------------------
    def get_current_topic(self):
        return self.current_topic

    # ------------------------------------------------------------------------
    def get_knowledge_state(self):
        ts = []
        for t in self.topic_sequence:
            if t in self.state:
                ts.append((t, self.state[t]['level']))
            else:
                ts.append((t, 0.0))
        return ts

    # ------------------------------------------------------------------------
    def get_topic_progress(self):
        return len(self.finished_questions) / (len(self.finished_questions) + len(self.questions))

    # ------------------------------------------------------------------------
    # if answer to current question is correct generates a new question
    # otherwise returns None
    def new_question(self):
        if self.current_question is None or \
            self.current_question.get('grade', 0.0) > 0.9:

            # if no more questions in this topic, go to the next one
            # keep going if there are no questions in the next topics
            while not self.questions:
                self.state[self.current_topic] = {
                    'level': self.correct_answers / (self.correct_answers + self.wrong_answers),
                    'date': datetime.now()
                }
                self.new_topic()

            self.current_question = self.questions.pop(0)
            self.current_question['start_time'] = datetime.now()
            self.finished_questions.append(self.current_question)

            logger.debug(f'Student {self.student}: new_question({self.current_question["ref"]})')
            return self.current_question

    # --- checks answer ------------------------------------------------------
    #     returns current question with correction, time and comments updated
    def check_answer(self, answer):
        question = self.current_question
        if question is not None:
            question['finish_time'] = datetime.now()
            grade = question.correct(answer)
            if grade > 0.9:
                self.correct_answers += 1
            else:
                self.wrong_answers +=1

            logger.debug(f'Student {self.student}: check_answer({answer}) = {grade}')

        return question