Marinho Brandão
16 years ago
10 changed files with 64 additions and 186 deletions
-
3pynfe/entidades/__init__.py
-
128pynfe/entidades/fontes_dados.py
-
2pynfe/entidades/notafiscal.py
-
43pynfe/processamento/serializacao.py
-
41tests/01-basico.txt
-
2tests/02-modelo-00-definicoes-gerais.txt
-
19tests/02-modelo-06-certificado.txt
-
5tests/03-processamento-00-definicoes-gerais.txt
-
5tests/03-processamento-01-serializacao-xml.txt
-
2tests/saida/README
@ -1,128 +0,0 @@ |
|||||
# -*- coding: utf-8 -*- |
|
||||
from pynfe.excecoes import NenhumObjetoEncontrado, MuitosObjetosEncontrados |
|
||||
|
|
||||
class FonteDados(object): |
|
||||
u"""Classe responsável por ser o repositório dos objetos em memória e que |
|
||||
pode ser extendida para persistir esses objetos. Também tem a função de |
|
||||
memorizar os objetos redundantes como um só e assim otimizar o desempenho.""" |
|
||||
|
|
||||
_objetos = None |
|
||||
|
|
||||
def __init__(self, objetos=None): |
|
||||
# Inicializa variável que armazena os objetos contidos na Fonte de Dados |
|
||||
if objetos: |
|
||||
self._objetos = objetos |
|
||||
else: |
|
||||
self._objetos = [] |
|
||||
|
|
||||
def carregar_objetos(self, **kwargs): |
|
||||
u"""Método responsavel por retornar os objetos que casem com os atributos |
|
||||
informados no argumento **kwargs (argumentos nomeados). |
|
||||
|
|
||||
Um argumento especial é o '_classe', que representa a classe da entidade |
|
||||
desejada. |
|
||||
|
|
||||
FIXME: Este algoritimo pode ser melhorado pra fazer pesquisas melhores, |
|
||||
mas por enquanto vamos nos focar no processo em geral para só depois nos |
|
||||
preocupar com otimizações e desempenho.""" |
|
||||
|
|
||||
# Função de filtro |
|
||||
def filtrar(obj): |
|
||||
ret = True |
|
||||
|
|
||||
for k,v in kwargs.items(): |
|
||||
# Filtra pela classe e pelos atributos |
|
||||
ret = (k == '_classe' and isinstance(obj, v)) or\ |
|
||||
(k != '_classe' and getattr(obj, k, None) == v) |
|
||||
|
|
||||
if not ret: |
|
||||
break |
|
||||
|
|
||||
return ret |
|
||||
|
|
||||
# Filtra a lista de objetos |
|
||||
lista = filter(filtrar, self._objetos) |
|
||||
|
|
||||
return lista |
|
||||
|
|
||||
def adicionar_objeto(self, _objeto): |
|
||||
u"""Método responsável por adicionar o(s) objeto(s) informado(s) ao |
|
||||
repositorio de objetos da fonte de dados.""" |
|
||||
|
|
||||
from base import Entidade |
|
||||
|
|
||||
# Adiciona _objeto como objeto |
|
||||
if isinstance(_objeto, Entidade): |
|
||||
self._objetos.append(_objeto) |
|
||||
|
|
||||
# Adiciona _objeto como lista |
|
||||
elif isinstance(_objeto, (list, tuple)): |
|
||||
self._objetos += _objeto |
|
||||
|
|
||||
else: |
|
||||
raise Exception('Objeto informado e invalido!') |
|
||||
|
|
||||
def remover_objeto(self, _objeto=None, **kwargs): |
|
||||
u"""Método responsavel por remover os objetos que casem com os atributos |
|
||||
informados no argumento **kwargs (argumentos nomeados). |
|
||||
|
|
||||
Um argumento especial é o '_classe', que representa a classe da entidade |
|
||||
desejada. |
|
||||
|
|
||||
Outro argumetno especial é o '_objeto', que representa o objeto a ser |
|
||||
removido. Caso o argumento _objeto seja uma lista de objetos, eles serão |
|
||||
removidos também.""" |
|
||||
|
|
||||
from base import Entidade |
|
||||
|
|
||||
lista = None |
|
||||
|
|
||||
# Remove objetos |
|
||||
if not _objeto: |
|
||||
lista = self.carregar_objetos(**kwargs) |
|
||||
|
|
||||
# Remove _objeto como objeto |
|
||||
elif isinstance(_objeto, Entidade): |
|
||||
lista = [_objeto] |
|
||||
|
|
||||
# Remove _objeto como objeto |
|
||||
elif isinstance(_objeto, (list, tuple)): |
|
||||
lista = _objeto |
|
||||
|
|
||||
else: |
|
||||
raise Exception('Objeto informado e invalido!') |
|
||||
|
|
||||
# Efetiva a remoção |
|
||||
for obj in lista: |
|
||||
self._objetos.remove(obj) |
|
||||
|
|
||||
def obter_objeto(self, **kwargs): |
|
||||
u"""Faz a ponte para o método 'carregar_objetos' mas obriga o retorno de |
|
||||
apenas um objeto, levantando exceção se nenhum for encontrado ou se forem |
|
||||
encontrados mais de um.""" |
|
||||
|
|
||||
lista = self.carregar_objetos(**kwargs) |
|
||||
|
|
||||
if len(lista) == 0: |
|
||||
raise NenhumObjetoEncontrado('Nenhum objeto foi encontrado!') |
|
||||
elif len(lista) > 1: |
|
||||
raise MuitosObjetosEncontrados('Muitos objetos foram encontrados!') |
|
||||
|
|
||||
return lista[0] |
|
||||
|
|
||||
def obter_lista(self, **kwargs): |
|
||||
u"""Método de proxy, que somente repassa a chamada ao metodo 'carregar_objetos'""" |
|
||||
return self.carregar_objetos(**kwargs) |
|
||||
|
|
||||
def contar_objetos(self, **kwargs): |
|
||||
u"""Método que repassa a chamada ao metodo 'carregar_objetos' mas retorna |
|
||||
somente a quantidade de objetos encontrados.""" |
|
||||
|
|
||||
if kwargs: |
|
||||
return len(self.carregar_objetos(**kwargs)) |
|
||||
else: |
|
||||
return len(self._objetos) |
|
||||
|
|
||||
# Instancia da fonte de dados default |
|
||||
_fonte_dados = FonteDados() |
|
||||
|
|
||||
@ -0,0 +1,2 @@ |
|||||
|
Esta pasta provavelmente vai permanecer vazia, e foi criada apenas para armazenar alguns |
||||
|
arquivos de saida da exportacao do serializador. |
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue