from datetime import datetime, timedelta

import django_tables2 as tables
from django_tables2.utils import A

from core.tables import CoreTable

from .models import Device


class DevicesTable(CoreTable):
    id = tables.Column(visible=False)
    name = tables.Column("Device", linkify=True)
    description = tables.Column(
        attrs={"td": {"class": "text-truncate", "style": "max-width: 150px;"}}
    )
    delete = tables.LinkColumn(
        "device_delete",
        text="delete",
        args=[A("pk")],
        attrs={
            "a": {
                "class": "delete material-icons",
            }
        },
        orderable=False,
    )

    class Meta(CoreTable.Meta):
        model = Device


class WarrantiesTable(CoreTable):
    customer = tables.Column(linkify=True, orderable=False)
    device = tables.Column(linkify=True)
    valid_from = tables.Column()
    valid_until = tables.Column()
    warranty_type = tables.Column()
    delete = tables.LinkColumn(
        "warranty_delete",
        text="delete",
        args=[A("pk")],
        attrs={
            "a": {
                "class": "delete material-icons",
            }
        },
        orderable=False,
    )

    class Meta(CoreTable.Meta):
        pass

    def render_valid_until(self, value, column):
        today = datetime.date(datetime.today())
        one_year_from_today = datetime.date(datetime.today() + timedelta(365))

        if value > one_year_from_today:
            column.attrs = {"td": {}}
            return value
        if value <= today:
            column.attrs = {"td": {"bgcolor": "red"}}
            return value
        if value <= one_year_from_today:
            column.attrs = {"td": {"bgcolor": "orange"}}
            return value