#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# Copyright (C) 2019  Hauke Petersen <devel@haukepetersen.de>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import sys
import yaml
import json
import logging
import argparse
from datetime import datetime
import urllib.request
from influxdb import InfluxDBClient
from apscheduler.schedulers.blocking import BlockingScheduler


CONFIGFILE = "/opt/steckie/config.yml"
CMD_SENSOR = "cm?cmnd=Status%208"


class Steckie:
    def __init__(self, cfgfile):
        self.cfg = dict()

        # read configuration
        try:
            with open(cfgfile, 'r', encoding="utf-8") as f:
                self.cfg.update(yaml.load(f, Loader=yaml.BaseLoader))
        except:
            logging.error("unable to read config file '{}'".format(cfgfile))
            sys.exit(1)

        self.scheduler = BlockingScheduler()
        self.db = InfluxDBClient(host=self.cfg['db']['host'],
                                 port=int(self.cfg['db']['port']),
                                 database=self.cfg['db']['name'],
                                 username=self.cfg['db']['user'],
                                 password=self.cfg['db']['pass'])

    def run(self):
        self.scheduler.add_job(self.update, 'interval',
                               seconds=int(self.cfg['update_itvl']))
        self.scheduler.start()

    def update(self):
        for dev in self.cfg['devs']:
            url = "{}/{}".format(dev['url'], CMD_SENSOR)
            data = {}
            try:
                with urllib.request.urlopen(url) as resp:
                    data = json.loads(resp.read().decode("utf-8"))
                    print(data)
            except:
                logging.warning("unable to read data from '{}'".format(dev['name']))
                continue

            point = [{
                "measurement": "stromie",
                "tags": {
                    "name": dev['name'],
                    "url": dev['url'],
                    "type": "tasmota_awp07l",
                },
                "time": data['StatusSNS']['Time'],
                "fields": {
                    "voltage": float(data['StatusSNS']['ENERGY']['Voltage']),
                    "current": float(data['StatusSNS']['ENERGY']['Current']),
                    "pwr": float(data['StatusSNS']['ENERGY']['ApparentPower']),
                    "pwr_r": float(data['StatusSNS']['ENERGY']['ReactivePower']),
                    "factor": float(data['StatusSNS']['ENERGY']['Factor']),
                    "e_daily": float(data['StatusSNS']['ENERGY']['Today']),
                }
            }]

            try:
                self.db.write_points(point)
            except:
                logging.warning("{}: unable to write to DB".format(dev['name']))


def main(args):
    logging.basicConfig(format='%(asctime)s %(message)s',
                        level=logging.WARNING)
    steckie = Steckie(args.cfgfile)
    steckie.run()


if __name__ == "__main__":
    p = argparse.ArgumentParser()
    p.add_argument("cfgfile", default=CONFIGFILE, nargs='?', help="output dump")
    args = p.parse_args()
    main(args)