idecomp 1.3.0__tar.gz → 1.4.0__tar.gz

{idecomp-1.3.0 → idecomp-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: idecomp
-Version: 1.3.0
+Version: 1.4.0
 Summary: Interface para arquivos do DECOMP
 Home-page: https://github.com/rjmalves/idecomp
 Author: Rogerio Alves
@@ -14,7 +14,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE.md
 Requires-Dist: numpy
 Requires-Dist: pandas
-Requires-Dist: cfinterface>=1.5.3
+Requires-Dist: cfinterface>=1.7.0
 
 # idecomp
 

{idecomp-1.3.0 → idecomp-1.4.0}/idecomp/__init__.py

@@ -6,6 +6,6 @@ idecomp é um módulo em Python para manipulação dos arquivos de entrada
 e saída do DECOMP.
 """
 
-__version__ = "1.3.0"
+__version__ = "1.4.0"
 
 from . import decomp  # noqa

{idecomp-1.3.0 → idecomp-1.4.0}/idecomp/config.py

@@ -48,3 +48,4 @@ MESES_ABREV = [
     "NOV",
     "DEZ",
 ]
+FORMATO_CAMPOS_DATA_LIBS = ["%Y/%m/%d %H:%M", "%Y/%m/%d", "%Y/%m"]

{idecomp-1.3.0 → idecomp-1.4.0}/idecomp/decomp/dadger.py

@@ -84,6 +84,7 @@ from idecomp.decomp.modelos.dadger import (
     VL,
     VA,
     VU,
+    DA,
 )
 
 import pandas  # type: ignore
@@ -211,6 +212,7 @@ class Dadger(RegisterFile):
         VL,
         VA,
         VU,
+        DA,
     ]
 
     def __init__(self, data=...) -> None:
@@ -1000,6 +1002,35 @@ class Dadger(RegisterFile):
         """
         return self.__registros_ou_df(TI, codigo_usina=codigo_usina, df=df)
 
+    def da(
+        self,
+        codigo_usina_retirada: Optional[int] = None,
+        codigo_usina_retorno: Optional[int] = None,
+        df: bool = False,
+    ) -> Optional[Union[DA, List[DA], pandas.DataFrame]]:
+        """
+        Obtém um registro que especifica as retiradas de água para
+        outros usos (desvios de água) por usina (UHE) existente no
+        estudo especificado no :class:`Dadger`
+
+        :param codigo_usina_retirada: Código da UHE a montante da qual será feita a retirada
+        :type codigo_usina_retirada: int | None
+        :param codigo_usina_retorno: Código da UHE a montante da qual se derá o retorno.
+        :type codigo_usina_retorno: int | None
+        :param df: ignorar os filtros e retornar
+            todos os dados de registros como um DataFrame
+        :type df: bool
+
+        :return: Um ou mais registros, se existirem.
+        :rtype: :class:`DA` | list[:class:`DA`] | :class:`pandas.DataFrame` | None
+        """
+        return self.__registros_ou_df(
+            DA,
+            codigo_usina_retirada=codigo_usina_retirada,
+            codigo_usina_retorno=codigo_usina_retorno,
+            df=df,
+        )
+
     def mp(
         self,
         codigo_usina: Optional[int] = None,

{idecomp-1.3.0 → idecomp-1.4.0}/idecomp/decomp/modelos/arquivos.py

@@ -11,6 +11,8 @@ class BlocoNomesArquivos(Section):
     arquivos de entrada do DECOMP.
     """
 
+    __slots__ = ["__linha"]
+
     def __init__(self, previous=None, next=None, data=None) -> None:
         super().__init__(previous, next, data)
         self.__linha = Line([LiteralField(80, 0)])

{idecomp-1.3.0 → idecomp-1.4.0}/idecomp/decomp/modelos/avl_cortesfpha_dec.py

@@ -13,6 +13,8 @@ class TabelaCortesFpha(TabelaCSV):
     Bloco com as informações dos cortes da função de produção hidráulica.
     """
 
+    __slots__ = []
+
     BEGIN_PATTERN = "-----;-----;--------------;"
     LINE_MODEL = Line(
         [

{idecomp-1.3.0 → idecomp-1.4.0}/idecomp/decomp/modelos/blocos/tabelacsv.py

@@ -11,6 +11,8 @@ class TabelaCSV(Block):
     a partir de um modelo de linha, para arquivos de saída do DECOMP.
     """
 
+    __slots__ = []
+
     BEGIN_PATTERN = "-----;------;"
     LINE_MODEL = Line([])
     COLUMN_NAMES: List[str] = []
@@ -66,6 +68,8 @@ class TabelaCSVLibs(Block):
     a partir de um modelo de linha, para arquivos de saída do DECOMP.
     """
 
+    __slots__ = []
+
     BEGIN_PATTERN = "&IIIIIII;IIIIIII;"
     LINE_MODEL = Line([])
     COLUMN_NAMES: List[str] = []

{idecomp-1.3.0 → idecomp-1.4.0}/idecomp/decomp/modelos/blocos/versaomodelo.py

@@ -11,6 +11,8 @@ class VersaoModelo(Block):
     título do arquivo.
     """
 
+    __slots__ = []
+
     BEGIN_PATTERN = r"CEPEL: DECOMP"
     END_PATTERN = ""
 
@@ -34,6 +36,8 @@ class VersaoModeloLibs(Block):
     título do arquivo.
     """
 
+    __slots__ = []
+
     BEGIN_PATTERN = r"Nome do Modelo: DECOMP Versão:"
     END_PATTERN = ""
 

{idecomp-1.3.0 → idecomp-1.4.0}/idecomp/decomp/modelos/caso.py

@@ -10,6 +10,8 @@ class NomeCaso(Section):
     entrada do DECOMP `caso.dat`.
     """
 
+    __slots__ = ["__linha"]
+
     def __init__(self, previous=None, next=None, data=None) -> None:
         super().__init__(previous, next, data)
         self.__linha = Line([LiteralField(80, 0)])

{idecomp-1.3.0 → idecomp-1.4.0}/idecomp/decomp/modelos/custos.py

@@ -13,6 +13,14 @@ from typing import IO, List
 class BlocoRelatorioCustos(Block):
     """ """
 
+    __slots__ = [
+        "__scenario_line",
+        "__dual_variables_line",
+        "__fcf_line",
+        "dados_cenario",
+        "data",
+    ]
+
     BEGIN_PATTERN = r" RELATORIO DAS VARIAVEIS DUAIS NO ESTAGIO"
     END_PATTERN = "X----X-"