rasolar/logger.py

#!/usr/bin/env python

import numpy as np
import os, serial,time,socket,sys,json,logging,requests,getopt
# import the server implementation
# import ADC
import vedirect
#import upload_osm

configfile="config.json"
cf=open(configfile,"r")
log_conf=json.load(cf)
cf.close()
channel_names=["time","CPU_temp"]
devicename=socket.gethostname()
if "device" in log_conf:
  devicename=log_conf['device']
mean_count=5
if "mean_count" in log_conf:
  mean_count=int(log_conf['mean_count'])

a = 2

def upload_osm(sensebox_id,sensor_id,value):
	url="https://ingress.opensensemap.org/boxes/%s/%s" % (sensebox_id,sensor_id)
	r = requests.post(url,json={'value': value})
	if r.status_code != requests.codes.ok:
		print("Error %d: %s" % (r.status_code,r.text))

# push options to internet
bosm=False
if "opensensemap" in log_conf:
  bosm=True
  conf_osm=log_conf['opensensemap']
  if "enable" in conf_osm:
    if conf_osm['enable'] == 0:
      bosm = False
  if bosm:
    try:
      push_count=conf_osm['push_count']
    except:
      push_count=20 # wait 5 cycles till upload to opensensemap
    push_counter=0
    sensebox_id=conf_osm['sensebox_id'] # id of opensensemap
    push_vars=conf_osm['sensors'].keys()
    push_data={}
    push_mean_counts={}
    sensebox_sid={}
    for i in push_vars:
      push_data[i]=0
      push_mean_counts[i]=0
      sensebox_sid[i]=conf_osm['sensors'][i]
#luftdaten_id=56009074600018881

# host and port of internal logging server
bsql=False
if "sqlserver" in log_conf:
  bsql=True
  if "enable" in log_conf['sqlserver']:
    if log_conf['sqlserver']['enable'] == 0:
      bsql=False
  if bsql:
    sqlhost=log_conf['sqlserver']['host']
    sqlport=log_conf['sqlserver']['port']

# config of lux sensor tls2591
btls=False
if "tsl2591" in log_conf:
  btls=True
  if "enable" in log_conf['tsl2591']:
    if log_conf['tsl2591']['enable'] == 0:
      btls=False
  if btls:
    import tsl2591
    tsl_port=1
    if "port" in log_conf['tsl2591']:
      tsl_port=int(log_conf['tsl2591']['port'])
    tsl_add=0x29
    if "address" in log_conf['tsl2591']:
      tsl_add=int(log_conf['tsl2591']['address'],16)
    try:
      tsl = tsl2591.Tsl2591(i2c_bus=tsl_port,sensor_address=tsl_add)  # initialize
    except:
      btls=False
    else:
      channel_names.append("lux")

# config of bme280 sensor
# use pip install RPi.bme280 for library
bbme=False
if "bme280" in log_conf:
  bbme=True
  if "enable" in log_conf['bme280']:
    if log_conf['bme280']['enable'] == 0:
      bbme=False
  if bbme:
    import smbus2
    import bme280
    if "port" in log_conf['bme280']:
      bme_port=log_conf['bme280']['port']
    else:
      bme_port=1
    if "address" in log_conf['bme280']:
      bme_add=int(log_conf['bme280']['address'],16)
    else:
      bme_add=0x77
    try:
      bme_bus=smbus2.SMBus(bme_port)
    except:
      bbme=False
    else:
      calibration_params=bme280.load_calibration_params(bme_bus,bme_add)
      channel_names.append("temperature")
      channel_names.append("humidity")
      channel_names.append("pressure")


# configure the client logging
logging.basicConfig()
log = logging.getLogger('./modbus.error')
log.setLevel(logging.ERROR)

# configure tristar
btristar=False
if "tristar" in log_conf:
  btristar=True
  if "enable" in log_conf['tristar']:
    if log_conf['tristar']['enable'] == 0:
      btristar=False
  if btristar:
    import smbus
    from pymodbus.client.sync import ModbusSerialClient as ModbusClient
    tri_port='/dev/ttyUSB0'
    if "port" in log_conf['tristar']:
      tri_port=log_conf['tristar']['port']
    tri_baud=9600
    if "baud" in log_conf['tristar']:
      tri_baud=log_conf['tristar']['baud']
    tri_timeout=1
    if "timeout" in log_conf['tristar']:
      tri_timeout=log_conf['tristar']['timeout']
    try:
      triclient = ModbusClient(method='rtu', port=tri_port, baudrate=tri_baud, timeout=tri_timeout)
    except:
      btristar=False
    else:
      triclient.connect()
      for i in ["volt_scale","amp_scale","volt_bat_term","volt_bat_sens","volt_arr","amp_bat","amp_arr","temp_heatsink","temp_bat","ah_res","ah_tot","kwh_res","kwh_tot","watt_in","watt_out","hour_tot","state","volt_sweep_mp","volt_sweep_oc"]:
        channel_names.append(i)
      
# declare ve mppt
bve=False
if "vedirect" in log_conf:
  bve=True
  if "enable" in log_conf['vedirect']:
    if log_conf['vedirect']['enable'] == 0:
      bve=False
  if "port" in log_conf['vedirect']:
    bve_port=log_conf['vedirect']['port']
  else:
    bve_port='/dev/serial/by-id/usb-VictronEnergy_BV_VE_Direct_cable_VE1SSBVT-if00-port0'
  try:
    ve=vedirect(bve_port,60)
  except:
    bve=False
  else:
    for i in ["volt_bat_ve","volt_arr_ve","amp_ve","watt_ve","days_ve"]:
      channel_names.append(i)

# declare adc
badc=False
if "ads1x15" in log_conf:
  badc,adc = ads1x15.init(log_conf['ads1x15'])

bmcp9808=False
if "mcp9808" in log_conf:
  bmcp9808=True
  if "enable" in log_conf['mcp9808']:
    if log_conf['mcp9808']['enable'] == 0:
      bmcp9808=False
  if bmcp9808:
    import Adafruit_MCP9808.MCP9808 as MCP9808
    try:
      sens_9808 = MCP9808.MCP9808()
    except:
      bmcp9808 = False
    else:
      sens_9808.begin()
      channel_names.append("temp_9808")


ch_val=np.zeros(len(channel_names))
for i in range(len(channel_names)):
	ch_val[i]=(-1)

while a > 1:
  # copy channel values to backup
  ch_old=ch_val.copy()
  
  ch_val=np.zeros(len(channel_names))
  ch_mean=ch_val.copy()
    
  # set actual time
  timestamp=int(1000*time.time())
  ch_val[channel_names.index("time")]=timestamp
  
  for n in range(mean_count):
  # get cpu temperature
    ch_val[channel_names.index("CPU_temp")] = int(open('/sys/class/thermal/thermal_zone0/temp').read())
	  
  # get mcp9808 temperatur
    if bmcp9808:
      ch_val[channel_names.index("temp_9808")] = int(1000*sens_9808.readTempC())
  # get tls lux
    if btls:
      tsl_full,tsl_ir=tsl.get_full_luminosity()
      ch_val[channel_names.index("lux")] = int(1000*tsl.calculate_lux(tsl_full,tsl_ir))

# read adc's
    if badc:
      for i in adc:
        cname="a"+str(i)+"_{0}"
        for j in range(4):
          ch_val[channel_names.index(cname.format(j))]=adc[i].read_adc(j,gain=GAIN)
  
    if btristar:
      try:
        rr = triclient.read_holding_registers(0,90,unit=1)
      except:
        print("could not get data from tristar")
      else:
        try:
          ch_val[channel_names.index("volt_scale")]=rr.registers[0]*65536+rr.registers[1]
        except:
          print("could not read from tristar")
        else:
          ch_val[channel_names.index("amp_scale")]= rr.registers[2]*65536+rr.registers[3]
          ch_val[channel_names.index("volt_bat_term")]=rr.registers[25]
          ch_val[channel_names.index("volt_bat_sens")]=rr.registers[26]
          ch_val[channel_names.index("volt_arr")]=rr.registers[27] # Array voltage
          ch_val[channel_names.index("amp_bat")]=rr.registers[28] # Battery current
          ch_val[channel_names.index("amp_arr")]=rr.registers[29] # Array current
          ch_val[channel_names.index("temp_heatsink")]=rr.registers[35] # Temperature heatsink
          ch_val[channel_names.index("temp_bat")]=rr.registers[36] # Temperature battery
          ch_val[channel_names.index("ah_res")]=rr.registers[52] * 65536 + rr.registers[53] # Ah resetable
          ch_val[channel_names.index("ah_tot")]=rr.registers[54] * 65536 + rr.registers[55] # Ah total
          ch_val[channel_names.index("kwh_res")]=rr.registers[56] # kwh resetable
          ch_val[channel_names.index("kwh_tot")]=rr.registers[57] # kwh total
          ch_val[channel_names.index("watt_in")]=rr.registers[58] # Power in
          ch_val[channel_names.index("watt_out")]=rr.registers[59] # Power out
          ch_val[channel_names.index("hour_tot")]=rr.registers[42] * 65536 + rr.registers[43] # hour total
          ch_val[channel_names.index("state")]=rr.registers[50] # State
          ch_val[channel_names.index("volt_sweep_mp")]=rr.registers[61] # Array voltage
          ch_val[channel_names.index("volt_sweep_oc")]=rr.registers[62] # Array voltage
  
  # read bme280 (temperature, humidity, pressure)
    if bbme:
      bme_data=bme280.sample(bme_bus,bme_add,calibration_params)
      ch_val[channel_names.index("temperature")]=int(1000*bme_data.temperature) # Temperature
      ch_val[channel_names.index("pressure")]=int(1000*bme_data.pressure) # Pressure
      ch_val[channel_names.index("humidity")]=int(1000*bme_data.humidity) # Humidity
  
  # read ve data
    if bve:
      try:
        vedata=ve.read_data_single()
      except:
        print("could not read VE")
      else:
        ch_val[channel_names.index("volt_bat_ve")]=int(vedata['V']) # Battery voltage measured by ve
        ch_val[channel_names.index("volt_arr_ve")]=int(vedata['VPV']) # Array voltage measured by ve
        ch_val[channel_names.index("amp_ve")]=int(vedata['I']) # loading current by ve
        ch_val[channel_names.index("watt_ve")]=int(vedata['PPV']) # Array power measured by ve
        ch_val[channel_names.index("days_ve")]=int(vedata['HSDS']) # total days online ve
    for i in range(len(ch_val)):
      ch_mean[i]=ch_mean[i]+ch_val[i]
  
  for i in range(len(ch_val)):
    ch_val[i]=int(ch_mean[i]/mean_count)
  
  timefile=round(timestamp/3600000)
  f1=open("/home/pi/log/data_{:d}.txt".format(int(timefile)),"a")
  payload={}
  for i in range(len(ch_val)):
    if ch_val[i] != ch_old[i]:
      f1.write(channel_names[i] + ":{0};".format(int(ch_val[i])))
      if channel_names[i] != "time":
        payload[channel_names[i]]=int(ch_val[i])
    if bosm:
      if channel_names[i] in push_vars:
        if push_counter == push_count:
          sense_data=push_data[channel_names[i]]/(1000*push_mean_counts[channel_names[i]])
#          upload_osm.upload_osm(sensebox_id,sensebox_sid[channel_names[i]],round(sense_data,2))
          upload_osm(sensebox_id,sensebox_sid[channel_names[i]],round(sense_data,2))
          push_data[channel_names[i]]=0
          push_mean_counts[channel_names[i]]=0
        else:
          push_data[channel_names[i]]=push_data[channel_names[i]]+ch_val[i]
          push_mean_counts[channel_names[i]]=push_mean_counts[channel_names[i]]+1
  f1.write("\n")
  f1.close()
  if bsql:
    json_out={"time":  ch_val[channel_names.index("time")],"device": devicename,"payload":payload}
    try:
      s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    except:
      print("{}: could not connect to database".format(time.time()))
    else:
      s.connect((sqlhost, sqlport))
      s.sendall(json.dumps(json_out))
      s.close()

#  volt_batt=rr.registers[24] * volt_scaling / 65536 / 32768
#  volt_batt_t=rr.registers[25] * volt_scaling / 65536 / 32768
#  volt_batt_sens=rr.registers[26] * volt_scaling / 65536 / 32768
#  volt_arr=rr.registers[27] * volt_scaling / 65536 / 32768
#  curr_batt=rr.registers[28] * amp_scaling / 65536 / 32768
#  curr_arr=rr.registers[29] * amp_scaling / 65536 / 32768
#  temp_heatsink=rr.registers[35]
#  temp_batt=rr.registers[36]
#  ah_reset = rr.registers[52] * 65536 + rr.registers[53] 
#  ah_total = rr.registers[54] * 65536 + rr.registers[55] 
#  kwh_reset = rr.registers[56]
#  kwh_total = rr.registers[57]
#  power_in = rr.registers[58] * volt_scaling * amp_scaling / 131072 / 65536 / 65536 
#  power_out = rr.registers[59] * volt_scaling * amp_scaling / 131072 / 65536 / 65536
#  hourm = rr.registers[42]*65536+rr.registers[43]
#  charge_state = rr.registers[50]
  if bosm:
    if push_counter>=push_count:
      push_counter = 1
    else:
      push_counter=push_counter+1
  
  time.sleep(5)


# close the client
client.close()
f1.close()

print("done")