inewave 1.5.2__tar.gz → 1.5.4__tar.gz

{inewave-1.5.2/inewave.egg-info → inewave-1.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: inewave
-Version: 1.5.2
+Version: 1.5.4
 Summary: Interface para arquivos do NEWAVE
 Home-page: https://github.com/rjmalves/inewave
 Author: Rogerio Alves

{inewave-1.5.2 → inewave-1.5.4}/inewave/__init__.py

@@ -6,7 +6,7 @@ inewave é um módulo em Python para manipulação dos arquivos de entrada
 e saída do NEWAVE.
 """
 
-__version__ = "1.5.2"
+__version__ = "1.5.4"
 
 from . import newave  # noqa
 from . import nwlistcf  # noqa

{inewave-1.5.2 → inewave-1.5.4}/inewave/newave/modelos/modif.py

@@ -1,11 +1,68 @@
 from cfinterface.components.register import Register
 from cfinterface.components.line import Line
+from cfinterface.components.field import Field
 from cfinterface.components.integerfield import IntegerField
 from cfinterface.components.literalfield import LiteralField
 from cfinterface.components.floatfield import FloatField
-from cfinterface.components.datetimefield import DatetimeField
+from cfinterface.adapters.components.repository import factory
+from copy import deepcopy
 from datetime import datetime
-from typing import Optional
+from typing import Optional, IO
+
+
+class ModifRegister(Register):
+    """
+    Registro base para o arquivo modif.dat, que contém
+    características específicas deste arquivo:
+
+    - Possibilidade de ler uma linha assumindo que o conteúdo
+        de cada campo é separado por um número variável
+        de espaços.
+    - Capacidade de escrever uma linha assumindo uma formatação
+        constante para melhor visualização.
+
+    OBS: Atualmente só utilizados para registros VOLMIN e VOLMAX.
+    """
+
+    def __init__(
+        self,
+        previous=None,
+        next=None,
+        data=None,
+    ) -> None:
+        self.__identifier_field: Field = LiteralField(
+            self.__class__.IDENTIFIER_DIGITS, 0
+        )
+        self.__previous = previous
+        self.__next = next
+        if data is None:
+            self.__data = [None] * len(self.__class__.LINE.fields)
+        else:
+            self.__data = data
+
+    def read(self, file: IO, storage: str = "", *args, **kwargs) -> bool:
+        delimited_fields = [deepcopy(f) for f in self.__class__.LINE.fields]
+        line = Line(
+            [self.__identifier_field] + delimited_fields,
+            delimiter=" ",
+            storage=storage,
+        )
+        line_full_spaces = file.readline().strip()
+        line_parts = [p for p in line_full_spaces.split(" ") if len(p) > 0]
+        line_simple_spaces = " ".join(line_parts)
+        self.data = line.read(line_simple_spaces)[1:]
+        return True
+
+    def write(self, file: IO, storage: str = "", *args, **kwargs) -> bool:
+        if not self.empty:
+            line = Line(
+                [self.__identifier_field] + self.__class__.LINE.fields,
+                delimiter=self.__class__.LINE.delimiter,
+                storage=storage,
+            )
+            linedata = line.write([self.__class__.IDENTIFIER] + self.data)
+            factory(storage).write(file, linedata)
+        return True
 
 
 class USINA(Register):
@@ -46,7 +103,7 @@ class USINA(Register):
         self.data[1] = t
 
 
-class VOLMIN(Register):
+class VOLMIN(ModifRegister):
     """
     Registro que contém uma modificação de volume mínimo
     operativo para uma usina.
@@ -85,7 +142,7 @@ class VOLMIN(Register):
         self.data[1] = t
 
 
-class VOLMAX(Register):
+class VOLMAX(ModifRegister):
     """
     Registro que contém uma modificação de volume máximo
     operativo para uma usina.
@@ -124,7 +181,7 @@ class VOLMAX(Register):
         self.data[1] = t
 
 
-class NUMCNJ(Register):
+class NUMCNJ(ModifRegister):
     """
     Registro que contém uma modificação de número de conjunto
     de máquinas.
@@ -132,7 +189,7 @@ class NUMCNJ(Register):
 
     IDENTIFIER = " NUMCNJ"
     IDENTIFIER_DIGITS = 8
-    LINE = Line([IntegerField(3, 10)])
+    LINE = Line([IntegerField(2, 11)])
 
     @property
     def numero(self) -> int:
@@ -149,7 +206,7 @@ class NUMCNJ(Register):
         self.data[0] = t
 
 
-class NUMMAQ(Register):
+class NUMMAQ(ModifRegister):
     """
     Registro que contém uma modificação do número de máquinas em um
     conjunto de máquinas.
@@ -157,7 +214,7 @@ class NUMMAQ(Register):
 
     IDENTIFIER = " NUMMAQ"
     IDENTIFIER_DIGITS = 8
-    LINE = Line([IntegerField(3, 10), IntegerField(3, 14)])
+    LINE = Line([IntegerField(3, 11), IntegerField(3, 14)])
 
     @property
     def conjunto(self) -> Optional[int]:
@@ -188,7 +245,7 @@ class NUMMAQ(Register):
         self.data[1] = t
 
 
-class VAZMIN(Register):
+class VAZMIN(ModifRegister):
     """
     Registro que contém uma modificação de vazão mínima (m3/s).
     """
@@ -212,14 +269,16 @@ class VAZMIN(Register):
         self.data[0] = t
 
 
-class CFUGA(Register):
+class CFUGA(ModifRegister):
     """
     Registro que contém uma modificação do nível do canal de fuga.
     """
 
     IDENTIFIER = " CFUGA"
     IDENTIFIER_DIGITS = 8
-    LINE = Line([DatetimeField(7, 10, format="%m %Y"), FloatField(7, 18, 3)])
+    LINE = Line(
+        [IntegerField(2, 10), IntegerField(4, 13), FloatField(7, 18, 3)]
+    )
 
     @property
     def data_inicio(self) -> datetime:
@@ -229,11 +288,12 @@ class CFUGA(Register):
         :return: A data de início da modificação
         :rtype: Optional[datetime]
         """
-        return self.data[0]
+        return datetime(self.data[1], self.data[0], 1)
 
     @data_inicio.setter
     def data_inicio(self, t: datetime):
-        self.data[0] = t
+        self.data[0] = t.month
+        self.data[1] = t.year
 
     @property
     def nivel(self) -> float:
@@ -243,21 +303,23 @@ class CFUGA(Register):
         :return: O novo nível
         :rtype: Optional[int]
         """
-        return self.data[1]
+        return self.data[2]
 
     @nivel.setter
     def nivel(self, t: float):
-        self.data[1] = t
+        self.data[2] = t
 
 
-class CMONT(Register):
+class CMONT(ModifRegister):
     """
     Registro que contém uma modificação do nível do canal de montante.
     """
 
     IDENTIFIER = " CMONT"
     IDENTIFIER_DIGITS = 8
-    LINE = Line([DatetimeField(7, 10, format="%m %Y"), FloatField(7, 18, 3)])
+    LINE = Line(
+        [IntegerField(2, 10), IntegerField(4, 13), FloatField(7, 18, 3)]
+    )
 
     @property
     def data_inicio(self) -> datetime:
@@ -267,11 +329,12 @@ class CMONT(Register):
         :return: A data de início da modificação
         :rtype: Optional[datetime]
         """
-        return self.data[0]
+        return datetime(self.data[1], self.data[0], 1)
 
     @data_inicio.setter
     def data_inicio(self, t: datetime):
-        self.data[0] = t
+        self.data[0] = t.month
+        self.data[1] = t.year
 
     @property
     def nivel(self) -> Optional[float]:
@@ -281,11 +344,11 @@ class CMONT(Register):
         :return: O novo nível
         :rtype: Optional[float]
         """
-        return self.data[1]
+        return self.data[2]
 
     @nivel.setter
     def nivel(self, t: float):
-        self.data[1] = t
+        self.data[2] = t
 
 
 class VMAXT(Register):
@@ -298,7 +361,8 @@ class VMAXT(Register):
     IDENTIFIER_DIGITS = 8
     LINE = Line(
         [
-            DatetimeField(7, 10, format="%m %Y"),
+            IntegerField(2, 10),
+            IntegerField(4, 13),
             FloatField(7, 18, 3),
             LiteralField(3, 26),
         ]
@@ -312,11 +376,12 @@ class VMAXT(Register):
         :return: A data de início da modificação
         :rtype: Optional[datetime]
         """
-        return self.data[0]
+        return datetime(self.data[1], self.data[0], 1)
 
     @data_inicio.setter
     def data_inicio(self, t: datetime):
-        self.data[0] = t
+        self.data[0] = t.month
+        self.data[1] = t.year
 
     @property
     def volume(self) -> Optional[float]:
@@ -326,11 +391,11 @@ class VMAXT(Register):
         :return: O novo volume
         :rtype: Optional[float]
         """
-        return self.data[1]
+        return self.data[2]
 
     @volume.setter
     def volume(self, t: float):
-        self.data[1] = t
+        self.data[2] = t
 
     @property
     def unidade(self) -> Optional[str]:
@@ -340,11 +405,11 @@ class VMAXT(Register):
         :return: A unidade
         :rtype: Optional[str]
         """
-        return self.data[2]
+        return self.data[3]
 
     @unidade.setter
     def unidade(self, t: str):
-        self.data[2] = t
+        self.data[3] = t
 
 
 class VMINT(Register):
@@ -357,7 +422,8 @@ class VMINT(Register):
     IDENTIFIER_DIGITS = 8
     LINE = Line(
         [
-            DatetimeField(7, 10, format="%m %Y"),
+            IntegerField(2, 10),
+            IntegerField(4, 13),
             FloatField(7, 18, 3),
             LiteralField(3, 26),
         ]
@@ -371,11 +437,12 @@ class VMINT(Register):
         :return: A data de início da modificação
         :rtype: Optional[datetime]
         """
-        return self.data[0]
+        return datetime(self.data[1], self.data[0], 1)
 
     @data_inicio.setter
     def data_inicio(self, t: datetime):
-        self.data[0] = t
+        self.data[0] = t.month
+        self.data[1] = t.year
 
     @property
     def volume(self) -> Optional[float]:
@@ -385,11 +452,11 @@ class VMINT(Register):
         :return: O novo volume
         :rtype: Optional[float]
         """
-        return self.data[1]
+        return self.data[2]
 
     @volume.setter
     def volume(self, t: float):
-        self.data[1] = t
+        self.data[2] = t
 
     @property
     def unidade(self) -> Optional[str]:
@@ -399,11 +466,11 @@ class VMINT(Register):
         :return: A unidade
         :rtype: Optional[str]
         """
-        return self.data[2]
+        return self.data[3]
 
     @unidade.setter
     def unidade(self, t: str):
-        self.data[2] = t
+        self.data[3] = t
 
 
 class VMINP(Register):
@@ -416,7 +483,8 @@ class VMINP(Register):
     IDENTIFIER_DIGITS = 8
     LINE = Line(
         [
-            DatetimeField(7, 10, format="%m %Y"),
+            IntegerField(2, 10),
+            IntegerField(4, 13),
             FloatField(7, 18, 3),
             LiteralField(3, 26),
         ]
@@ -430,11 +498,12 @@ class VMINP(Register):
         :return: A data de início da modificação
         :rtype: Optional[datetime]
         """
-        return self.data[0]
+        return datetime(self.data[1], self.data[0], 1)
 
     @data_inicio.setter
     def data_inicio(self, t: datetime):
-        self.data[0] = t
+        self.data[0] = t.month
+        self.data[1] = t.year
 
     @property
     def volume(self) -> Optional[float]:
@@ -444,11 +513,11 @@ class VMINP(Register):
         :return: O novo volume
         :rtype: Optional[float]
         """
-        return self.data[1]
+        return self.data[2]
 
     @volume.setter
     def volume(self, t: float):
-        self.data[1] = t
+        self.data[2] = t
 
     @property
     def unidade(self) -> Optional[str]:
@@ -458,11 +527,11 @@ class VMINP(Register):
         :return: A unidade
         :rtype: Optional[str]
         """
-        return self.data[2]
+        return self.data[3]
 
     @unidade.setter
     def unidade(self, t: str):
-        self.data[2] = t
+        self.data[3] = t
 
 
 class VAZMINT(Register):
@@ -475,7 +544,8 @@ class VAZMINT(Register):
     IDENTIFIER_DIGITS = 8
     LINE = Line(
         [
-            DatetimeField(7, 10, format="%m %Y"),
+            IntegerField(2, 10),
+            IntegerField(4, 13),
             FloatField(7, 18, 2),
         ]
     )
@@ -488,11 +558,12 @@ class VAZMINT(Register):
         :return: A data de início da modificação
         :rtype: Optional[datetime]
         """
-        return self.data[0]
+        return datetime(self.data[1], self.data[0], 1)
 
     @data_inicio.setter
     def data_inicio(self, t: datetime):
-        self.data[0] = t
+        self.data[0] = t.month
+        self.data[1] = t.year
 
     @property
     def vazao(self) -> Optional[float]:
@@ -502,11 +573,11 @@ class VAZMINT(Register):
         :return: A nova vazão
         :rtype: Optional[float]
         """
-        return self.data[1]
+        return self.data[2]
 
     @vazao.setter
     def vazao(self, t: float):
-        self.data[1] = t
+        self.data[2] = t
 
 
 class VAZMAXT(Register):
@@ -519,7 +590,8 @@ class VAZMAXT(Register):
     IDENTIFIER_DIGITS = 8
     LINE = Line(
         [
-            DatetimeField(7, 10, format="%m %Y"),
+            IntegerField(2, 10),
+            IntegerField(4, 13),
             FloatField(7, 18, 2),
         ]
     )
@@ -532,11 +604,12 @@ class VAZMAXT(Register):
         :return: A data de início da modificação
         :rtype: Optional[datetime]
         """
-        return self.data[0]
+        return datetime(self.data[1], self.data[0], 1)
 
     @data_inicio.setter
     def data_inicio(self, t: datetime):
-        self.data[0] = t
+        self.data[0] = t.month
+        self.data[1] = t.year
 
     @property
     def vazao(self) -> Optional[float]:
@@ -546,11 +619,11 @@ class VAZMAXT(Register):
         :return: A nova vazão
         :rtype: Optional[float]
         """
-        return self.data[1]
+        return self.data[2]
 
     @vazao.setter
     def vazao(self, t: float):
-        self.data[1] = t
+        self.data[2] = t
 
 
 class TURBMAXT(Register):
@@ -563,7 +636,8 @@ class TURBMAXT(Register):
     IDENTIFIER_DIGITS = 9
     LINE = Line(
         [
-            DatetimeField(7, 10, format="%m %Y"),
+            IntegerField(2, 10),
+            IntegerField(4, 13),
             FloatField(7, 18, 2),
         ]
     )
@@ -576,11 +650,12 @@ class TURBMAXT(Register):
         :return: A data de início da modificação
         :rtype: Optional[datetime]
         """
-        return self.data[0]
+        return datetime(self.data[1], self.data[0], 1)
 
     @data_inicio.setter
     def data_inicio(self, t: datetime):
-        self.data[0] = t
+        self.data[0] = t.month
+        self.data[1] = t.year
 
     @property
     def turbinamento(self) -> Optional[float]:
@@ -590,11 +665,11 @@ class TURBMAXT(Register):
         :return: O novo turbinamento
         :rtype: Optional[float]
         """
-        return self.data[1]
+        return self.data[2]
 
     @turbinamento.setter
     def turbinamento(self, t: float):
-        self.data[1] = t
+        self.data[2] = t
 
 
 class TURBMINT(Register):
@@ -607,7 +682,8 @@ class TURBMINT(Register):
     IDENTIFIER_DIGITS = 9
     LINE = Line(
         [
-            DatetimeField(7, 10, format="%m %Y"),
+            IntegerField(2, 10),
+            IntegerField(4, 13),
             FloatField(7, 18, 2),
         ]
     )
@@ -620,11 +696,12 @@ class TURBMINT(Register):
         :return: A data de início da modificação
         :rtype: Optional[datetime]
         """
-        return self.data[0]
+        return datetime(self.data[1], self.data[0], 1)
 
     @data_inicio.setter
     def data_inicio(self, t: datetime):
-        self.data[0] = t
+        self.data[0] = t.month
+        self.data[1] = t.year
 
     @property
     def turbinamento(self) -> Optional[float]:
@@ -634,8 +711,8 @@ class TURBMINT(Register):
         :return: O novo turbinamento
         :rtype: Optional[float]
         """
-        return self.data[1]
+        return self.data[2]
 
     @turbinamento.setter
     def turbinamento(self, t: float):
-        self.data[1] = t
+        self.data[2] = t

{inewave-1.5.2 → inewave-1.5.4/inewave.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: inewave
-Version: 1.5.2
+Version: 1.5.4
 Summary: Interface para arquivos do NEWAVE
 Home-page: https://github.com/rjmalves/inewave
 Author: Rogerio Alves

{inewave-1.5.2 → inewave-1.5.4}/tests/newave/test_modif.py

@@ -75,7 +75,7 @@ def test_registro_vmaxt_modif():
         with open("", "") as fp:
             r.read(fp)
 
-    assert r.data == [datetime(2021, 12, 1), 61.310, "'%'"]
+    assert r.data == [12, 2021, 61.310, "'%'"]
     assert r.data_inicio == datetime(2021, 12, 1)
     assert r.volume == 61.310
     assert r.unidade == "'%'"
@@ -88,7 +88,7 @@ def test_registro_vazmint_modif():
         with open("", "") as fp:
             r.read(fp)
 
-    assert r.data == [datetime(2021, 10, 1), 10.00]
+    assert r.data == [10, 2021, 10.00]
     assert r.data_inicio == datetime(2021, 10, 1)
     assert r.vazao == 10.00
 
@@ -146,7 +146,7 @@ def test_registro_vmint_modif():
         with open("", "") as fp:
             r.read(fp)
 
-    assert r.data == [datetime(2021, 10, 1), 20.0, "'%'"]
+    assert r.data == [10, 2021, 20.0, "'%'"]
     assert r.data_inicio == datetime(2021, 10, 1)
     assert r.volume == 20.0
     assert r.unidade == "'%'"
@@ -159,7 +159,7 @@ def test_registro_vminp_modif():
         with open("", "") as fp:
             r.read(fp)
 
-    assert r.data == [datetime(2021, 10, 1), 20.0, "'%'"]
+    assert r.data == [10, 2021, 20.0, "'%'"]
     assert r.data_inicio == datetime(2021, 10, 1)
     assert r.volume == 20.0
     assert r.unidade == "'%'"
@@ -172,7 +172,7 @@ def test_registro_cfuga_modif():
         with open("", "") as fp:
             r.read(fp)
 
-    assert r.data == [datetime(2021, 10, 1), 5.60]
+    assert r.data == [10, 2021, 5.60]
     assert r.data_inicio == datetime(2021, 10, 1)
     assert r.nivel == 5.60
 
@@ -184,7 +184,7 @@ def test_registro_cmont_modif():
         with open("", "") as fp:
             r.read(fp)
 
-    assert r.data == [datetime(2021, 10, 1), 71.30]
+    assert r.data == [10, 2021, 71.30]
     assert r.data_inicio == datetime(2021, 10, 1)
     assert r.nivel == 71.30
 
@@ -196,7 +196,7 @@ def test_registro_vazmaxt_modif():
         with open("", "") as fp:
             r.read(fp)
 
-    assert r.data == [datetime(2020, 1, 1), 0.00]
+    assert r.data == [1, 2020, 0.00]
     assert r.data_inicio == datetime(2020, 1, 1)
     assert r.vazao == 0.0
 
@@ -208,7 +208,7 @@ def test_registro_turbmaxt_modif():
         with open("", "") as fp:
             r.read(fp)
 
-    assert r.data == [datetime(2020, 1, 1), 0.00]
+    assert r.data == [1, 2020, 0.00]
     assert r.data_inicio == datetime(2020, 1, 1)
     assert r.turbinamento == 0.0
 
@@ -220,7 +220,7 @@ def test_registro_turbmint_modif():
         with open("", "") as fp:
             r.read(fp)
 
-    assert r.data == [datetime(2020, 1, 1), 0.00]
+    assert r.data == [1, 2020, 0.00]
     assert r.data_inicio == datetime(2020, 1, 1)
     assert r.turbinamento == 0.0
 
@@ -258,6 +258,7 @@ def test_leitura_escrita_modif():
         linhas_escritas = [
             chamadas[i].args[0] for i in range(1, len(chamadas) - 1)
         ]
+
     m_releitura: MagicMock = mock_open(read_data="".join(linhas_escritas))
     with patch("builtins.open", m_releitura):
         cf2 = Modif.read(ARQ_TESTE)