dbus-btoutback.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
from outbackbt import OutbackBtDev
from dbus.mainloop.glib import DBusGMainLoop
from threading import Thread
import sys

if sys.version_info.major == 2:
	import gobject
else:
	from gi.repository import GLib as gobject

# Victron packages
# from ve_utils import exit_on_error

from dbushelper import DbusHelper
from utils import *
import utils
from inverter import Inverter
from outbackbt import OutbackBt

logger.info("Starting dbus-btoutback")

def main():

	def get_btaddr() -> list:
		# Get the bluetooth address we need to use from the argument
		if len(sys.argv) > 1:
			return sys.argv[1:]
		else:
			return [utils.OUTBACK_ADDRESS]

	logger.info("dbus-btoutback v" + str(utils.DRIVER_VERSION) + utils.DRIVER_SUBVERSION)

	btaddr = get_btaddr()

	outbackBtDevConnection = OutbackBtDev(btaddr[0])
	outbackInverterObject: Inverter = OutbackBt(outbackBtDevConnection, btaddr[0])
	outbackBtDevConnection.connect()

	if outbackInverterObject is None:
		logger.error("ERROR >>> No Inverter connection at " + str(btaddr))
		sys.exit(1)

	outbackInverterObject.log_settings()

	# Have a mainloop, so we can send/receive asynchronous calls to and from dbus
	DBusGMainLoop(set_as_default=True)
	if sys.version_info.major == 2:
		gobject.threads_init()
	mainloop = gobject.MainLoop()

	# Get the initial values for the battery used by setup_vedbus
	inverterDevice = DbusHelper(outbackInverterObject, 1)

	if not inverterDevice.setup_vedbus():
		logger.error("ERROR >>> Problem with inverter " + str(btaddr))
		sys.exit(1)

	def poll_inverter(loop):
		# Run in separate thread. Pass in the mainloop so the thread can kill us if there is an exception.
		poller = Thread(target=lambda: inverterDevice.publish_inverter(loop))
		# Thread will die with us if deamon
		poller.daemon = True
		poller.start()

		return True

	# Poll the battery at INTERVAL and run the main loop
	gobject.timeout_add(outbackInverterObject.poll_interval, lambda: poll_inverter(mainloop))
	try:
		mainloop.run()
	except KeyboardInterrupt:
		pass


if __name__ == "__main__":
	main()