Skip to content

Commit 05adc38

Browse files
authored
Use a custom logger in the data sourcing actor (#769)
2 parents 5183476 + 07e1379 commit 05adc38

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

src/frequenz/sdk/actor/_data_sourcing/microgrid_api_source.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@
2424
from .._channel_registry import ChannelRegistry
2525
from ._component_metric_request import ComponentMetricRequest
2626

27+
_logger = logging.getLogger(__name__)
28+
2729
_MeterDataMethods: dict[ComponentMetricId, Callable[[MeterData], float]] = {
2830
ComponentMetricId.ACTIVE_POWER: lambda msg: msg.active_power,
2931
ComponentMetricId.CURRENT_PHASE_1: lambda msg: msg.current_per_phase[0],
@@ -156,7 +158,7 @@ async def _check_battery_request(
156158
for metric in requests:
157159
if metric not in _BatteryDataMethods:
158160
err = f"Unknown metric {metric} for Battery id {comp_id}"
159-
logging.error(err)
161+
_logger.error(err)
160162
raise ValueError(err)
161163
if comp_id not in self.comp_data_receivers:
162164
self.comp_data_receivers[
@@ -181,7 +183,7 @@ async def _check_ev_charger_request(
181183
for metric in requests:
182184
if metric not in _EVChargerDataMethods:
183185
err = f"Unknown metric {metric} for EvCharger id {comp_id}"
184-
logging.error(err)
186+
_logger.error(err)
185187
raise ValueError(err)
186188
if comp_id not in self.comp_data_receivers:
187189
self.comp_data_receivers[
@@ -206,7 +208,7 @@ async def _check_inverter_request(
206208
for metric in requests:
207209
if metric not in _InverterDataMethods:
208210
err = f"Unknown metric {metric} for Inverter id {comp_id}"
209-
logging.error(err)
211+
_logger.error(err)
210212
raise ValueError(err)
211213
if comp_id not in self.comp_data_receivers:
212214
self.comp_data_receivers[
@@ -231,7 +233,7 @@ async def _check_meter_request(
231233
for metric in requests:
232234
if metric not in _MeterDataMethods:
233235
err = f"Unknown metric {metric} for Meter id {comp_id}"
234-
logging.error(err)
236+
_logger.error(err)
235237
raise ValueError(err)
236238
if comp_id not in self.comp_data_receivers:
237239
self.comp_data_receivers[
@@ -269,7 +271,7 @@ async def _check_requested_component_and_metrics(
269271
await self._check_meter_request(comp_id, requests)
270272
else:
271273
err = f"Unknown component category {category}"
272-
logging.error(err)
274+
_logger.error(err)
273275
raise ValueError(err)
274276

275277
def _get_data_extraction_method(
@@ -297,7 +299,7 @@ def _get_data_extraction_method(
297299
if category == ComponentCategory.EV_CHARGER:
298300
return _EVChargerDataMethods[metric]
299301
err = f"Unknown component category {category}"
300-
logging.error(err)
302+
_logger.error(err)
301303
raise ValueError(err)
302304

303305
def _get_metric_senders(
@@ -411,7 +413,7 @@ async def add_metric(self, request: ComponentMetricRequest) -> None:
411413
category = await self._get_component_category(comp_id)
412414

413415
if category is None:
414-
logging.error("Unknown component ID: %d in request %s", comp_id, request)
416+
_logger.error("Unknown component ID: %d in request %s", comp_id, request)
415417
return
416418

417419
self._req_streaming_metrics.setdefault(comp_id, {}).setdefault(

0 commit comments

Comments
 (0)