Add from_pb decorator to all conversion methods

Signed-off-by: cwasicki <[email protected]>

Author: cwasicki

src/frequenz/client/electricity_trading/_types.py

@@ -83,6 +83,7 @@ class Currency(enum.Enum):
     SGD = price_pb2.Price.Currency.CURRENCY_SGD
 
     @classmethod
+    @from_pb
     def from_pb(cls, currency: price_pb2.Price.Currency.ValueType) -> "Currency":
         """Convert a protobuf Currency value to Currency enum.
 
@@ -213,6 +214,7 @@ class EnergyMarketCodeType(enum.Enum):
     """North American Electric Reliability Corporation identifiers."""
 
     @classmethod
+    @from_pb
     def from_pb(
         cls, energy_market_code_type: delivery_area_pb2.EnergyMarketCodeType.ValueType
     ) -> "EnergyMarketCodeType":
@@ -258,6 +260,7 @@ class DeliveryArea:
     """Type of code used for identifying the delivery area itself."""
 
     @classmethod
+    @from_pb
     def from_pb(cls, delivery_area: delivery_area_pb2.DeliveryArea) -> Self:
         """Convert a protobuf DeliveryArea to DeliveryArea object.
 
@@ -307,6 +310,7 @@ class DeliveryDuration(enum.Enum):
     """1-hour contract duration."""
 
     @classmethod
+    @from_pb
     def from_pb(
         cls, delivery_duration: delivery_duration_pb2.DeliveryDuration.ValueType
     ) -> "DeliveryDuration":
@@ -437,6 +441,7 @@ def __repr__(self) -> str:
         return self.__str__()
 
     @classmethod
+    @from_pb
     def from_pb(cls, delivery_period: delivery_duration_pb2.DeliveryPeriod) -> Self:
         """Convert a protobuf DeliveryPeriod to DeliveryPeriod object.
 
@@ -511,6 +516,7 @@ class OrderExecutionOption(enum.Enum):
     immediately will be cancelled."""
 
     @classmethod
+    @from_pb
     def from_pb(
         cls,
         order_execution_option: electricity_trading_pb2.OrderExecutionOption.ValueType,
@@ -576,6 +582,7 @@ class OrderType(enum.Enum):
     order book and has no market impact. (Not yet supported)."""
 
     @classmethod
+    @from_pb
     def from_pb(
         cls, order_type: electricity_trading_pb2.OrderType.ValueType
     ) -> "OrderType":
@@ -615,6 +622,7 @@ class MarketSide(enum.Enum):
     """Order to sell electricity, referred to as an 'ask' or 'offer' in the order book."""
 
     @classmethod
+    @from_pb
     def from_pb(
         cls, market_side: electricity_trading_pb2.MarketSide.ValueType
     ) -> "MarketSide":
@@ -677,6 +685,7 @@ class OrderState(enum.Enum):
     could be due to certain conditions not yet being met."""
 
     @classmethod
+    @from_pb
     def from_pb(
         cls, order_state: electricity_trading_pb2.OrderState.ValueType
     ) -> "OrderState":
@@ -740,6 +749,7 @@ class TradeState(enum.Enum):
     """An approval has been requested."""
 
     @classmethod
+    @from_pb
     def from_pb(
         cls, trade_state: electricity_trading_pb2.TradeState.ValueType
     ) -> "TradeState":
@@ -840,6 +850,7 @@ class StateReason(enum.Enum):
     """A quote was partially executed."""
 
     @classmethod
+    @from_pb
     def from_pb(
         cls,
         state_reason: electricity_trading_pb2.OrderDetail.StateDetail.StateReason.ValueType,
@@ -893,6 +904,7 @@ class MarketActor(enum.Enum):
     """The system was the actor."""
 
     @classmethod
+    @from_pb
     def from_pb(
         cls,
         market_actor: electricity_trading_pb2.OrderDetail.StateDetail.MarketActor.ValueType,
@@ -1145,6 +1157,7 @@ def __post_init__(self) -> None:
             self.execution_time = self.execution_time.astimezone(timezone.utc)
 
     @classmethod
+    @from_pb
     def from_pb(cls, trade: electricity_trading_pb2.Trade) -> Self:
         """Convert a protobuf Trade to Trade object.
 
@@ -1202,6 +1215,7 @@ class StateDetail:
     """Actor responsible for the current state."""
 
     @classmethod
+    @from_pb
     def from_pb(
         cls, state_detail: electricity_trading_pb2.OrderDetail.StateDetail
     ) -> Self:
@@ -1282,6 +1296,7 @@ def __post_init__(self) -> None:
             self.modification_time = self.modification_time.astimezone(timezone.utc)
 
     @classmethod
+    @from_pb
     def from_pb(cls, order_detail: electricity_trading_pb2.OrderDetail) -> Self:
         """Convert a protobuf OrderDetail to OrderDetail object.
 
@@ -1362,6 +1377,7 @@ def __post_init__(self) -> None:
             self.execution_time = self.execution_time.astimezone(timezone.utc)
 
     @classmethod
+    @from_pb
     def from_pb(cls, public_trade: electricity_trading_pb2.PublicTrade) -> Self:
         """Convert a protobuf PublicTrade to PublicTrade object.
 
@@ -1460,6 +1476,7 @@ def __hash__(self) -> int:
         )
 
     @classmethod
+    @from_pb
     def from_pb(
         cls, gridpool_order_filter: electricity_trading_pb2.GridpoolOrderFilter
     ) -> Self:
@@ -1584,6 +1601,7 @@ def __hash__(self) -> int:
         )
 
     @classmethod
+    @from_pb
     def from_pb(
         cls, gridpool_trade_filter: electricity_trading_pb2.GridpoolTradeFilter
     ) -> "GridpoolTradeFilter":
@@ -1697,6 +1715,7 @@ def __hash__(self) -> int:
         )
 
     @classmethod
+    @from_pb
     def from_pb(
         cls, public_trade_filter: electricity_trading_pb2.PublicTradeFilter
     ) -> Self:
@@ -1809,6 +1828,7 @@ def __post_init__(self) -> None:
                 self.valid_until = self.valid_until.astimezone(timezone.utc)
 
     @classmethod
+    @from_pb
     def from_pb(
         cls,
         update_order: electricity_trading_pb2.UpdateGridpoolOrderRequest.UpdateOrder,