"""Daily Usage Sensors"""
from .fplEntity import FplEnergyEntity, FplMoneyEntity


class FplDailyUsageSensor(FplMoneyEntity):
    """Daily Usage Cost Sensor"""

    def __init__(self, coordinator, config, account):
        super().__init__(coordinator, config, account, "Daily Usage")

    @property
    def state(self):
        data = self.getData("daily_usage")

        if data is not None and len(data) > 0 and "cost" in data[-1].keys():
            return data[-1]["cost"]

        return None

    def defineAttributes(self):
        """Return the state attributes."""
        data = self.getData("daily_usage")
        attributes = {}
        attributes["state_class"] = "total_increasing"
        if data is not None and len(data) > 0 and "readTime" in data[-1].keys():
            attributes["date"] = data[-1]["readTime"]

        return attributes


class FplDailyUsageKWHSensor(FplEnergyEntity):
    """Daily Usage Kwh Sensor"""

    def __init__(self, coordinator, config, account):
        super().__init__(coordinator, config, account, "Daily Usage KWH")

    @property
    def state(self):
        data = self.getData("daily_usage")

        if data is not None and len(data) > 0 and "usage" in data[-1].keys():
            return data[-1]["usage"]

        return None

    def defineAttributes(self):
        """Return the state attributes."""
        data = self.getData("daily_usage")
        attributes = {}
        attributes["state_class"] = "total_increasing"

        if data is not None:
            if data[-1] is not None and "readTime" in data[-1].keys():
                attributes["date"] = data[-1]["readTime"]
            if data[-2] is not None and "readTime" in data[-2].keys():
                attributes["last_reset"] = data[-2]["readTime"]

        return attributes


class FplDailyReceivedKWHSensor(FplEnergyEntity):
    """daily received Kwh sensor"""

    def __init__(self, coordinator, config, account):
        super().__init__(coordinator, config, account, "Daily Received KWH")

    @property
    def state(self):
        data = self.getData("daily_usage")
        if data is not None and len(data) > 0 and "netReceivedKwh" in data[-1].keys():
            return data[-1]["netReceivedKwh"]
        return 0

    def defineAttributes(self):
        """Return the state attributes."""
        data = self.getData("daily_usage")

        attributes = {}
        attributes["state_class"] = "total_increasing"
        attributes["date"] = data[-1]["readTime"]
        attributes["last_reset"] = data[-2]["readTime"]
        return attributes


class FplDailyDeliveredKWHSensor(FplEnergyEntity):
    """daily delivered Kwh sensor"""

    def __init__(self, coordinator, config, account):
        super().__init__(coordinator, config, account, "Daily Delivered KWH")

    @property
    def state(self):
        data = self.getData("daily_usage")
        if data is not None and len(data) > 0 and "netDeliveredKwh" in data[-1].keys():
            return data[-1]["netDeliveredKwh"]
        return 0

    def defineAttributes(self):
        """Return the state attributes."""
        data = self.getData("daily_usage")

        attributes = {}
        attributes["state_class"] = "total_increasing"
        attributes["date"] = data[-1]["readTime"]
        attributes["last_reset"] = data[-2]["readTime"]
        return attributes