#!/usr/bin/env python3
# -*- coding: utf_8 -*-

from sdm630 import *

import sys
import serial
import time
import paho.mqtt.client as mqtt
import logging
import os
import traceback
import configparser as ConfigParser
import time

CONFIG_FILE = '/usr/local/bin/sdm630_mqtt/sdm630-mqtt.conf'

logging.getLogger().setLevel(logging.WARNING)

# time.sleep(20)

def publish(topic,value):
        info = mqclient.publish("/".join((config.get("mqtt","topic_prefix"),topic)),value,qos=1)
        info.wait_for_publish()


if ((len(sys.argv) == 3) and sys.argv[1] == '-c'):
        CONFIG_FILE = sys.argv[2]
logging.info("Reading config...")
config = ConfigParser.ConfigParser()
confread = config.read(CONFIG_FILE)
logging.info("Read config {}".format(confread))


logging.info("Connecting to MQTT server...")
mqclient = mqtt.Client()
try:
        mqclient.connect(config.get("mqtt","server"),
                config.getint("mqtt","port"),
                config.getint("mqtt","keepalive"))
        mqclient.loop_start()
except:
        traceback.print_exc()
        logging.error("Cannot connect to MQTT server")
        sys.exit(1)

logging.info("Setup...")
num_meters = config.getint("sdm630","num_meters")
meters = []
#connection_type = config.get("sdm630", "connection_type")

#for i in range(num_meters):
#       if connection_type == 'tcp':
#               logging.info("Opening port {}".format(config.get("tcp","port")))
#               meter = SDM630TCP(host=config.get("tcp","host"),
#                       port=config.get("tcp","port"),
#                       aid=config.getint("sdm630","id"+str(i+1)),
#                       regfile=config.get("sdm630", "regfile"))
#       elif connection_type == 'rs485':
logging.info("Opening port {}".format(config.get("rs485","port")))
meter = SDM630RS485(config.get("rs485","PORT"),
config.getint("rs485","BAUDRATE"),
config.getint("sdm630","id"+str(+1)),
regfile=config.get("sdm630", "regfile"))
meters.append(meter)

logging.info("Entering endless loop")
while (True):
        for (num,sdm630) in enumerate(meters):
                num += 1
                publish(str(num)+"/voltage/all",str(sdm630.voltx3))
                publish(str(num)+"/current/all",str(sdm630.ampx3))
                publish(str(num)+"/power/all",str(sdm630.powx3))
                publish(str(num)+"/voltage/l1",sdm630.v1)
                publish(str(num)+"/voltage/l2",sdm630.v2)
                publish(str(num)+"/voltage/l3",sdm630.v3)
                publish(str(num)+"/current/l1",sdm630.a1)
                publish(str(num)+"/current/l2",sdm630.a2)
                publish(str(num)+"/current/l3",sdm630.a3)
                publish(str(num)+"/power/l1",sdm630.p1)
                publish(str(num)+"/power/l2",sdm630.p2)
                publish(str(num)+"/power/l3",sdm630.p3)
                publish(str(num)+"/power/total",sdm630.total_power)
                publish(str(num)+"/voltamps/l1",sdm630.va1)
                publish(str(num)+"/voltamps/l2",sdm630.va2)
                publish(str(num)+"/voltamps/l3",sdm630.va3)
                publish(str(num)+"/voltage/avg",sdm630.avg_voltage)
                publish(str(num)+"/current/avg",sdm630.avg_current)
                publish(str(num)+"/current/sum",sdm630.sum_current)
                publish(str(num)+"/frequency",sdm630.frequency)
                publish(str(num)+"/voltage/l1l2",sdm630.v1v2)
                publish(str(num)+"/voltage/l2l3",sdm630.v2v3)
                publish(str(num)+"/voltage/l3l1",sdm630.v3v1)
                publish(str(num)+"/curent/neutral",sdm630.neutral_current)
                publish(str(num)+"/wh/import",sdm630.import_wh)
                publish(str(num)+"/wh/export",sdm630.export_wh)
                publish(str(num)+"/time",time.time())
        time.sleep(1)