mirror of
https://github.com/skalavala/mysmarthome.git
synced 2025-08-18 11:16:34 +00:00
minor updates
This commit is contained in:
Mahasri Kalavala
267
custom_components/life365/sensor.py
Normal file
267
custom_components/life365/sensor.py
Normal file
@@ -0,0 +1,267 @@
|
||||
"""
|
||||
@ Author : Suresh Kalavala
|
||||
@ Date : 05/24/2017
|
||||
@ Description : Life365 Sensor - It queries Life360 API and retrieves
|
||||
data at a specified interval and dumps into MQTT
|
||||
|
||||
@ Notes: Copy this file and place it in your
|
||||
"Home Assistant Config folder\custom_components\sensor\" folder
|
||||
Copy corresponding Life365 Package frommy repo,
|
||||
and make sure you have MQTT installed and Configured
|
||||
Make sure the life365 password doesn't contain '#' or '$' symbols
|
||||
"""
|
||||
|
||||
from datetime import timedelta
|
||||
import logging
|
||||
import subprocess
|
||||
import json
|
||||
|
||||
import voluptuous as vol
|
||||
import homeassistant.components.mqtt as mqtt
|
||||
|
||||
from io import StringIO
|
||||
from homeassistant.components.mqtt import (CONF_STATE_TOPIC, CONF_COMMAND_TOPIC, CONF_QOS, CONF_RETAIN)
|
||||
from homeassistant.helpers import template
|
||||
from homeassistant.exceptions import TemplateError
|
||||
from homeassistant.components.sensor import PLATFORM_SCHEMA
|
||||
from homeassistant.const import (
|
||||
CONF_NAME, CONF_VALUE_TEMPLATE, CONF_UNIT_OF_MEASUREMENT,
|
||||
STATE_UNKNOWN)
|
||||
from homeassistant.helpers.entity import Entity
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
DEPENDENCIES = ['mqtt']
|
||||
|
||||
DEFAULT_NAME = 'Life365 Sensor'
|
||||
CONST_MQTT_TOPIC = "mqtt_topic"
|
||||
CONST_STATE_ERROR = "error"
|
||||
CONST_STATE_RUNNING = "running"
|
||||
CONST_USERNAME = "username"
|
||||
CONST_PASSWORD = "password"
|
||||
|
||||
COMMAND1 = "curl -s -X POST -H \"Authorization: Basic cFJFcXVnYWJSZXRyZTRFc3RldGhlcnVmcmVQdW1hbUV4dWNyRUh1YzptM2ZydXBSZXRSZXN3ZXJFQ2hBUHJFOTZxYWtFZHI0Vg==\" -F \"grant_type=password\" -F \"username=USERNAME360\" -F \"password=PASSWORD360\" https://api.life360.com/v3/oauth2/token.json | grep -Po '(?<=\"access_token\":\")\\w*'"
|
||||
COMMAND2 = "curl -s -X GET -H \"Authorization: Bearer ACCESS_TOKEN\" https://api.life360.com/v3/circles.json | grep -Po '(?<=\"id\":\")[\\w-]*'"
|
||||
COMMAND3 = "curl -s -X GET -H \"Authorization: Bearer ACCESS_TOKEN\" https://api.life360.com/v3/circles/ID"
|
||||
|
||||
SCAN_INTERVAL = timedelta(seconds=60)
|
||||
|
||||
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
||||
vol.Required(CONST_USERNAME): cv.string,
|
||||
vol.Required(CONST_PASSWORD): cv.string,
|
||||
vol.Required(CONST_MQTT_TOPIC): cv.string,
|
||||
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
|
||||
vol.Optional(CONF_UNIT_OF_MEASUREMENT): cv.string,
|
||||
vol.Optional(CONF_VALUE_TEMPLATE): cv.template,
|
||||
})
|
||||
|
||||
# pylint: disable=unused-argument
|
||||
def setup_platform(hass, config, add_devices, discovery_info=None):
|
||||
"""Set up the Life365 Sensor."""
|
||||
name = config.get(CONF_NAME)
|
||||
username = config.get(CONST_USERNAME)
|
||||
password = config.get(CONST_PASSWORD)
|
||||
mqtt_topic = config.get(CONST_MQTT_TOPIC)
|
||||
|
||||
unit = config.get(CONF_UNIT_OF_MEASUREMENT)
|
||||
value_template = config.get(CONF_VALUE_TEMPLATE)
|
||||
if value_template is not None:
|
||||
value_template.hass = hass
|
||||
|
||||
data = Life365SensorData(username, password, COMMAND1, COMMAND2, COMMAND3, mqtt_topic, hass)
|
||||
|
||||
add_devices([Life365Sensor(hass, data, name, unit, value_template)])
|
||||
|
||||
|
||||
class Life365Sensor(Entity):
|
||||
"""Representation of a sensor."""
|
||||
|
||||
def __init__(self, hass, data, name, unit_of_measurement, value_template):
|
||||
"""Initialize the sensor."""
|
||||
self._hass = hass
|
||||
self.data = data
|
||||
self._name = name
|
||||
self._state = STATE_UNKNOWN
|
||||
self._unit_of_measurement = unit_of_measurement
|
||||
self._value_template = value_template
|
||||
self.update()
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
"""Return the name of the sensor."""
|
||||
return self._name
|
||||
|
||||
@property
|
||||
def unit_of_measurement(self):
|
||||
"""Return the unit the value is expressed in."""
|
||||
return self._unit_of_measurement
|
||||
|
||||
@property
|
||||
def state(self):
|
||||
"""Return the state of the device."""
|
||||
return self._state
|
||||
|
||||
def update(self):
|
||||
"""Get the latest data and updates the state."""
|
||||
self.data.update()
|
||||
value = self.data.value
|
||||
|
||||
if value is None:
|
||||
value = STATE_UNKNOWN
|
||||
elif self._value_template is not None:
|
||||
self._state = self._value_template.render_with_possible_json_value(
|
||||
value, STATE_UNKNOWN)
|
||||
else:
|
||||
self._state = value
|
||||
|
||||
|
||||
class Life365SensorData(object):
|
||||
"""The class for handling the data retrieval."""
|
||||
|
||||
def __init__(self, username, password, command1, command2, command3, mqtt_topic, hass):
|
||||
"""Initialize the data object."""
|
||||
self.username = username
|
||||
self.password = password
|
||||
self.COMMAND_ACCESS_TOKEN = command1
|
||||
self.COMMAND_ID = command2
|
||||
self.COMMAND_MEMBERS = command3
|
||||
self.hass = hass
|
||||
self.value = None
|
||||
self.mqtt_topic = mqtt_topic
|
||||
self.mqtt_retain = True
|
||||
self.mqtt_qos = 0
|
||||
|
||||
def update(self):
|
||||
|
||||
try:
|
||||
""" Prepare and Execute Commands """
|
||||
self.COMMAND_ACCESS_TOKEN = self.COMMAND_ACCESS_TOKEN.replace("USERNAME360", self.username)
|
||||
self.COMMAND_ACCESS_TOKEN = self.COMMAND_ACCESS_TOKEN.replace("PASSWORD360", self.password)
|
||||
access_token = self.exec_shell_command( self.COMMAND_ACCESS_TOKEN )
|
||||
|
||||
if access_token == None:
|
||||
self.value = CONST_STATE_ERROR
|
||||
return None
|
||||
|
||||
self.COMMAND_ID = self.COMMAND_ID.replace("ACCESS_TOKEN", access_token)
|
||||
id = self.exec_shell_command( self.COMMAND_ID )
|
||||
|
||||
if id == None:
|
||||
self.value = CONST_STATE_ERROR
|
||||
return None
|
||||
|
||||
self.COMMAND_MEMBERS = self.COMMAND_MEMBERS.replace("ACCESS_TOKEN", access_token)
|
||||
self.COMMAND_MEMBERS = self.COMMAND_MEMBERS.replace("ID", id)
|
||||
payload = self.exec_shell_command( self.COMMAND_MEMBERS )
|
||||
|
||||
if payload != None:
|
||||
self.save_payload_to_mqtt ( self.mqtt_topic, payload )
|
||||
data = json.loads ( payload )
|
||||
for member in data["members"]:
|
||||
topic = StringBuilder()
|
||||
topic.Append("owntracks/")
|
||||
topic.Append(member["firstName"].lower())
|
||||
topic.Append("/")
|
||||
topic.Append(member["firstName"].lower())
|
||||
topic = topic
|
||||
|
||||
msgPayload = StringBuilder()
|
||||
msgPayload.Append("{")
|
||||
msgPayload.Append("\"t\":\"p\"")
|
||||
msgPayload.Append(",")
|
||||
|
||||
msgPayload.Append("\"tst\":")
|
||||
msgPayload.Append(member['location']['timestamp'])
|
||||
msgPayload.Append(",")
|
||||
|
||||
msgPayload.Append("\"acc\":")
|
||||
msgPayload.Append(member['location']['accuracy'])
|
||||
msgPayload.Append(",")
|
||||
|
||||
msgPayload.Append("\"_type\":\"location\"")
|
||||
msgPayload.Append(",")
|
||||
|
||||
msgPayload.Append("\"alt\":\"0\"")
|
||||
msgPayload.Append(",")
|
||||
|
||||
msgPayload.Append("\"_cp\":\"false\"")
|
||||
msgPayload.Append(",")
|
||||
|
||||
msgPayload.Append("\"lon\":")
|
||||
msgPayload.Append(member['location']['longitude'])
|
||||
msgPayload.Append(",")
|
||||
|
||||
msgPayload.Append("\"lat\":")
|
||||
msgPayload.Append(member['location']['latitude'])
|
||||
msgPayload.Append(",")
|
||||
|
||||
msgPayload.Append("\"batt\":")
|
||||
msgPayload.Append(member['location']['battery'])
|
||||
msgPayload.Append(",")
|
||||
|
||||
if str(member['location']['wifiState']) == "1":
|
||||
msgPayload.Append("\"conn\":\"w\"")
|
||||
msgPayload.Append(",")
|
||||
|
||||
msgPayload.Append("\"vel\":")
|
||||
msgPayload.Append(str(member['location']['speed']))
|
||||
msgPayload.Append(",")
|
||||
|
||||
msgPayload.Append("\"charging\":")
|
||||
msgPayload.Append(member['location']['charge'])
|
||||
msgPayload.Append("}")
|
||||
|
||||
self.save_payload_to_mqtt ( str(topic), str(msgPayload) )
|
||||
self.value = CONST_STATE_RUNNING
|
||||
else:
|
||||
self.value = CONST_STATE_ERROR
|
||||
|
||||
except Exception as e:
|
||||
self.value = CONST_STATE_ERROR
|
||||
|
||||
def exec_shell_command( self, command ):
|
||||
|
||||
output = None
|
||||
try:
|
||||
output = subprocess.check_output( command, shell=True, timeout=50 )
|
||||
output = output.strip().decode('utf-8')
|
||||
|
||||
except subprocess.CalledProcessError:
|
||||
""" _LOGGER.error("Command failed: %s", command)"""
|
||||
self.value = CONST_STATE_ERROR
|
||||
output = None
|
||||
except subprocess.TimeoutExpired:
|
||||
""" _LOGGER.error("Timeout for command: %s", command)"""
|
||||
self.value = CONST_STATE_ERROR
|
||||
output = None
|
||||
|
||||
if output == None:
|
||||
_LOGGER.error( "Life365 has not responsed well. Nothing to worry, will try again!" )
|
||||
self.value = CONST_STATE_ERROR
|
||||
return None
|
||||
else:
|
||||
return output
|
||||
|
||||
def save_payload_to_mqtt( self, topic, payload ):
|
||||
|
||||
try:
|
||||
"""mqtt.async_publish ( self.hass, topic, payload, self.mqtt_qos, self.mqtt_retain )"""
|
||||
_LOGGER.info("topic: %s", topic)
|
||||
_LOGGER.info("payload: %s", payload)
|
||||
mqtt.publish ( self.hass, topic, payload, self.mqtt_qos, self.mqtt_retain )
|
||||
|
||||
except:
|
||||
_LOGGER.error( "Error saving Life365 data to mqtt." )
|
||||
|
||||
class StringBuilder:
|
||||
_file_str = None
|
||||
|
||||
def __init__(self):
|
||||
self._file_str = StringIO()
|
||||
|
||||
def Append(self, str):
|
||||
self._file_str.write(str)
|
||||
|
||||
def __str__(self):
|
||||
return self._file_str.getvalue()
|
||||
Reference in New Issue
Block a user