CardManager.py

import json
import math


class CardManager:
    def __init__(self):
        try:
            self.card_db = json.load(open("vocab.json", "r+"))
        except IOError:
            self.card_db = json.load(open("vocab.json", "w+"))

    def save(self):
        json.dump(self.card_db, open("vocab.json", "w"), indent=2, ensure_ascii=False)

    def get_card_sets(self):
        return self.card_db

    def get_card_set(self, id):
        for set in self.card_db:
            if set["id"] == id:
                return set
        return None

    def create_card_set(self, name):
        self.card_db.append({
            "id": self.card_db[-1]["id"] + 1 if self.card_db else 1,
            "name": name,
            "cards": []
        })
        self.save()

    def delete_card_set(self, id):
        self.card_db = [set for set in self.card_db if set["id"] != id]
        self.save()

    def create_card(self, set_id, question, answer):
        for set in self.card_db:
            if set["id"] == set_id:
                set["cards"].append({
                    "id": set["cards"][-1]["id"] + 1 if set["cards"] else 1,
                    "question": question,
                    "answer": answer,
                    "level": 1
                })
        self.save()

    def delete_card(self, set_id, id):
        for set in self.card_db:
            if set["id"] == set_id:
                set["cards"] = [card for card in set["cards"] if card["id"] != id]
        self.save()

    def level_up_card(self, set_id, id):
        for set in self.card_db:
            if set["id"] == set_id:
                for card in set["cards"]:
                    if card["id"] == id:
                        if card["level"] == 9: return
                        card["level"] += 1
        self.save()

    def level_down_card(self, set_id, id):
        for set in self.card_db:
            if set["id"] == set_id:
                for card in set["cards"]:
                    if card["id"] == id:
                        if card["level"] == 1: return
                        card["level"] -= 1
        self.save()

    def get_card_mastery(self, set_id, id):
        for set in self.card_db:
            if set["id"] == set_id:
                for card in set["cards"]:
                    if card["id"] == id:
                        return math.ceil((card["level"] + 1) / 2)
        return None