from delphi_api.bigtable import BigTableModel, IntegerAttribute, TextAttribute
from delphi_api.v3.enums import ColumnFamily, ColumnName, TableName


class ChartPosition(BigTableModel):
    """Bigtable model for chart track positions"""
    class Meta:
        table_name = TableName.charts.value
        country_codes = None
        len_sub_dsps = 1

    date = TextAttribute(
        column_family=ColumnFamily.META.value, attr_name=ColumnName.DATE.value)
    dsp = TextAttribute(
        column_family=ColumnFamily.META.value, attr_name=ColumnName.DSP.value)
    chart_id = TextAttribute(
        column_family=ColumnFamily.META.value, attr_name=ColumnName.CHART_ID.value)
    isrc = TextAttribute(
        column_family=ColumnFamily.META.value, attr_name=ColumnName.ISRC.value)
    current = IntegerAttribute(
        column_family=ColumnFamily.POS.value, attr_name=ColumnName.CURRENT.value)