table.py

# Built-ins
from dataclasses import dataclass

# Third-party
from flask_table import Table, Col


class ItemTable(Table):
    def sort_url(self, col_id, reverse=False):
        pass

    gem_name = Col("Gem")
    quest_name = Col("Quest")
    gem_level = Col("Level")
    gem_colour = Col("Colour")


@dataclass
class Item:
    gem_name: str
    quest_name: str
    gem_level: int
    gem_colour: str


def generate_table(store):
    for character_class, skill_gems in store:
        if skill_gems:
            yield (
                {
                    "character_class": character_class,
                    "skill_gems": ItemTable(
                        [Item(*skill_gem) for skill_gem in skill_gems]
                    ),
                }
            )