###################################################################################
# configurando novos padrões prontos de regex
from pesquisabr import UtilExtracaoRe
novos_prontos = {'OAB' : r'\b([a-z]{2}[\.\-,; ]?\d{2,7}|\d{2,7}[\.\-,; ]?[a-z]{2})\b'}
UtilExtracaoRe.PRONTOS.update(novos_prontos)

###################################################################################
# define o tipo de objeto de regras que será carregado
# pode-se criar um tipo personalizado carregando do banco ou de onde for necessário
from regras_model import RegrasModelArquivo
obj_regras_model = RegrasModelArquivo()

class RegrasModelBase():
    # sobrescrever esse método para retornar uma lista de regras 
    # pode carregar do banco, do disco, de outro serviço, etc
    # [{"grupo" : "nome_grupo", "rotulo": "rotulo1", "regra": "critérios da regra", "tags": "receita bolo", "qtd_cabecalho":0, "qtd_rodape":0, "ordem": 0},]
    # incluir algum filtro com a chave filtro_tipo facilita testes na tela exemplo do serviço
    def get_regras_db(self):
        return []

    # pode-se transforar os dados de saída do serviço com base em alguma
    # regra específica do contexto da aplicação
    # pode-se tratar a chave regras com as regras aplicadas, a chave rótulos, etc
    def conversao_retorno(self, retorno: dict):
        pass 

    # pode-se transforar os dados de entrada no serviço com base em alguma
    # regra específica do contexto da aplicação
    # pode-se converter dados ou incluir outras regras de negócio específicas
    # front_end, analisar_regras, analisar_criterios indicam o contexto dos dados de entrada
    def conversao_entrada(self, dados: dict, front_end = False, analisar_regras=False, analisar_criterios=False):
        pass 
...
    
# Exemplo de personalização da classe     
class RegrasModelArquivo(RegrasModelBase):
    ARQ_REGRAS = './regras.json'   # arquivo texto no formato [{regra}, {regra}, {regra}]
    def get_regras_db(self):
        regras = []
        if os.path.isfile(self.ARQ_REGRAS):
            with open(self.ARQ_REGRAS,mode='r') as f:
                regras = f.read()
            regras = json.loads(regras.strip()) 
        regras = regras.get('regras',[])
        return RegrasPesquisaBR.ordenar_regras(regras)

    # converte os dados retornados pelo controller 
    def conversao_retorno(self, retorno: dict):
        # para manter a compatibilidade com os testes básicos do serviço
        # os testes preenchem uma chave 'rodando-testes' para 
        # que essa informação possa ser usada no controle de conversão
        if 'rodando-testes' in retorno:
            print('Em teste - ignorando conversão de retorno')
            return
        # exemplo para injetar uma chave com os grupos retornados
        # poderia buscar outros dados no BD, em outro serviço, transformar dados, etc
        if 'regras' in retorno:
           regras = retorno['regras']
           grupos = []
           for r in regras:
               if r.get('grupo') and r.get('grupo') not in grupos:
                  grupos.append(r.get('grupo'))
           retorno['grupos'] = grupos

    # converte os dados de entrada para o serviço
    def conversao_entrada(self, dados: dict, front_end = False, analisar_regras=False, analisar_criterios=False):
        # para manter a compatibilidade com os testes básicos do serviço
        if 'rodando-testes' in dados:
            print('Em teste - ignorando conversão de entrada')
            return
        # front-end de exemplo do serviço, para não ser impactado pela conversão de saída
        if front_end or analisar_criterios:
            return
        # exemplo para converter uma chave em tags de pesquisa
        # ou retirar dados desnecessários dos resultados
        # ou tratar dados antes de devolver 
        # poderia buscar outros dados no BD, em outro serviço, transformar dados, etc
        if analisar_regras:
            if 'valor_filtro' in dados:
               dados['tags'] = f"{dados.get('tags','')} {dados['valor_filtro']}".strip()
               dados.pop('valor_filtro')
            # remove as chaves sem valor de filtro - chaves enviadas como null por exemplo
            # para não serem filtradas nas chaves das regras
            for chave in list(dados.keys()):
                if dados[chave] is None:
                   dados.pop(chave)
            # força o resultado sempre detalhado
            dados['detalhar'] = 1