#!/usr/bin/env python3 import json import os import traceback import sys # Importamos a nossa trava de segurança do arquivo vizinho from duckdb_engine import get_connection def application(environ, start_response): # --- ROTA DE INTERFACE (VISUALIZAÇÃO NO CHROME) --- if environ.get('REQUEST_METHOD') == 'GET': start_response('200 OK', [('Content-Type', 'text/html; charset=utf-8')]) return [f""" LinkTudo - DuckDB Protected Engine

⚡ DUCKDB ENGINE PROTEGIDO

Limites ativos: 1 Core / 128MB RAM (via duckdb_engine.py)

🛡️ HARDWARE LIMITER STATUS: ATIVO
SCANNER GERAL DE ARQUIVO PARQUET (3.6GB)

A execução será feita em thread única para não derrubar o Passenger.


        


        
        
    


    


""".encode('utf-8')]

    # --- ROTA DE PROCESSAMENTO (POST) ---
    if environ.get('REQUEST_METHOD') == 'POST':
        start_response('200 OK', [('Content-Type', 'application/json; charset=utf-8')])
        try:
            path = "/home/link6350/arquivos_restritos/banco_de_dados/parquet/cnpj/tabelas_auxiliares_pesadas/cnpj___tb_aux___estabelecimentos___por_cnae_por_logradouro___mais_agregacoes___para_filtro_personalizado___cliente_final.parquet"
            
            # CHAVE DE SEGURANÇA: Obtém a conexão já limitada
            con = get_connection()

            # Processamento otimizado
            count_total = con.execute(f"SELECT count(*) FROM read_parquet('{path}')").fetchone()[0]
            count_4733 = con.execute(f"SELECT count(*) FROM read_parquet('{path}') WHERE municipio_codigo = '4733'").fetchone()[0]
            cols_info = con.execute(f"DESCRIBE SELECT * FROM read_parquet('{path}') LIMIT 0").fetchall()
            colunas = [row[0] for row in cols_info]

            return [json.dumps({
                "status": "sucesso_blindado",
                "configuracao": "1_core_128mb_limit",
                "total_linhas": count_total,
                "municipio_4733": count_4733,
                "colunas": colunas,
                "engine_file": "duckdb_engine.py"
            }).encode('utf-8')]

        except Exception as e:
            return [json.dumps({
                "status": "error", 
                "msg": str(e), 
                "trace": traceback.format_exc()
            }).encode('utf-8')]

    return [b"Metodo nao permitido"]