data_handler.py

import os
import connection

ANSWER_DATA_FILE_PATH = os.getcwd() + "/data/answer.csv"
QUESTION_DATA_FILE_PATH = os.getcwd() + "/data/question.csv"


def get_answers():
    database = connection.csv_to_dict(ANSWER_DATA_FILE_PATH)
    return database


def get_questions():
    database = connection.csv_to_dict(QUESTION_DATA_FILE_PATH)
    return database


def save_questions(data):
    database = connection.dict_to_csv(QUESTION_DATA_FILE_PATH, data)
    return database


def save_answers(data):
    database = connection.dict_to_csv(ANSWER_DATA_FILE_PATH, data, True)
    return database


def add_entry(entry, is_answer=False):
    if not is_answer:
        connection.append_to_csv(QUESTION_DATA_FILE_PATH, entry)
    else:
        connection.append_to_csv(ANSWER_DATA_FILE_PATH, entry)


def get_question(question_id, question_database):
    for question_data in question_database:
        if question_data['id'] == question_id:
            return question_data


def get_answer(answer_id, answer_database):
    for answer_data in answer_database:
        if answer_data['id'] == answer_id:
            return answer_data


def get_question_related_answers(question_id, answer_database):
    answers_of_question = []
    for answer_data in answer_database:
        if answer_data['question_id'] == question_id:
            answers_of_question.append(answer_data)
    return answers_of_question


def update_questions(question_id, updated_data):
    all_questions = get_questions()
    question = get_question(question_id, all_questions)
    question_index = all_questions.index(question)

    for key, value in updated_data.items():
        question[key] = value
    all_questions[question_index] = question
    save_questions(all_questions)
    return question


def delete_record(id, answer=False, delete=False):
    if answer:
        answers = get_answers()
        for i, answer in enumerate(answers):
            question_id = answer['question_id']
            if answer['id'] == id and delete:
                del answers[i]
                save_answers(answers)
            return question_id