snorklerjoe/CubeServer

"""Tables for data-related stuff."""

from typing import List
from flask_table import Table, Col

from cubeserver_common.models.datapoint import DataPoint, DataClass
from cubeserver_app.tables.columns import (
    OptionsCol,
    EnumCol,
    TextEditCol,
    FloatCol,
    AdminTeamNameCol,
)

__all__ = ["AdminDataTable", "LeaderboardDataTable"]


class AdminDataTable(Table):
    """Allows a group of DataPoint objects to be displayed in an HTML table"""

    allow_sort = False  # Let's avoid flask-table's built-in sorting
    classes = [
        "table",
        "table-striped",
        "datatable",
        "display",
        "bg-dark",
        "datapoints-table",
    ]
    thead_classes = ["thead-dark"]
    border = True
    allow_empty = True

    moment = Col("Datetime")
    team_str = AdminTeamNameCol("Team", th_html_attrs={"data-orderable": "false"})
    category = EnumCol("Category", allow_sort="category")
    value_with_unit = Col("Value", allow_sort="value")
    rawscore = TextEditCol("Raw Point Value", model_type="DataPoint")
    score = FloatCol("Score", allow_sort="rawscore")
    id = OptionsCol(
        "Options", model_type="DataPoint", th_html_attrs={"data-orderable": "false"}
    )

    def __init__(self, items: List[DataPoint], **kwargs):
        """Initializes the table"""
        super().__init__(items, **kwargs)

    def sort_url(self, col_id, reverse=False):
        pass
        # return url_for(self._endpoint, sort=col_id,
        #               direction='desc' if reverse else 'asc')


class LeaderboardDataTable(Table):
    """Allows a group of DataPoint objects to be displayed in an HTML table"""

    allow_sort = False  # Let's avoid flask-table's built-in sorting
    classes = ["table", "table-striped", "datatable", "display", "datapoints-table"]
    thead_classes = ["thead-dark"]
    border = True
    allow_empty = True

    moment = Col("Datetime")
    category = EnumCol("Category", allow_sort="category")
    value_with_unit = Col("Value", allow_sort="value")
    rawscore = FloatCol("Point Value")
    score = FloatCol("Score", allow_sort="rawscore")

    def __init__(self, items: List[DataPoint], **kwargs):
        """Initializes the table"""
        super().__init__(items, **kwargs)

    def sort_url(self, col_id, reverse=False):
        pass
        # return url_for(self._endpoint, sort=col_id,
        #               direction='desc' if reverse else 'asc')