#1047 change BO Energiemenge and use it instead of COM Verbrauch

Author: Simon Blank

src/bo4e/bo/energiemenge.py

@@ -11,8 +11,9 @@
 from .geschaeftsobjekt import Geschaeftsobjekt
 
 if TYPE_CHECKING:
+    from ..com.menge import Menge
     from ..com.verbrauch import Verbrauch
-    from ..enum.lokationstyp import Lokationstyp
+    from ..com.zeitraum import Zeitraum
 
 
 # pylint: disable=too-few-public-methods
@@ -34,11 +35,11 @@ class Energiemenge(Geschaeftsobjekt):
     """
 
     typ: Annotated[Literal[BoTyp.ENERGIEMENGE], Field(alias="_typ")] = BoTyp.ENERGIEMENGE
-    lokations_id: Optional[str] = None
-    """Eindeutige Nummer der Marktlokation bzw. der Messlokation, zu der die Energiemenge gehört"""
-    lokationstyp: Optional["Lokationstyp"] = None
-    """Gibt an, ob es sich um eine Markt- oder Messlokation handelt"""
-
-    energieverbrauch: Optional[list["Verbrauch"]] = None
-    """Gibt den Verbrauch in einer Zeiteinheit an"""
-    # there are no optional attributes
+    obis_kennzahl: Optional[str] = None
+    """Die OBIS-Kennzahl der Energiemenge"""
+    beschreibung: Optional[str] = None
+    """Ergänzende Beschreibung zur Energiemenge"""
+    zeitraum: Optional["Zeitraum"] = None
+    """Zeitraum, in dem die Energiemenge angefallen ist/gemessen wurde"""
+    menge: Optional["Menge"] = None
+    """Die angefallene/gemessene Menge"""

src/bo4e/bo/marktlokation.py

@@ -12,10 +12,10 @@
 from .geschaeftsobjekt import Geschaeftsobjekt
 
 if TYPE_CHECKING:
+    from ..bo.energiemenge import Energiemenge
     from ..com.adresse import Adresse
     from ..com.geokoordinaten import Geokoordinaten
     from ..com.katasteradresse import Katasteradresse
-    from ..com.verbrauch import Verbrauch
     from ..com.zaehlwerk import Zaehlwerk
     from ..enum.bilanzierungsmethode import Bilanzierungsmethode
     from ..enum.energierichtung import Energierichtung
@@ -103,7 +103,7 @@ class Marktlokation(Geschaeftsobjekt):
     marktgebiet: Optional[str] = None
     """für Gas. Code vom EIC, https://www.entsog.eu/data/data-portal/codes-list"""
     zaehlwerke: Optional[list["Zaehlwerk"]] = None
-    verbrauchsmengen: Optional[list["Verbrauch"]] = None
+    verbrauchsmengen: Optional[list["Energiemenge"]] = None
     zaehlwerke_der_beteiligten_marktrolle: Optional[list["Zaehlwerk"]] = None
 
     lokationszuordnungen: Optional[list["Lokationszuordnung"]] = None

tests/test_energiemenge.py

@@ -1,6 +1,7 @@
 import pytest
 
-from bo4e import Energiemenge, Lokationstyp, Verbrauch
+from bo4e import Energiemenge, Menge, Zeitraum
+
 from tests.serialization_helper import assert_serialization_roundtrip
 
 
@@ -10,9 +11,10 @@ class TestEnergiemenge:
         [
             pytest.param(
                 Energiemenge(
-                    lokations_id="DE0123456789012345678901234567890",
-                    lokationstyp=Lokationstyp.MELO,
-                    energieverbrauch=[Verbrauch()],
+                    obis_kennzahl="1-0:1.8.1",
+                    beschreibung="Eine Beschreibung",
+                    zeitraum=Zeitraum(),
+                    menge=Menge(),
                 ),
                 id="all attributes at first level",
             ),

tests/test_marktlokation.py

@@ -3,6 +3,7 @@
 from bo4e import (
     Adresse,
     Bilanzierungsmethode,
+    Energiemenge,
     Energierichtung,
     Geschaeftspartner,
     Kundentyp,
@@ -29,6 +30,7 @@ class TestMaLo:
                     netzebene=Netzebene.NSP,
                     endkunde=Geschaeftspartner(),
                     kundengruppen=[Kundentyp.GEWERBE, Kundentyp.PRIVAT],
+                    verbrauchsmengen=[Energiemenge()],
                     lokationsbuendel_objektcode="9992 00000 125 6",
                     lokationszuordnungen=[Lokationszuordnung()],
                 )