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

Get the currency from the api (#56806)

Browse Source
This commit is contained in:
João Pedro Hickmann 2021-09-29 20:25:07 -03:00 committed by GitHub
parent 12b2076351
commit a967a1d1df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
homeassistant/components/growatt_server/sensor.py
View File

@ -5,7 +5,6 @@ from dataclasses import dataclass
import datetime import datetime
import json import json
import logging import logging
import re
import growattServer import growattServer
@ -19,7 +18,6 @@ from homeassistant.const import (
CONF_PASSWORD, CONF_PASSWORD,
CONF_URL, CONF_URL,
CONF_USERNAME, CONF_USERNAME,
CURRENCY_EURO,
DEVICE_CLASS_BATTERY, DEVICE_CLASS_BATTERY,
DEVICE_CLASS_CURRENT, DEVICE_CLASS_CURRENT,
DEVICE_CLASS_ENERGY, DEVICE_CLASS_ENERGY,
@ -57,6 +55,7 @@ class GrowattSensorEntityDescription(SensorEntityDescription, GrowattRequiredKey
"""Describes Growatt sensor entity.""" """Describes Growatt sensor entity."""
precision: int | None = None precision: int | None = None
currency: bool = False
TOTAL_SENSOR_TYPES: tuple[GrowattSensorEntityDescription, ...] = ( TOTAL_SENSOR_TYPES: tuple[GrowattSensorEntityDescription, ...] = (
@ -64,13 +63,13 @@ TOTAL_SENSOR_TYPES: tuple[GrowattSensorEntityDescription, ...] = (
key="total_money_today", key="total_money_today",
name="Total money today", name="Total money today",
api_key="plantMoneyText", api_key="plantMoneyText",
native_unit_of_measurement=CURRENCY_EURO, currency=True,
), ),
GrowattSensorEntityDescription( GrowattSensorEntityDescription(
key="total_money_total", key="total_money_total",
name="Money lifetime", name="Money lifetime",
api_key="totalMoneyText", api_key="totalMoneyText",
native_unit_of_measurement=CURRENCY_EURO, currency=True,
), ),
GrowattSensorEntityDescription( GrowattSensorEntityDescription(
key="total_energy_today", key="total_energy_today",
@ -975,6 +974,13 @@ class GrowattInverter(SensorEntity):
result = round(result, self.entity_description.precision) result = round(result, self.entity_description.precision)
return result return result
@property
def native_unit_of_measurement(self) -> str | None:
"""Return the unit of measurement of the sensor, if any."""
if self.entity_description.currency:
return self.probe.get_data("currency")
return super().native_unit_of_measurement
def update(self): def update(self):
"""Get the latest data from the Growat API and updates the state.""" """Get the latest data from the Growat API and updates the state."""
self.probe.update() self.probe.update()
@ -1003,10 +1009,10 @@ class GrowattData:
if self.growatt_type == "total": if self.growatt_type == "total":
total_info = self.api.plant_info(self.device_id) total_info = self.api.plant_info(self.device_id)
del total_info["deviceList"] del total_info["deviceList"]
# PlantMoneyText comes in as "3.1/€" remove anything that isn't part of the number # PlantMoneyText comes in as "3.1/€" split between value and currency
total_info["plantMoneyText"] = re.sub( plant_money_text, currency = total_info["plantMoneyText"].split("/")
r"[^\d.,]", "", total_info["plantMoneyText"] total_info["plantMoneyText"] = plant_money_text
) total_info["currency"] = currency
self.data = total_info self.data = total_info
elif self.growatt_type == "inverter": elif self.growatt_type == "inverter":
inverter_info = self.api.inverter_detail(self.device_id) inverter_info = self.api.inverter_detail(self.device_id)