mms-client 1.5.1__tar.gz → 1.6.0__tar.gz

{mms_client-1.5.1 → mms_client-1.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mms-client
-Version: 1.5.1
+Version: 1.6.0
 Summary: API client for accessing the MMS
 Home-page: https://github.com/ElectroRoute-Japan/mms-client
 Author: Ryan Wood
@@ -209,6 +209,7 @@ This same input allows for the user to create their own plugins and add them to
 
 # Completeness
 This client is not complete. Currently, it supports the following endpoints:
+- MarketQuery_ReserveRequirementQuery
 - MarketSubmit_OfferData
 - MarketQuery_OfferQuery
 - MarketCancel_OfferCancel

{mms_client-1.5.1 → mms_client-1.6.0}/README.md

@@ -177,6 +177,7 @@ This same input allows for the user to create their own plugins and add them to
 
 # Completeness
 This client is not complete. Currently, it supports the following endpoints:
+- MarketQuery_ReserveRequirementQuery
 - MarketSubmit_OfferData
 - MarketQuery_OfferQuery
 - MarketCancel_OfferCancel

{mms_client-1.5.1 → mms_client-1.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mms_client"
-version = "v1.5.1"
+version = "v1.6.0"
 description = "API client for accessing the MMS"
 authors = ["Ryan Wood <ryan.wood@electroroute.co.jp>"]
 readme = "README.md"

{mms_client-1.5.1 → mms_client-1.6.0}/src/mms_client/services/base.py

@@ -28,6 +28,7 @@ from mms_client.types.transport import RequestType
 from mms_client.types.transport import ResponseDataType
 from mms_client.utils.errors import AudienceError
 from mms_client.utils.errors import MMSClientError
+from mms_client.utils.errors import MMSServerError
 from mms_client.utils.errors import MMSValidationError
 from mms_client.utils.serialization import Serializer
 from mms_client.utils.web import ClientType
@@ -459,7 +460,11 @@ class BaseClient:  # pylint: disable=too-many-instance-attributes
         MMSClientError: If the response is not valid.
         """
         # Verify that the response is in the correct format. If it's not, raise an error.
-        if resp.data_type != ResponseDataType.XML:
+        # NOTE: We're disabling the no-else-raise rule here because both comparisons are on the same enum so if one is
+        # removed then the other will raise an error. This is a false positive.
+        if resp.data_type == ResponseDataType.TXT:  # pylint: disable=no-else-raise
+            raise MMSServerError(config.name, resp.payload.decode("UTF-8"))
+        elif resp.data_type != ResponseDataType.XML:
             raise MMSClientError(
                 config.name,
                 f"Invalid MMS response data type: {resp.data_type.name}. Only XML is supported.",

{mms_client-1.5.1 → mms_client-1.6.0}/src/mms_client/services/market.py

@@ -18,6 +18,8 @@ from mms_client.types.market import MarketType
 from mms_client.types.offer import OfferCancel
 from mms_client.types.offer import OfferData
 from mms_client.types.offer import OfferQuery
+from mms_client.types.reserve import ReserveRequirement
+from mms_client.types.reserve import ReserveRequirementQuery
 from mms_client.types.transport import RequestType
 from mms_client.utils.serialization import SchemaType
 from mms_client.utils.serialization import Serializer
@@ -34,6 +36,35 @@ class MarketClientMixin:  # pylint: disable=unused-argument
     # The configuration for the market service
     config = ServiceConfiguration(Interface.MI, Serializer(SchemaType.MARKET, "MarketData"))
 
+    @mms_endpoint(
+        "MarketQuery_ReserveRequirementQuery",
+        config,
+        RequestType.INFO,
+        resp_envelope_type=MarketSubmit,
+        resp_data_type=ReserveRequirement,
+    )
+    def query_reserve_requirements(
+        self: ClientProto, request: ReserveRequirementQuery, date: Optional[Date] = None
+    ) -> List[ReserveRequirement]:
+        """Query the MMS server for reserve requirements.
+
+        This endpoint is accessible to all client types.
+
+        Arguments:
+        request (ReserveRequirementQuery):  The query to submit to the MMS server.
+        date (Date):                        The date of the transaction in the format "YYYY-MM-DD". This value defaults
+                                            to the current date.
+
+        Returns:    A list of reserve requirements that match the query.
+        """
+        # NOTE: The return type does not match the method definition but the decorator will return the correct type
+        return MarketQuery(  # type: ignore[return-value]
+            date=date or Date.today(),
+            participant=self.participant,
+            user=self.user,
+            days=1,
+        )
+
     @mms_endpoint("MarketSubmit_OfferData", config, RequestType.INFO, [ClientType.BSP])
     def put_offer(
         self: ClientProto, request: OfferData, market_type: MarketType, days: int, date: Optional[Date] = None
@@ -124,6 +155,8 @@ class MarketClientMixin:  # pylint: disable=unused-argument
         days (int):                 The number of days ahead for which the data is being cancelled.
         date (Date):                The date of the transaction in the format "YYYY-MM-DD". This value defaults to the
                                     current date.
+
+        Returns:    Data identifying the offer that was cancelled.
         """
         # NOTE: The return type does not match the method definition but the decorator will return the correct type
         return MarketCancel(  # type: ignore[return-value]

mms_client-1.6.0/src/mms_client/types/reserve.py

@@ -0,0 +1,71 @@
+"""Contains objects for MMS reserve requirements."""
+
+from typing import List
+from typing import Optional
+
+from pydantic_extra_types.pendulum_dt import DateTime
+from pydantic_xml import attr
+from pydantic_xml import element
+
+from mms_client.types.base import Payload
+from mms_client.types.enums import AreaCode
+from mms_client.types.enums import Direction
+from mms_client.types.fields import power_positive
+from mms_client.types.market import MarketType
+
+
+class Requirement(Payload):
+    """Represents a reserve requirement."""
+
+    # The start block of the requirement
+    start: DateTime = attr(name="StartTime")
+
+    # The end block of the requirement
+    end: DateTime = attr(name="EndTime")
+
+    # The direction of the requirement
+    direction: Direction = attr(name="Direction")
+
+    # The primary reserve quantity in kW
+    primary_qty_kw: Optional[int] = power_positive("PrimaryReserveQuantityInKw", True)
+
+    # The first secondary reserve quantity in kW
+    secondary_1_qty_kw: Optional[int] = power_positive("Secondary1ReserveQuantityInKw", True)
+
+    # The second secondary reserve quantity in kW
+    secondary_2_qty_kw: Optional[int] = power_positive("Secondary2ReserveQuantityInKw", True)
+
+    # The first tertiary reserve quantity in kW
+    tertiary_1_qty_kw: Optional[int] = power_positive("Tertiary1ReserveQuantityInKw", True)
+
+    # The second tertiary reserve quantity in kW
+    tertiary_2_qty_kw: Optional[int] = power_positive("Tertiary2ReserveQuantityInKw", True)
+
+    # The minimum reserve of compound primary and secondary 1 in kW
+    primary_secondary_1_qty_kw: Optional[int] = power_positive("CompoundPriSec1ReserveQuantityInKw", True)
+
+    # The minimum reserve of compound primary and secondary 2 in kW
+    primary_secondary_2_qty_kw: Optional[int] = power_positive("CompoundPriSec2ReserveQuantityInKw", True)
+
+    # The minimum reserve of compound primary and tertiary 1 in kW
+    primary_tertiary_1_qty_kw: Optional[int] = power_positive("CompoundPriTer1ReserveQuantityInKw", True)
+
+
+class ReserveRequirement(Payload):
+    """Represents a set of reserve requirements."""
+
+    # The area for which the reserve requirement applies
+    area: AreaCode = attr(name="Area")
+
+    # The requirements associated with the area
+    requirements: List[Requirement] = element(tag="Requirement", min_length=1)
+
+
+class ReserveRequirementQuery(Payload):
+    """Represents a request to query reserve requirements."""
+
+    # The market type for which to query reserve requirements
+    market_type: MarketType = attr(name="MarketType")
+
+    # The area for which to query reserve requirements
+    area: Optional[AreaCode] = attr(default=None, name="Area")

{mms_client-1.5.1 → mms_client-1.6.0}/src/mms_client/utils/errors.py

@@ -38,6 +38,21 @@ class AudienceError(ValueError):
         super().__init__(self.message)
 
 
+class MMSServerError(RuntimeError):
+    """Error raised when the MMS server returns an error."""
+
+    def __init__(self, method: str, message: str):
+        """Initialize the error.
+
+        Arguments:
+        method (str):   The method that caused the error.
+        message (str):  The error message.
+        """
+        super().__init__(f"{method}: {message}")
+        self.message = message
+        self.method = method
+
+
 class MMSClientError(RuntimeError):
     """Base class for MMS client errors."""