Remoção de código do cnab400

Author: danimaribeiro

cnab240/registro.py

@@ -5,12 +5,7 @@
 
 from glob import iglob
 from decimal import Decimal, InvalidOperation
-try:
-    from collections import OrderedDict
-except ImportError:
-    # Fallback for python 2.6
-    from ordereddict import OrderedDict
-
+from collections import OrderedDict
 from cnab240 import errors
 
 
@@ -168,10 +163,7 @@ def carregar(self, registro_str):
             if campo.decimais:
                 exponente = campo.decimais * -1
                 dec = valor[:exponente] + '.' + valor[exponente:]
-                try:
-                    campo.valor = Decimal(dec)
-                except InvalidOperation:
-                    raise  # raise custom?
+                campo.valor = Decimal(dec)
 
             elif campo.formato == 'num':
                 try:

cnab240/tipos.py

@@ -319,113 +319,3 @@ def incluir_pagamento(self, **kwargs):
         lote_pag.adicionar_evento(evento)
         # Incrementar numero de registros no trailer do arquivo
         self.trailer.totais_quantidade_registros += len(evento)
-
-
-class ArquivoCobranca400(object):
-
-    def __init__(self, banco, **kwargs):
-        """Arquivo Cnab400."""
-
-        self._lotes = []
-        self.banco = banco
-        arquivo = kwargs.get('arquivo')
-        if isinstance(arquivo, (IOBase, codecs.StreamReaderWriter)):
-            return self.carregar_retorno(arquivo)
-
-        self.header = self.banco.registros.HeaderArquivo(**kwargs)
-        self.trailer = self.banco.registros.TrailerArquivo(**kwargs)
-
-        if self.header.arquivo_data_de_geracao is None:
-            now = datetime.now()
-            self.header.arquivo_data_de_geracao = int(now.strftime("%d%m%Y"))
-
-    def carregar_retorno(self, arquivo):
-
-        lote_aberto = None
-        evento_aberto = None
-
-        for linha in arquivo:
-            tipo_registro = linha[0]
-
-            if tipo_registro == '0':
-                self.header = self.banco.registros.HeaderArquivo()
-                self.header.carregar(linha)
-                lote_aberto = Lote(self.banco)
-                self._lotes.append(lote_aberto)
-
-            elif tipo_registro == '1':
-                tipo_segmento = linha[0]
-                # codigo_evento = linha[15:17]
-
-                if tipo_segmento == '1':
-                    trans_tipo1 = self.banco.registros.TransacaoTipo1()
-                    trans_tipo1.carregar(linha)
-
-                    evento_aberto = Evento(self.banco, 1)
-                    lote_aberto._eventos.append(evento_aberto)
-                    evento_aberto._segmentos.append(trans_tipo1)
-
-                    evento_aberto = None
-
-            elif tipo_registro == '9':
-                self.trailer = self.banco.registros.TrailerArquivo()
-                self.trailer.carregar(linha)
-
-    @property
-    def lotes(self):
-        return self._lotes
-
-    def incluir_cobranca(self, **kwargs):
-        # 1 eh o codigo de cobranca
-        codigo_evento = 1
-        evento = Evento(self.banco, codigo_evento)
-
-        trans_tp1 = self.banco.registros.TransacaoTipo1(**kwargs)
-        evento.adicionar_segmento(trans_tp1)
-
-        lote_cobranca = self.encontrar_lote(codigo_evento)
-
-        if lote_cobranca is None:
-            header = None
-            trailer = None
-            lote_cobranca = Lote(self.banco, header, trailer)
-            self.adicionar_lote(lote_cobranca)
-
-        lote_cobranca.adicionar_evento(evento)
-
-    def encontrar_lote(self, codigo_servico):
-        for lote in self.lotes:
-            # if lote.header.codigo_servico == codigo_servico:
-            #     return lote
-            return lote
-
-    def adicionar_lote(self, lote):
-        if not isinstance(lote, Lote):
-            raise TypeError('Objeto deve ser instancia de "Lote"')
-
-        self._lotes.append(lote)
-        lote.codigo = len(self._lotes)
-
-        if self.trailer != None:
-            if hasattr(self.trailer, 'totais_quantidade_lotes'):
-                # Incrementar numero de lotes no trailer do arquivo
-                self.trailer.totais_quantidade_lotes += 1
-
-            if hasattr(self.trailer, 'totais_quantidade_registros'):
-                # Incrementar numero de registros no trailer do arquivo
-                self.trailer.totais_quantidade_registros += len(lote)
-
-    def escrever(self, file_):
-        file_.write(str(self).encode('ascii'))
-
-    def __str__(self):
-        if not self._lotes:
-            raise errors.ArquivoVazioError()
-
-        result = []
-        result.append(str(self.header))
-        result.extend(str(lote) for lote in self._lotes)
-        result.append(str(self.trailer))
-        # Adicionar elemento vazio para arquivo terminar com \r\n
-        result.append('')
-        return '\r\n'.join(result)