jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-21 04:07:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add android option for linux_battery.py (#11852)

Browse Source 
* Add android option for linux_battery.py

* Add system option for linux_battery.py, support android.
This commit is contained in:
lich 2018-01-24 20:51:39 +08:00 committed by Fabian Affolter
parent a0a001db71
commit becd94fe2f
1 changed files with 41 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
homeassistant/components/sensor/linux_battery.py
View File

@ -35,17 +35,25 @@ ATTR_STATUS = 'status'
ATTR_VOLTAGE_MIN_DESIGN = 'voltage_min_design' ATTR_VOLTAGE_MIN_DESIGN = 'voltage_min_design'
ATTR_VOLTAGE_NOW = 'voltage_now' ATTR_VOLTAGE_NOW = 'voltage_now'
ATTR_HEALTH = 'health'
ATTR_STATUS = 'status'
CONF_BATTERY = 'battery' CONF_BATTERY = 'battery'
CONF_SYSTEM = 'system'
DEFAULT_BATTERY = 1 DEFAULT_BATTERY = 1
DEFAULT_NAME = 'Battery' DEFAULT_NAME = 'Battery'
DEFAULT_PATH = '/sys/class/power_supply' DEFAULT_PATH = '/sys/class/power_supply'
DEFAULT_SYSTEM = 'linux'
SYSTEMS = ['android', 'linux']
ICON = 'mdi:battery' ICON = 'mdi:battery'
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Optional(CONF_BATTERY, default=DEFAULT_BATTERY): cv.positive_int, vol.Optional(CONF_BATTERY, default=DEFAULT_BATTERY): cv.positive_int,
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string, vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
vol.Optional(CONF_SYSTEM, default=DEFAULT_SYSTEM): vol.In(SYSTEMS),
}) })
@ -53,20 +61,24 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
"""Set up the Linux Battery sensor.""" """Set up the Linux Battery sensor."""
name = config.get(CONF_NAME) name = config.get(CONF_NAME)
battery_id = config.get(CONF_BATTERY) battery_id = config.get(CONF_BATTERY)
system = config.get(CONF_SYSTEM)
try: try:
os.listdir(os.path.join(DEFAULT_PATH, 'BAT{}'.format(battery_id))) if system == 'android':
os.listdir(os.path.join(DEFAULT_PATH, 'battery'))
else:
os.listdir(os.path.join(DEFAULT_PATH, 'BAT{}'.format(battery_id)))
except FileNotFoundError: except FileNotFoundError:
_LOGGER.error("No battery found") _LOGGER.error("No battery found")
return False return False
add_devices([LinuxBatterySensor(name, battery_id)], True) add_devices([LinuxBatterySensor(name, battery_id, system)], True)
class LinuxBatterySensor(Entity): class LinuxBatterySensor(Entity):
"""Representation of a Linux Battery sensor.""" """Representation of a Linux Battery sensor."""
def __init__(self, name, battery_id): def __init__(self, name, battery_id, system):
"""Initialize the battery sensor.""" """Initialize the battery sensor."""
import batinfo import batinfo
self._battery = batinfo.Batteries() self._battery = batinfo.Batteries()
@ -74,6 +86,7 @@ class LinuxBatterySensor(Entity):
self._name = name self._name = name
self._battery_stat = None self._battery_stat = None
self._battery_id = battery_id - 1 self._battery_id = battery_id - 1
self._system = system
self._unit_of_measurement = '%' self._unit_of_measurement = '%'
@property @property
@ -99,23 +112,31 @@ class LinuxBatterySensor(Entity):
@property @property
def device_state_attributes(self): def device_state_attributes(self):
"""Return the state attributes of the sensor.""" """Return the state attributes of the sensor."""
return { if self._system == 'android':
ATTR_NAME: self._battery_stat.name, return {
ATTR_PATH: self._battery_stat.path, ATTR_NAME: self._battery_stat.name,
ATTR_ALARM: self._battery_stat.alarm, ATTR_PATH: self._battery_stat.path,
ATTR_CAPACITY_LEVEL: self._battery_stat.capacity_level, ATTR_HEALTH: self._battery_stat.health,
ATTR_CYCLE_COUNT: self._battery_stat.cycle_count, ATTR_STATUS: self._battery_stat.status,
ATTR_ENERGY_FULL: self._battery_stat.energy_full, }
ATTR_ENERGY_FULL_DESIGN: self._battery_stat.energy_full_design, else:
ATTR_ENERGY_NOW: self._battery_stat.energy_now, return {
ATTR_MANUFACTURER: self._battery_stat.manufacturer, ATTR_NAME: self._battery_stat.name,
ATTR_MODEL_NAME: self._battery_stat.model_name, ATTR_PATH: self._battery_stat.path,
ATTR_POWER_NOW: self._battery_stat.power_now, ATTR_ALARM: self._battery_stat.alarm,
ATTR_SERIAL_NUMBER: self._battery_stat.serial_number, ATTR_CAPACITY_LEVEL: self._battery_stat.capacity_level,
ATTR_STATUS: self._battery_stat.status, ATTR_CYCLE_COUNT: self._battery_stat.cycle_count,
ATTR_VOLTAGE_MIN_DESIGN: self._battery_stat.voltage_min_design, ATTR_ENERGY_FULL: self._battery_stat.energy_full,
ATTR_VOLTAGE_NOW: self._battery_stat.voltage_now, ATTR_ENERGY_FULL_DESIGN: self._battery_stat.energy_full_design,
} ATTR_ENERGY_NOW: self._battery_stat.energy_now,
ATTR_MANUFACTURER: self._battery_stat.manufacturer,
ATTR_MODEL_NAME: self._battery_stat.model_name,
ATTR_POWER_NOW: self._battery_stat.power_now,
ATTR_SERIAL_NUMBER: self._battery_stat.serial_number,
ATTR_STATUS: self._battery_stat.status,
ATTR_VOLTAGE_MIN_DESIGN: self._battery_stat.voltage_min_design,
ATTR_VOLTAGE_NOW: self._battery_stat.voltage_now,
}
def update(self): def update(self):
"""Get the latest data and updates the states.""" """Get the latest data and updates the states."""