s2-python 0.3.0__tar.gz → 0.4.0__tar.gz

{s2_python-0.3.0 → s2_python-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: s2-python
-Version: 0.3.0
+Version: 0.4.0
 Summary: S2 Protocol Python Wrapper
 Home-page: https://github.com/flexiblepower/s2-ws-json-python
 Author: Flexiblepower

{s2_python-0.3.0 → s2_python-0.4.0}/setup.cfg

@@ -8,7 +8,7 @@ license_files = LICENSE.txt
 long_description = file: README.rst
 long_description_content_type = text/x-rst; charset=UTF-8
 url = https://github.com/flexiblepower/s2-ws-json-python
-version = 0.3.0
+version = 0.4.0
 platforms = Linux
 classifiers = 
 	Development Status :: 4 - Beta

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: s2-python
-Version: 0.3.0
+Version: 0.4.0
 Summary: S2 Protocol Python Wrapper
 Home-page: https://github.com/flexiblepower/s2-ws-json-python
 Author: Flexiblepower

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2_python.egg-info/SOURCES.txt

@@ -10,6 +10,7 @@ src/s2_python.egg-info/not-zip-safe
 src/s2_python.egg-info/requires.txt
 src/s2_python.egg-info/top_level.txt
 src/s2python/__init__.py
+src/s2python/message.py
 src/s2python/py.typed
 src/s2python/reception_status_awaiter.py
 src/s2python/s2_connection.py
@@ -58,4 +59,14 @@ src/s2python/frbc/frbc_usage_forecast.py
 src/s2python/frbc/frbc_usage_forecast_element.py
 src/s2python/frbc/rm.py
 src/s2python/generated/__init__.py
-src/s2python/generated/gen_s2.py
+src/s2python/generated/gen_s2.py
+src/s2python/ppbc/__init__.py
+src/s2python/ppbc/ppbc_end_interruption_instruction.py
+src/s2python/ppbc/ppbc_power_profile_definition.py
+src/s2python/ppbc/ppbc_power_profile_status.py
+src/s2python/ppbc/ppbc_power_sequence.py
+src/s2python/ppbc/ppbc_power_sequence_container.py
+src/s2python/ppbc/ppbc_power_sequence_container_status.py
+src/s2python/ppbc/ppbc_power_sequence_element.py
+src/s2python/ppbc/ppbc_schedule_instruction.py
+src/s2python/ppbc/ppbc_start_interruption_instruction.py

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/duration.py

@@ -4,12 +4,12 @@ import math
 from s2python.generated.gen_s2 import Duration as GenDuration
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class Duration(GenDuration, S2Message["Duration"]):
+class Duration(GenDuration, S2MessageComponent["Duration"]):
     def to_timedelta(self) -> timedelta:
         return timedelta(milliseconds=self.root)
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/handshake.py

@@ -3,12 +3,12 @@ import uuid
 from s2python.generated.gen_s2 import Handshake as GenHandshake
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class Handshake(GenHandshake, S2Message["Handshake"]):
+class Handshake(GenHandshake, S2MessageComponent["Handshake"]):
     model_config = GenHandshake.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/handshake_response.py

@@ -3,12 +3,12 @@ import uuid
 from s2python.generated.gen_s2 import HandshakeResponse as GenHandshakeResponse
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class HandshakeResponse(GenHandshakeResponse, S2Message["HandshakeResponse"]):
+class HandshakeResponse(GenHandshakeResponse, S2MessageComponent["HandshakeResponse"]):
     model_config = GenHandshakeResponse.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/instruction_status_update.py

@@ -5,12 +5,12 @@ from s2python.generated.gen_s2 import (
 )
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class InstructionStatusUpdate(GenInstructionStatusUpdate, S2Message["InstructionStatusUpdate"]):
+class InstructionStatusUpdate(GenInstructionStatusUpdate, S2MessageComponent["InstructionStatusUpdate"]):
     model_config = GenInstructionStatusUpdate.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/number_range.py

@@ -1,11 +1,11 @@
 from typing import Any
 
-from s2python.validate_values_mixin import S2Message, catch_and_convert_exceptions
+from s2python.validate_values_mixin import S2MessageComponent, catch_and_convert_exceptions
 from s2python.generated.gen_s2 import NumberRange as GenNumberRange
 
 
 @catch_and_convert_exceptions
-class NumberRange(GenNumberRange, S2Message["NumberRange"]):
+class NumberRange(GenNumberRange, S2MessageComponent["NumberRange"]):
     model_config = GenNumberRange.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/power_forecast.py

@@ -5,12 +5,12 @@ from s2python.common.power_forecast_element import PowerForecastElement
 from s2python.generated.gen_s2 import PowerForecast as GenPowerForecast
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class PowerForecast(GenPowerForecast, S2Message["PowerForecast"]):
+class PowerForecast(GenPowerForecast, S2MessageComponent["PowerForecast"]):
     model_config = GenPowerForecast.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/power_forecast_element.py

@@ -3,14 +3,14 @@ from typing import List
 from s2python.generated.gen_s2 import PowerForecastElement as GenPowerForecastElement
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 from s2python.common.duration import Duration
 from s2python.common.power_forecast_value import PowerForecastValue
 
 
 @catch_and_convert_exceptions
-class PowerForecastElement(GenPowerForecastElement, S2Message["PowerForecastElement"]):
+class PowerForecastElement(GenPowerForecastElement, S2MessageComponent["PowerForecastElement"]):
     model_config = GenPowerForecastElement.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/power_forecast_value.py

@@ -1,11 +1,11 @@
 from s2python.generated.gen_s2 import PowerForecastValue as GenPowerForecastValue
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class PowerForecastValue(GenPowerForecastValue, S2Message["PowerForecastValue"]):
+class PowerForecastValue(GenPowerForecastValue, S2MessageComponent["PowerForecastValue"]):
     model_config = GenPowerForecastValue.model_config
     model_config["validate_assignment"] = True

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/power_measurement.py

@@ -5,12 +5,12 @@ from s2python.common.power_value import PowerValue
 from s2python.generated.gen_s2 import PowerMeasurement as GenPowerMeasurement
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class PowerMeasurement(GenPowerMeasurement, S2Message["PowerMeasurement"]):
+class PowerMeasurement(GenPowerMeasurement, S2MessageComponent["PowerMeasurement"]):
     model_config = GenPowerMeasurement.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/power_range.py

@@ -4,13 +4,13 @@ from pydantic import model_validator
 
 from s2python.generated.gen_s2 import PowerRange as GenPowerRange
 from s2python.validate_values_mixin import (
-    S2Message,
+    S2MessageComponent,
     catch_and_convert_exceptions,
 )
 
 
 @catch_and_convert_exceptions
-class PowerRange(GenPowerRange, S2Message["PowerRange"]):
+class PowerRange(GenPowerRange, S2MessageComponent["PowerRange"]):
     model_config = GenPowerRange.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/power_value.py

@@ -1,11 +1,11 @@
 from s2python.generated.gen_s2 import PowerValue as GenPowerValue
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class PowerValue(GenPowerValue, S2Message["PowerValue"]):
+class PowerValue(GenPowerValue, S2MessageComponent["PowerValue"]):
     model_config = GenPowerValue.model_config
     model_config["validate_assignment"] = True

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/reception_status.py

@@ -3,12 +3,12 @@ import uuid
 from s2python.generated.gen_s2 import ReceptionStatus as GenReceptionStatus
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class ReceptionStatus(GenReceptionStatus, S2Message["ReceptionStatus"]):
+class ReceptionStatus(GenReceptionStatus, S2MessageComponent["ReceptionStatus"]):
     model_config = GenReceptionStatus.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/resource_manager_details.py

@@ -8,12 +8,12 @@ from s2python.generated.gen_s2 import (
 )
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class ResourceManagerDetails(GenResourceManagerDetails, S2Message["ResourceManagerDetails"]):
+class ResourceManagerDetails(GenResourceManagerDetails, S2MessageComponent["ResourceManagerDetails"]):
     model_config = GenResourceManagerDetails.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/revoke_object.py

@@ -3,12 +3,12 @@ import uuid
 from s2python.generated.gen_s2 import RevokeObject as GenRevokeObject
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class RevokeObject(GenRevokeObject, S2Message["RevokeObject"]):
+class RevokeObject(GenRevokeObject, S2MessageComponent["RevokeObject"]):
     model_config = GenRevokeObject.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/role.py

@@ -1,11 +1,11 @@
 from s2python.generated.gen_s2 import Role as GenRole
 from s2python.validate_values_mixin import (
-    S2Message,
+    S2MessageComponent,
     catch_and_convert_exceptions,
 )
 
 
 @catch_and_convert_exceptions
-class Role(GenRole, S2Message["Role"]):
+class Role(GenRole, S2MessageComponent["Role"]):
     model_config = GenRole.model_config
     model_config["validate_assignment"] = True

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/select_control_type.py

@@ -3,12 +3,12 @@ import uuid
 from s2python.generated.gen_s2 import SelectControlType as GenSelectControlType
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class SelectControlType(GenSelectControlType, S2Message["SelectControlType"]):
+class SelectControlType(GenSelectControlType, S2MessageComponent["SelectControlType"]):
     model_config = GenSelectControlType.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/session_request.py

@@ -3,12 +3,12 @@ import uuid
 from s2python.generated.gen_s2 import SessionRequest as GenSessionRequest
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class SessionRequest(GenSessionRequest, S2Message["SessionRequest"]):
+class SessionRequest(GenSessionRequest, S2MessageComponent["SessionRequest"]):
     model_config = GenSessionRequest.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/timer.py

@@ -3,13 +3,13 @@ import uuid
 from s2python.common.duration import Duration
 from s2python.generated.gen_s2 import Timer as GenTimer
 from s2python.validate_values_mixin import (
-    S2Message,
+    S2MessageComponent,
     catch_and_convert_exceptions,
 )
 
 
 @catch_and_convert_exceptions
-class Timer(GenTimer, S2Message["Timer"]):
+class Timer(GenTimer, S2MessageComponent["Timer"]):
     model_config = GenTimer.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/common/transition.py

@@ -4,13 +4,13 @@ from typing import Optional, List
 from s2python.common.duration import Duration
 from s2python.generated.gen_s2 import Transition as GenTransition
 from s2python.validate_values_mixin import (
-    S2Message,
+    S2MessageComponent,
     catch_and_convert_exceptions,
 )
 
 
 @catch_and_convert_exceptions
-class Transition(GenTransition, S2Message["Transition"]):
+class Transition(GenTransition, S2MessageComponent["Transition"]):
     model_config = GenTransition.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_actuator_description.py

@@ -12,13 +12,13 @@ from s2python.generated.gen_s2 import (
     FRBCActuatorDescription as GenFRBCActuatorDescription,
 )
 from s2python.validate_values_mixin import (
-    S2Message,
+    S2MessageComponent,
     catch_and_convert_exceptions,
 )
 
 
 @catch_and_convert_exceptions
-class FRBCActuatorDescription(GenFRBCActuatorDescription, S2Message["FRBCActuatorDescription"]):
+class FRBCActuatorDescription(GenFRBCActuatorDescription, S2MessageComponent["FRBCActuatorDescription"]):
     model_config = GenFRBCActuatorDescription.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_actuator_status.py

@@ -4,12 +4,12 @@ import uuid
 from s2python.generated.gen_s2 import FRBCActuatorStatus as GenFRBCActuatorStatus
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class FRBCActuatorStatus(GenFRBCActuatorStatus, S2Message["FRBCActuatorStatus"]):
+class FRBCActuatorStatus(GenFRBCActuatorStatus, S2MessageComponent["FRBCActuatorStatus"]):
     model_config = GenFRBCActuatorStatus.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_fill_level_target_profile.py

@@ -9,12 +9,12 @@ from s2python.generated.gen_s2 import (
 )
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class FRBCFillLevelTargetProfile(GenFRBCFillLevelTargetProfile, S2Message["FRBCFillLevelTargetProfile"]):
+class FRBCFillLevelTargetProfile(GenFRBCFillLevelTargetProfile, S2MessageComponent["FRBCFillLevelTargetProfile"]):
     model_config = GenFRBCFillLevelTargetProfile.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_fill_level_target_profile_element.py

@@ -8,12 +8,12 @@ from s2python.common import Duration, NumberRange
 from s2python.generated.gen_s2 import (
     FRBCFillLevelTargetProfileElement as GenFRBCFillLevelTargetProfileElement,
 )
-from s2python.validate_values_mixin import catch_and_convert_exceptions, S2Message
+from s2python.validate_values_mixin import catch_and_convert_exceptions, S2MessageComponent
 
 
 @catch_and_convert_exceptions
 class FRBCFillLevelTargetProfileElement(
-    GenFRBCFillLevelTargetProfileElement, S2Message["FRBCFillLevelTargetProfileElement"]
+    GenFRBCFillLevelTargetProfileElement, S2MessageComponent["FRBCFillLevelTargetProfileElement"]
 ):
     model_config = GenFRBCFillLevelTargetProfileElement.model_config
     model_config["validate_assignment"] = True

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_instruction.py

@@ -3,12 +3,12 @@ import uuid
 from s2python.generated.gen_s2 import FRBCInstruction as GenFRBCInstruction
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class FRBCInstruction(GenFRBCInstruction, S2Message["FRBCInstruction"]):
+class FRBCInstruction(GenFRBCInstruction, S2MessageComponent["FRBCInstruction"]):
     model_config = GenFRBCInstruction.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_leakage_behaviour.py

@@ -5,12 +5,12 @@ from s2python.frbc.frbc_leakage_behaviour_element import FRBCLeakageBehaviourEle
 from s2python.generated.gen_s2 import FRBCLeakageBehaviour as GenFRBCLeakageBehaviour
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class FRBCLeakageBehaviour(GenFRBCLeakageBehaviour, S2Message["FRBCLeakageBehaviour"]):
+class FRBCLeakageBehaviour(GenFRBCLeakageBehaviour, S2MessageComponent["FRBCLeakageBehaviour"]):
     model_config = GenFRBCLeakageBehaviour.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_leakage_behaviour_element.py

@@ -5,12 +5,12 @@ from typing_extensions import Self
 
 from s2python.common import NumberRange
 from s2python.generated.gen_s2 import FRBCLeakageBehaviourElement as GenFRBCLeakageBehaviourElement
-from s2python.validate_values_mixin import catch_and_convert_exceptions, S2Message
+from s2python.validate_values_mixin import catch_and_convert_exceptions, S2MessageComponent
 
 
 @catch_and_convert_exceptions
 class FRBCLeakageBehaviourElement(
-    GenFRBCLeakageBehaviourElement, S2Message["FRBCLeakageBehaviourElement"]
+    GenFRBCLeakageBehaviourElement, S2MessageComponent["FRBCLeakageBehaviourElement"]
 ):
     model_config = GenFRBCLeakageBehaviourElement.model_config
     model_config["validate_assignment"] = True

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_operation_mode.py

@@ -9,14 +9,14 @@ from s2python.common import NumberRange
 from s2python.frbc.frbc_operation_mode_element import FRBCOperationModeElement
 from s2python.generated.gen_s2 import FRBCOperationMode as GenFRBCOperationMode
 from s2python.validate_values_mixin import (
-    S2Message,
+    S2MessageComponent,
     catch_and_convert_exceptions,
 )
 from s2python.utils import pairwise
 
 
 @catch_and_convert_exceptions
-class FRBCOperationMode(GenFRBCOperationMode, S2Message["FRBCOperationMode"]):
+class FRBCOperationMode(GenFRBCOperationMode, S2MessageComponent["FRBCOperationMode"]):
     model_config = GenFRBCOperationMode.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_operation_mode_element.py

@@ -5,13 +5,13 @@ from s2python.generated.gen_s2 import (
     FRBCOperationModeElement as GenFRBCOperationModeElement,
 )
 from s2python.validate_values_mixin import (
-    S2Message,
+    S2MessageComponent,
     catch_and_convert_exceptions,
 )
 
 
 @catch_and_convert_exceptions
-class FRBCOperationModeElement(GenFRBCOperationModeElement, S2Message["FRBCOperationModeElement"]):
+class FRBCOperationModeElement(GenFRBCOperationModeElement, S2MessageComponent["FRBCOperationModeElement"]):
     model_config = GenFRBCOperationModeElement.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_storage_description.py

@@ -4,12 +4,12 @@ from s2python.generated.gen_s2 import (
 )
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class FRBCStorageDescription(GenFRBCStorageDescription, S2Message["FRBCStorageDescription"]):
+class FRBCStorageDescription(GenFRBCStorageDescription, S2MessageComponent["FRBCStorageDescription"]):
     model_config = GenFRBCStorageDescription.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_storage_status.py

@@ -3,12 +3,12 @@ import uuid
 from s2python.generated.gen_s2 import FRBCStorageStatus as GenFRBCStorageStatus
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class FRBCStorageStatus(GenFRBCStorageStatus, S2Message["FRBCStorageStatus"]):
+class FRBCStorageStatus(GenFRBCStorageStatus, S2MessageComponent["FRBCStorageStatus"]):
     model_config = GenFRBCStorageStatus.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_system_description.py

@@ -4,14 +4,14 @@ import uuid
 from s2python.generated.gen_s2 import FRBCSystemDescription as GenFRBCSystemDescription
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 from s2python.frbc.frbc_actuator_description import FRBCActuatorDescription
 from s2python.frbc.frbc_storage_description import FRBCStorageDescription
 
 
 @catch_and_convert_exceptions
-class FRBCSystemDescription(GenFRBCSystemDescription, S2Message["FRBCSystemDescription"]):
+class FRBCSystemDescription(GenFRBCSystemDescription, S2MessageComponent["FRBCSystemDescription"]):
     model_config = GenFRBCSystemDescription.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_timer_status.py

@@ -3,12 +3,12 @@ import uuid
 from s2python.generated.gen_s2 import FRBCTimerStatus as GenFRBCTimerStatus
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class FRBCTimerStatus(GenFRBCTimerStatus, S2Message["FRBCTimerStatus"]):
+class FRBCTimerStatus(GenFRBCTimerStatus, S2MessageComponent["FRBCTimerStatus"]):
     model_config = GenFRBCTimerStatus.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_usage_forecast.py

@@ -4,13 +4,13 @@ import uuid
 from s2python.generated.gen_s2 import FRBCUsageForecast as GenFRBCUsageForecast
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 from s2python.frbc.frbc_usage_forecast_element import FRBCUsageForecastElement
 
 
 @catch_and_convert_exceptions
-class FRBCUsageForecast(GenFRBCUsageForecast, S2Message["FRBCUsageForecast"]):
+class FRBCUsageForecast(GenFRBCUsageForecast, S2MessageComponent["FRBCUsageForecast"]):
     model_config = GenFRBCUsageForecast.model_config
     model_config["validate_assignment"] = True
 

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/frbc/frbc_usage_forecast_element.py

@@ -5,12 +5,12 @@ from s2python.generated.gen_s2 import (
 )
 from s2python.validate_values_mixin import (
     catch_and_convert_exceptions,
-    S2Message,
+    S2MessageComponent,
 )
 
 
 @catch_and_convert_exceptions
-class FRBCUsageForecastElement(GenFRBCUsageForecastElement, S2Message["FRBCUsageForecastElement"]):
+class FRBCUsageForecastElement(GenFRBCUsageForecastElement, S2MessageComponent["FRBCUsageForecastElement"]):
     model_config = GenFRBCUsageForecastElement.model_config
     model_config["validate_assignment"] = True
 

s2_python-0.4.0/src/s2python/message.py

@@ -0,0 +1,50 @@
+from typing import Union
+
+from s2python.frbc import (
+    FRBCActuatorStatus,
+    FRBCFillLevelTargetProfile,
+    FRBCInstruction,
+    FRBCLeakageBehaviour,
+    FRBCStorageStatus,
+    FRBCSystemDescription,
+    FRBCTimerStatus,
+    FRBCUsageForecast
+)
+from s2python.ppbc import (
+    PPBCScheduleInstruction,
+)
+
+from s2python.common import (
+    Handshake,
+    HandshakeResponse,
+    InstructionStatusUpdate,
+    PowerForecast,
+    PowerMeasurement,
+    ReceptionStatus,
+    ResourceManagerDetails,
+    RevokeObject,
+    SelectControlType,
+    SessionRequest
+)
+
+S2Message = Union[
+    FRBCActuatorStatus,
+    FRBCFillLevelTargetProfile,
+    FRBCInstruction,
+    FRBCLeakageBehaviour,
+    FRBCStorageStatus,
+    FRBCSystemDescription,
+    FRBCTimerStatus,
+    FRBCUsageForecast,
+    PPBCScheduleInstruction,
+    Handshake,
+    HandshakeResponse,
+    InstructionStatusUpdate,
+    PowerForecast,
+    PowerMeasurement,
+    ReceptionStatus,
+    ResourceManagerDetails,
+    RevokeObject,
+    SelectControlType,
+    SessionRequest,
+]

s2_python-0.4.0/src/s2python/ppbc/__init__.py

@@ -0,0 +1,12 @@
+from s2python.ppbc.ppbc_schedule_instruction import PPBCScheduleInstruction
+from s2python.ppbc.ppbc_end_interruption_instruction import (
+    PPBCEndInterruptionInstruction,
+)
+from s2python.ppbc.ppbc_power_profile_definition import PPBCPowerProfileDefinition
+from s2python.ppbc.ppbc_power_sequence_container import PPBCPowerSequenceContainer
+from s2python.ppbc.ppbc_power_sequence import PPBCPowerSequence
+from s2python.ppbc.ppbc_power_profile_status import PPBCPowerProfileStatus
+from s2python.ppbc.ppbc_power_sequence_container_status import (
+    PPBCPowerSequenceContainerStatus,
+)
+from s2python.ppbc.ppbc_power_sequence_element import PPBCPowerSequenceElement

s2_python-0.4.0/src/s2python/ppbc/ppbc_end_interruption_instruction.py

@@ -0,0 +1,32 @@
+import uuid
+
+from s2python.generated.gen_s2 import (
+    PPBCEndInterruptionInstruction as GenPPBCEndInterruptionInstruction,
+)
+
+from s2python.validate_values_mixin import (
+    S2MessageComponent,
+    catch_and_convert_exceptions,
+)
+
+
+@catch_and_convert_exceptions
+class PPBCEndInterruptionInstruction(
+    GenPPBCEndInterruptionInstruction, S2MessageComponent["PPBCEndInterruptionInstruction"]
+):
+    model_config = GenPPBCEndInterruptionInstruction.model_config
+    model_config["validate_assignment"] = True
+
+    id: uuid.UUID = GenPPBCEndInterruptionInstruction.model_fields["id"]  # type: ignore[assignment]
+    power_profile_id: uuid.UUID = GenPPBCEndInterruptionInstruction.model_fields[
+        "power_profile_id"
+    ]  # type: ignore[assignment]
+    sequence_container_id: uuid.UUID = GenPPBCEndInterruptionInstruction.model_fields[
+        "sequence_container_id"
+    ]  # type: ignore[assignment]
+    power_sequence_id: uuid.UUID = GenPPBCEndInterruptionInstruction.model_fields[
+        "power_sequence_id"
+    ]  # type: ignore[assignment]
+    abnormal_condition: bool = GenPPBCEndInterruptionInstruction.model_fields[
+        "abnormal_condition"
+    ]  # type: ignore[assignment]

s2_python-0.4.0/src/s2python/ppbc/ppbc_power_profile_definition.py

@@ -0,0 +1,27 @@
+from typing import List
+import uuid
+
+from s2python.generated.gen_s2 import (
+    PPBCPowerProfileDefinition as GenPPBCPowerProfileDefinition,
+)
+
+from s2python.validate_values_mixin import (
+    S2MessageComponent,
+    catch_and_convert_exceptions,
+)
+
+from s2python.ppbc.ppbc_power_sequence_container import PPBCPowerSequenceContainer
+
+
+@catch_and_convert_exceptions
+class PPBCPowerProfileDefinition(
+    GenPPBCPowerProfileDefinition, S2MessageComponent["PPBCPowerProfileDefinition"]
+):
+    model_config = GenPPBCPowerProfileDefinition.model_config
+    model_config["validate_assignment"] = True
+
+    message_id: uuid.UUID = GenPPBCPowerProfileDefinition.model_fields["message_id"]  # type: ignore[assignment]
+    id: uuid.UUID = GenPPBCPowerProfileDefinition.model_fields["id"]  # type: ignore[assignment]
+    power_sequences_containers: List[PPBCPowerSequenceContainer] = (
+        GenPPBCPowerProfileDefinition.model_fields["power_sequences_containers"]  # type: ignore[assignment]
+    )

s2_python-0.4.0/src/s2python/ppbc/ppbc_power_profile_status.py

@@ -0,0 +1,26 @@
+from typing import List
+
+from s2python.generated.gen_s2 import (
+    PPBCPowerProfileStatus as GenPPBCPowerProfileStatus,
+)
+
+from s2python.validate_values_mixin import (
+    S2MessageComponent,
+    catch_and_convert_exceptions,
+)
+
+from s2python.ppbc.ppbc_power_sequence_container_status import (
+    PPBCPowerSequenceContainerStatus,
+)
+
+
+@catch_and_convert_exceptions
+class PPBCPowerProfileStatus(
+    GenPPBCPowerProfileStatus, S2MessageComponent["PPBCPowerProfileStatus"]
+):
+    model_config = GenPPBCPowerProfileStatus.model_config
+    model_config["validate_assignment"] = True
+
+    sequence_container_status: List[PPBCPowerSequenceContainerStatus] = (
+        GenPPBCPowerProfileStatus.model_fields["sequence_container_status"]  # type: ignore[assignment]
+    )

s2_python-0.4.0/src/s2python/ppbc/ppbc_power_sequence.py

@@ -0,0 +1,30 @@
+from typing import List
+import uuid
+
+from s2python.generated.gen_s2 import (
+    PPBCPowerSequence as GenPPBCPowerSequence,
+)
+
+from s2python.validate_values_mixin import (
+    S2MessageComponent,
+    catch_and_convert_exceptions,
+)
+
+from s2python.ppbc.ppbc_power_sequence_element import PPBCPowerSequenceElement
+from s2python.common import Duration
+
+
+@catch_and_convert_exceptions
+class PPBCPowerSequence(GenPPBCPowerSequence, S2MessageComponent["PPBCPowerSequence"]):
+    model_config = GenPPBCPowerSequence.model_config
+    model_config["validate_assignment"] = True
+
+    id: uuid.UUID = GenPPBCPowerSequence.model_fields["id"]  # type: ignore[assignment]
+    elements: List[PPBCPowerSequenceElement] = GenPPBCPowerSequence.model_fields[
+        "elements"
+    ]  # type: ignore[assignment]
+    is_interruptible: bool = GenPPBCPowerSequence.model_fields["is_interruptible"]  # type: ignore[assignment]
+    max_pause_before: Duration = GenPPBCPowerSequence.model_fields["max_pause_before"]  # type: ignore[assignment]
+    abnormal_condition_only: bool = GenPPBCPowerSequence.model_fields[
+        "abnormal_condition_only"
+    ]  # type: ignore[assignment]

s2_python-0.4.0/src/s2python/ppbc/ppbc_power_sequence_container.py

@@ -0,0 +1,27 @@
+from typing import List
+import uuid
+
+
+from s2python.generated.gen_s2 import (
+    PPBCPowerSequenceContainer as GenPPBCPowerSequenceContainer,
+)
+
+from s2python.validate_values_mixin import (
+    S2MessageComponent,
+    catch_and_convert_exceptions,
+)
+
+from s2python.ppbc.ppbc_power_sequence import PPBCPowerSequence
+
+
+@catch_and_convert_exceptions
+class PPBCPowerSequenceContainer(
+    GenPPBCPowerSequenceContainer, S2MessageComponent["PPBCPowerSequenceContainer"]
+):
+    model_config = GenPPBCPowerSequenceContainer.model_config
+    model_config["validate_assignment"] = True
+
+    id: uuid.UUID = GenPPBCPowerSequenceContainer.model_fields["id"]  # type: ignore[assignment]
+    power_sequences: List[PPBCPowerSequence] = (
+        GenPPBCPowerSequenceContainer.model_fields["power_sequences"]  # type: ignore[assignment]
+    )

s2_python-0.4.0/src/s2python/ppbc/ppbc_power_sequence_container_status.py

@@ -0,0 +1,32 @@
+import uuid
+from typing import Union
+
+from s2python.generated.gen_s2 import (
+    PPBCPowerSequenceContainerStatus as GenPPBCPowerSequenceContainerStatus,
+)
+
+from s2python.validate_values_mixin import (
+    S2MessageComponent,
+    catch_and_convert_exceptions,
+)
+
+
+@catch_and_convert_exceptions
+class PPBCPowerSequenceContainerStatus(
+    GenPPBCPowerSequenceContainerStatus, S2MessageComponent["PPBCPowerSequenceContainerStatus"]
+):
+    model_config = GenPPBCPowerSequenceContainerStatus.model_config
+    model_config["validate_assignment"] = True
+
+    power_profile_id: uuid.UUID = GenPPBCPowerSequenceContainerStatus.model_fields[
+        "power_profile_id"  # type: ignore[assignment]
+    ]
+    sequence_container_id: uuid.UUID = GenPPBCPowerSequenceContainerStatus.model_fields[
+        "sequence_container_id"  # type: ignore[assignment]
+    ]
+    selected_sequence_id: Union[uuid.UUID, None] = (
+        GenPPBCPowerSequenceContainerStatus.model_fields["selected_sequence_id"]  # type: ignore[assignment]
+    )
+    progress: Union[uuid.UUID, None] = GenPPBCPowerSequenceContainerStatus.model_fields[
+        "progress"  # type: ignore[assignment]
+    ]

s2_python-0.4.0/src/s2python/ppbc/ppbc_power_sequence_element.py

@@ -0,0 +1,25 @@
+from typing import List
+
+from s2python.generated.gen_s2 import (
+    PPBCPowerSequenceElement as GenPPBCPowerSequenceElement,
+)
+
+from s2python.validate_values_mixin import (
+    S2MessageComponent,
+    catch_and_convert_exceptions,
+)
+
+from s2python.common import Duration, PowerForecastValue
+
+
+@catch_and_convert_exceptions
+class PPBCPowerSequenceElement(
+    GenPPBCPowerSequenceElement, S2MessageComponent["PPBCPowerSequenceElement"]
+):
+    model_config = GenPPBCPowerSequenceElement.model_config
+    model_config["validate_assignment"] = True
+
+    duration: Duration = GenPPBCPowerSequenceElement.model_fields["duration"]  # type: ignore[assignment]
+    power_values: List[PowerForecastValue] = GenPPBCPowerSequenceElement.model_fields[
+        "power_values"
+    ]  # type: ignore[assignment]

s2_python-0.4.0/src/s2python/ppbc/ppbc_schedule_instruction.py

@@ -0,0 +1,33 @@
+import uuid
+
+from s2python.generated.gen_s2 import (
+    PPBCScheduleInstruction as GenPPBCScheduleInstruction,
+)
+from s2python.validate_values_mixin import (
+    catch_and_convert_exceptions,
+    S2MessageComponent,
+)
+
+
+@catch_and_convert_exceptions
+class PPBCScheduleInstruction(
+    GenPPBCScheduleInstruction, S2MessageComponent["PPBCScheduleInstruction"]
+):
+    model_config = GenPPBCScheduleInstruction.model_config
+    model_config["validate_assignment"] = True
+
+    id: uuid.UUID = GenPPBCScheduleInstruction.model_fields["id"]  # type: ignore[assignment]
+
+    power_profile_id: uuid.UUID = GenPPBCScheduleInstruction.model_fields[
+        "power_profile_id"
+    ]  # type: ignore[assignment]
+
+    message_id: uuid.UUID = GenPPBCScheduleInstruction.model_fields["message_id"]  # type: ignore[assignment]
+
+    sequence_container_id: uuid.UUID = GenPPBCScheduleInstruction.model_fields[
+        "sequence_container_id"
+    ]  # type: ignore[assignment]
+
+    power_sequence_id: uuid.UUID = GenPPBCScheduleInstruction.model_fields[
+        "power_sequence_id"
+    ]  # type: ignore[assignment]

s2_python-0.4.0/src/s2python/ppbc/ppbc_start_interruption_instruction.py

@@ -0,0 +1,32 @@
+import uuid
+
+from s2python.generated.gen_s2 import (
+    PPBCStartInterruptionInstruction as GenPPBCStartInterruptionInstruction,
+)
+
+from s2python.validate_values_mixin import (
+    S2MessageComponent,
+    catch_and_convert_exceptions,
+)
+
+
+@catch_and_convert_exceptions
+class PPBCStartInterruptionInstruction(
+    GenPPBCStartInterruptionInstruction, S2MessageComponent["PPBCStartInterruptionInstruction"]
+):
+    model_config = GenPPBCStartInterruptionInstruction.model_config
+    model_config["validate_assignment"] = True
+
+    id: uuid.UUID = GenPPBCStartInterruptionInstruction.model_fields["id"]  # type: ignore[assignment]
+    power_profile_id: uuid.UUID = GenPPBCStartInterruptionInstruction.model_fields[
+        "power_profile_id"
+    ]  # type: ignore[assignment]
+    sequence_container_id: uuid.UUID = GenPPBCStartInterruptionInstruction.model_fields[
+        "sequence_container_id"
+    ]  # type: ignore[assignment]
+    power_sequence_id: uuid.UUID = GenPPBCStartInterruptionInstruction.model_fields[
+        "power_sequence_id"
+    ]  # type: ignore[assignment]
+    abnormal_condition: bool = GenPPBCStartInterruptionInstruction.model_fields[
+        "abnormal_condition"
+    ]  # type: ignore[assignment]

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/s2_connection.py

@@ -27,7 +27,7 @@ from s2python.reception_status_awaiter import ReceptionStatusAwaiter
 from s2python.s2_control_type import S2ControlType
 from s2python.s2_parser import S2Parser
 from s2python.s2_validation_error import S2ValidationError
-from s2python.validate_values_mixin import S2Message
+from s2python.message import S2Message
 from s2python.version import S2_VERSION
 
 logger = logging.getLogger("s2python")
@@ -141,7 +141,7 @@ class MessageHandlers:
         """
         handler = self.handlers.get(type(msg))
         if handler is not None:
-            send_okay = SendOkay(connection, msg.message_id)  # type: ignore[attr-defined]
+            send_okay = SendOkay(connection, msg.message_id)  # type: ignore[attr-defined, union-attr]
 
             try:
                 if asyncio.iscoroutinefunction(handler):
@@ -158,9 +158,9 @@ class MessageHandlers:
             except Exception:
                 if not send_okay.status_is_send.is_set():
                     await connection.respond_with_reception_status(
-                        subject_message_id=str(msg.message_id),  # type: ignore[attr-defined]
+                        subject_message_id=str(msg.message_id),  # type: ignore[attr-defined, union-attr]
                         status=ReceptionStatusValues.PERMANENT_ERROR,
-                        diagnostic_label=f"While processing message {msg.message_id} "  # type: ignore[attr-defined]
+                        diagnostic_label=f"While processing message {msg.message_id} "  # type: ignore[attr-defined, union-attr]  # pylint: disable=line-too-long
                         f"an unrecoverable error occurred.",
                     )
                 raise
@@ -490,17 +490,17 @@ class S2Connection:  # pylint: disable=too-many-instance-attributes
         await self._send_and_forget(s2_msg)
         logger.debug(
             "Waiting for ReceptionStatus for %s %s seconds",
-            s2_msg.message_id,  # type: ignore[attr-defined]
+            s2_msg.message_id,  # type: ignore[attr-defined, union-attr]
             timeout_reception_status,
         )
         try:
             reception_status = await self.reception_status_awaiter.wait_for_reception_status(
-                s2_msg.message_id, timeout_reception_status  # type: ignore[attr-defined]
+                s2_msg.message_id, timeout_reception_status  # type: ignore[attr-defined, union-attr]
             )
         except TimeoutError:
             logger.error(
                 "Did not receive a reception status on time for %s",
-                s2_msg.message_id,  # type: ignore[attr-defined]
+                s2_msg.message_id,  # type: ignore[attr-defined, union-attr]
             )
             self._stop_event.set()
             raise

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/s2_control_type.py

@@ -3,7 +3,8 @@ import typing
 
 from s2python.common import ControlType as ProtocolControlType
 from s2python.frbc import FRBCInstruction
-from s2python.validate_values_mixin import S2Message
+from s2python.ppbc import PPBCScheduleInstruction
+from s2python.message import S2Message
 
 if typing.TYPE_CHECKING:
     from s2python.s2_connection import S2Connection, MessageHandlers
@@ -36,10 +37,33 @@ class FRBCControlType(S2ControlType):
     ) -> None: ...
 
     @abc.abstractmethod
-    def activate(self, conn: "S2Connection") -> None: ...
+    def activate(self, conn: "S2Connection") -> None:
+        """Overwrite with the actual dctivation logic of your Resource Manager for this particular control type."""
 
     @abc.abstractmethod
-    def deactivate(self, conn: "S2Connection") -> None: ...
+    def deactivate(self, conn: "S2Connection") -> None:
+        """Overwrite with the actual deactivation logic of your Resource Manager for this particular control type."""
+
+
+class PPBCControlType(S2ControlType):
+    def get_protocol_control_type(self) -> ProtocolControlType:
+        return ProtocolControlType.POWER_PROFILE_BASED_CONTROL
+
+    def register_handlers(self, handlers: "MessageHandlers") -> None:
+        handlers.register_handler(PPBCScheduleInstruction, self.handle_instruction)
+
+    @abc.abstractmethod
+    def handle_instruction(
+        self, conn: "S2Connection", msg: S2Message, send_okay: typing.Callable[[], None]
+    ) -> None: ...
+
+    @abc.abstractmethod
+    def activate(self, conn: "S2Connection") -> None:
+        """Overwrite with the actual dctivation logic of your Resource Manager for this particular control type."""
+
+    @abc.abstractmethod
+    def deactivate(self, conn: "S2Connection") -> None:
+        """Overwrite with the actual deactivation logic of your Resource Manager for this particular control type."""
 
 
 class NoControlControlType(S2ControlType):

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/s2_parser.py

@@ -24,14 +24,17 @@ from s2python.frbc import (
     FRBCTimerStatus,
     FRBCUsageForecast,
 )
-from s2python.validate_values_mixin import S2Message
+from s2python.ppbc import PPBCScheduleInstruction
+
+from s2python.message import S2Message
+from s2python.validate_values_mixin import S2MessageComponent
 from s2python.s2_validation_error import S2ValidationError
 
 
 LOGGER = logging.getLogger(__name__)
 S2MessageType = str
 
-M = TypeVar("M", bound=S2Message)
+M = TypeVar("M", bound=S2MessageComponent)
 
 
 # May be generated with development_utilities/generate_s2_message_type_to_class.py
@@ -44,6 +47,7 @@ TYPE_TO_MESSAGE_CLASS: Dict[str, Type[S2Message]] = {
     "FRBC.SystemDescription": FRBCSystemDescription,
     "FRBC.TimerStatus": FRBCTimerStatus,
     "FRBC.UsageForecast": FRBCUsageForecast,
+    "PPBC.ScheduleInstruction": PPBCScheduleInstruction,
     "Handshake": Handshake,
     "HandshakeResponse": HandshakeResponse,
     "InstructionStatusUpdate": InstructionStatusUpdate,
@@ -86,7 +90,9 @@ class S2Parser:
         return TYPE_TO_MESSAGE_CLASS[message_type].model_validate(message_json)
 
     @staticmethod
-    def parse_as_message(unparsed_message: Union[dict, str, bytes], as_message: Type[M]) -> M:
+    def parse_as_message(
+        unparsed_message: Union[dict, str, bytes], as_message: Type[M]
+    ) -> M:
         """Parse the message to a specific S2 python message.
 
         :param unparsed_message: The message as a JSON-formatted string or as a JSON-parsed dictionary.
@@ -98,7 +104,9 @@ class S2Parser:
         return as_message.from_dict(message_json)
 
     @staticmethod
-    def parse_message_type(unparsed_message: Union[dict, str, bytes]) -> Optional[S2MessageType]:
+    def parse_message_type(
+        unparsed_message: Union[dict, str, bytes],
+    ) -> Optional[S2MessageType]:
         """Parse only the message type from the unparsed message.
 
         This is useful to call before `parse_as_message` to retrieve the message type and allows for strictly-typed

{s2_python-0.3.0 → s2_python-0.4.0}/src/s2python/validate_values_mixin.py

@@ -15,7 +15,7 @@ MappingIntStrAny = Mapping[IntStr, Any]
 C = TypeVar("C", bound="BaseModel")
 
 
-class S2Message(BaseModel, Generic[C]):
+class S2MessageComponent(BaseModel, Generic[C]):
     def to_json(self: C) -> str:
         try:
             return self.model_dump_json(by_alias=True, exclude_none=True)
@@ -59,7 +59,7 @@ def convert_to_s2exception(f: Callable) -> Callable:
     return inner
 
 
-def catch_and_convert_exceptions(input_class: Type[S2Message[B_co]]) -> Type[S2Message[B_co]]:
+def catch_and_convert_exceptions(input_class: Type[S2MessageComponent[B_co]]) -> Type[S2MessageComponent[B_co]]:
     input_class.__init__ = convert_to_s2exception(input_class.__init__)  # type: ignore[method-assign]
     input_class.__setattr__ = convert_to_s2exception(input_class.__setattr__)  # type: ignore[method-assign]
     input_class.model_validate_json = convert_to_s2exception(  # type: ignore[method-assign]