from datetime import datetime
from influxdb_client import Point
import json
class Pulse:
    label = str(__name__).lower()
    def __init__(self,data):
        measurement = data.cache
        timestamp = measurement['timestamp']
        power = measurement['power']
        cost = measurement.get('accumulatedCost',None)
        lastMeterConsumption = measurement['lastMeterConsumption']
        tags = {self.label: ""}
        fields = {
            "power": float(power),
            "lastMeterConsumption": float(lastMeterConsumption)
        }
        if cost is not None and float(cost) > 0:
            fields.update({'accumulatedCost':cost})
        self.datapoint = {"fields": fields,
                     "tags": tags,
                     "measurement": "pulse",
                     "time": timestamp
                     }
    def get_datapoint(self):
        return Point(measurement_name=str(self.label)).from_dict(self.datapoint)

class Price:
    label = str(__name__).lower()
    def __init__(self,data):
        measurement = data.cache
        current_time_utc = datetime.utcnow()
        timestamp = current_time_utc.isoformat()
        total = measurement["total"]
        tax = measurement["tax"]
        tags = {self.label: ""}
        fields = {
            "total": float(total),
            "tax": float(tax)
        }
        self.datapoint = {"fields": fields,
                     "tags": tags,
                     "measurement": "pulse",
                     "time": timestamp
                     }
    def get_datapoint(self):
        return Point(measurement_name=str(self.label)).from_dict(self.datapoint)