Mercurial > templog
view py/uploader.py @ 581:b4d6f1ae0b9e
make overshoot_factor floating point
add a proper error message for noparamsyet
author | Matt Johnston <matt@ucc.asn.au> |
---|---|
date | Sun, 29 Nov 2015 09:30:36 +0800 |
parents | c5f8375b12a2 |
children | 78c542f03030 d15dda1b1f76 |
line wrap: on
line source
import json import hmac import hashlib import zlib import binascii import logging import asyncio import aiohttp import config from utils import L,D,EX,W,E import utils class Uploader(object): def __init__(self, server): self.server = server @asyncio.coroutine def run(self): # wait for the first read yield from asyncio.sleep(5) while True: yield from self.do() yield from asyncio.sleep(config.UPLOAD_SLEEP) def get_tosend(self, readings): tosend = {} tosend['fridge_on'] = self.server.fridge.is_on() tosend['now'] = self.server.now() tosend['readings'] = readings tosend['wort_name'] = self.server.wort_name tosend['fridge_name'] = self.server.wort_name tosend['current_params'] = dict(self.server.params) tosend['current_params_epoch'] = self.server.params.get_epoch() tosend['start_time'] = self.server.start_time tosend['uptime'] = utils.uptime() return tosend @asyncio.coroutine def send(self, tosend): js = json.dumps(tosend) if self.server.test_mode(): D("Would upload %s to %s" % (js, config.UPDATE_URL)) return js_enc = binascii.b2a_base64(zlib.compress(js.encode())).strip() mac = hmac.new(config.HMAC_KEY.encode(), js_enc, hashlib.sha256).hexdigest() send_data = {'data': js_enc.decode(), 'hmac': mac} r = yield from asyncio.wait_for(aiohttp.request('post', config.UPDATE_URL, data=send_data), 60) result = yield from asyncio.wait_for(r.text(), 60) if r.status == 200 and result != 'OK': raise Exception("Server returned %s" % result) @asyncio.coroutine def do(self): try: readings = self.server.take_readings() tosend = self.get_tosend(readings) D("tosend >>>%s<<<" % str(tosend)) nreadings = len(readings) yield from self.send(tosend) readings = None D("Sent updated %d readings" % nreadings) except Exception as e: EX("Error in uploader: %s" % str(e)) finally: if readings is not None: self.server.pushfront(readings)