Replace pytz with timezone module

This is done for handling UTC timezone without importing
the module pytz which is currently installed through pandas.

Signed-off-by: Daniel Zullo <[email protected]>

Author: daniel-zullo-frequenz

src/frequenz/sdk/microgrid/component/_component_data.py

@@ -6,13 +6,12 @@
 
 from abc import ABC, abstractmethod
 from dataclasses import dataclass, field
-from datetime import datetime
+from datetime import datetime, timezone
 from typing import List, Optional, Tuple
 
 import frequenz.api.microgrid.battery_pb2 as battery_pb
 import frequenz.api.microgrid.inverter_pb2 as inverter_pb
 import frequenz.api.microgrid.microgrid_pb2 as microgrid_pb
-import pytz
 
 from ._component_states import EVChargerCableState, EVChargerComponentState
 
@@ -93,7 +92,7 @@ def from_proto(cls, raw: microgrid_pb.ComponentData) -> MeterData:
         """
         meter_data = cls(
             component_id=raw.id,
-            timestamp=raw.ts.ToDatetime(tzinfo=pytz.UTC),
+            timestamp=raw.ts.ToDatetime(tzinfo=timezone.utc),
             active_power=raw.meter.data.ac.power_active.value,
             current_per_phase=(
                 raw.meter.data.ac.phase_1.current.value,
@@ -166,7 +165,7 @@ def from_proto(cls, raw: microgrid_pb.ComponentData) -> BatteryData:
         """
         battery_data = cls(
             component_id=raw.id,
-            timestamp=raw.ts.ToDatetime(tzinfo=pytz.UTC),
+            timestamp=raw.ts.ToDatetime(tzinfo=timezone.utc),
             soc=raw.battery.data.soc.avg,
             soc_lower_bound=raw.battery.data.soc.system_bounds.lower,
             soc_upper_bound=raw.battery.data.soc.system_bounds.upper,
@@ -221,7 +220,7 @@ def from_proto(cls, raw: microgrid_pb.ComponentData) -> InverterData:
         """
         inverter_data = cls(
             component_id=raw.id,
-            timestamp=raw.ts.ToDatetime(tzinfo=pytz.UTC),
+            timestamp=raw.ts.ToDatetime(tzinfo=timezone.utc),
             active_power=raw.inverter.data.ac.power_active.value,
             active_power_lower_bound=raw.inverter.data.ac.power_active.system_bounds.lower,
             active_power_upper_bound=raw.inverter.data.ac.power_active.system_bounds.upper,
@@ -272,7 +271,7 @@ def from_proto(cls, raw: microgrid_pb.ComponentData) -> EVChargerData:
         """
         ev_charger_data = cls(
             component_id=raw.id,
-            timestamp=raw.ts.ToDatetime(tzinfo=pytz.UTC),
+            timestamp=raw.ts.ToDatetime(tzinfo=timezone.utc),
             active_power=raw.ev_charger.data.ac.power_active.value,
             current_per_phase=(
                 raw.ev_charger.data.ac.phase_1.current.value,

tests/actor/test_battery_status.py

@@ -53,7 +53,7 @@ def battery_data(
     Args:
         component_id: component id
         timestamp: Timestamp of the component message.
-            Defaults to datetime.now(tz=pytz.utc).
+            Defaults to datetime.now(tz=timezone.utc).
         relay_state: Battery relay state.
             Defaults to BatteryRelayState.RELAY_STATE_CLOSED.
         component_state: Component state.
@@ -88,7 +88,7 @@ def inverter_data(
     Args:
         component_id: component id
         timestamp: Timestamp of the component message.
-            Defaults to datetime.now(tz=pytz.utc).
+            Defaults to datetime.now(tz=timezone.utc).
         component_state: Component state.
             Defaults to InverterState.CHARGING.
         errors: List of the components error. By default empty list will be created.