plotwatt-api/plotwattapi.py at master · davidsheldon/plotwatt-api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
"""
Author: Kyle Konrad and Zach Dwiel

This is a basic interface to the plotwatt api.

Please email comments and questions to zdwiel@plotwatt.com
"""

import requests
import base64
from datetime import datetime
import time


class PlotwattError(Exception):
    def __init__(self, message):
        self.message = message

    def __str__(self):
        return "Plotwatt Error: %s " % (self.message)


class Plotwatt(object):
    def __init__(self, house_id, secret, baseurl="http://plotwatt.com"):
        self.house_id = house_id
        self.secret = secret
        self.baseurl = baseurl
        self.base64string = base64.encodestring('%s:%s' % (secret, ''))[:-1]
        self.authheader = "Basic %s" % self.base64string
        self.session = requests.Session()
        self.session.headers.update({"Authorization": self.authheader})
        self.mysql_datetime_format = "'%Y-%m-%d %H:%M:%S'"
        # api actions
        self.list_meters_url = '/api/v2/list_meters'
        self.push_readings_url = '/api/v2/push_readings'
        self.new_meters_url = '/api/v2/new_meters'
        self.delete_meter_url = '/api/v2/delete_meter'

    def _request(self, url, data=None):
        """ make a request to plotwatt.com """
        if data is None:
            res = self.session.get(self.baseurl + url)
        else:
            res = self.session.post(self.baseurl + url, data=data)
        try:
            res.raise_for_status()
        except requests.exceptions.HTTPError as e:
            if e.response.status_code == 422:
                raise PlotwattError(e.response.text)
            raise
        return res

    def create_meters(self, num_meters):
        """ create meters on your plotwatt.com account """
        res = self._request(self.new_meters_url, "number_of_new_meters=%s" % num_meters)
        return res.json()

    def delete_meter(self, meter_id):
        """ delete a meter from your plotwatt.com account """
        return self._request(self.delete_meter_url, "meter_id=%s" % meter_id)

    def list_meters(self):
        """ list the meters connected to your plotwatt.com account """
        res = self._request(self.list_meters_url, '')
        return res.json()

    def push_readings(self, meter, readings, times):
        """call push_readings endpoint. times should be in POSIX format in UTC"""
        meters = [meter] * len(readings)

        def sanitize_times(t):
            """ if the input is a datetime, turn it into a timestamp,
            otherwise, make sure that it is an intiger"""
            if isinstance(t, datetime):
                return str(int(time.mktime(t.timetuple())))
            else:
                return str(int(t))

        times = map(sanitize_times, times)
        data = ','.join(map(lambda tup: "%s,%s,%s" % tup, zip(meters, readings, times)))
        res = self._request(self.push_readings_url, data)
        assert res.status_code == 200
        return res