Update app.py

app.py

@@ -161,6 +161,7 @@ class PostgresManager:
           para hacer SELECT sobre las tablas).
         - Ignoramos errores tipo “already exists” para que no reviente si
           el script crea dos veces la misma tabla/índice.
+        - Ignoramos sentencias que cambian OWNER a 'postgres'.
         """
         connection_id = self._new_connection_id()
         schema_name = f"sess_{uuid.uuid4().hex[:8]}"
@@ -184,13 +185,16 @@ class PostgresManager:
                 copy_sql = ""
                 copy_rows: list[str] = []
 
-                in_function = False  # ← estamos dentro de un CREATE FUNCTION ... $$ ... $$
-                function_end_delim: str | None = None
+                # NUEVO: estado para funciones
+                in_function = False
+                function_delim: str | None = None
+                function_delim_count: int = 0
 
                 stmt_lines: list[str] = []
 
                 def flush_statement():
                     """Ejecuta el statement acumulado si es útil."""
+                    nonlocal stmt_lines
                     stmt = "\n".join(stmt_lines).strip()
                     stmt_lines.clear()
                     if not stmt or stmt == ";":
@@ -198,7 +202,7 @@ class PostgresManager:
 
                     upper = stmt.upper()
 
-                    # 1) Saltar cosas peligrosas o globales
+                    # Saltar cosas globales/peligrosas
                     skip_prefixes = (
                         "SET ",
                         "SELECT PG_CATALOG.SET_CONFIG",
@@ -215,19 +219,11 @@ class PostgresManager:
                     if upper.startswith(skip_prefixes):
                         return
 
-                    # 2) Saltar cualquier sentencia que intente cambiar OWNER o ROLES
-                    #    Ejemplos:
-                    #      ALTER TABLE ... OWNER TO postgres;
-                    #      ALTER FUNCTION ... OWNER TO postgres;
-                    #      CREATE ROLE postgres;
-                    #      ALTER ROLE postgres ...;
-                    #      SET ROLE postgres;
+                    # Saltar ALTER ... OWNER TO postgres (u otros OWNER)
                     if " OWNER TO " in upper:
                         return
-                    if upper.startswith("CREATE ROLE") or upper.startswith("ALTER ROLE") or upper.startswith("SET ROLE"):
-                        return
 
-                    # 3) Quitar ';' final (psycopg2 no la necesita)
+                    # Quitar ';' final (psycopg2 no la necesita)
                     if stmt.endswith(";"):
                         stmt = stmt[:-1]
 
@@ -235,7 +231,7 @@ class PostgresManager:
                         cur.execute(stmt)
                     except Exception as e:
                         msg = str(e).lower()
-                        # Ignorar errores típicos de dumps (idempotencia)
+                        # Ignorar cosas no fatales típicas de dumps
                         if "already exists" in msg or "duplicate key value" in msg:
                             print("[WARN] Ignorando error no crítico:", e)
                             return
@@ -245,7 +241,7 @@ class PostgresManager:
                     line = raw_line.rstrip("\n")
                     stripped = line.strip()
 
-                    # Comentarios o líneas vacías
+                    # Comentarios o líneas vacías (si no estamos en COPY/función)
                     if not in_copy and not in_function:
                         if not stripped or stripped.startswith("--"):
                             continue
@@ -261,9 +257,23 @@ class PostgresManager:
                             copy_rows.append(line)
                         continue
 
+                    # ====== BLOQUE CREATE FUNCTION ... $$ ... $$ (IGNORAR) ======
+                    if in_function:
+                        # Contamos apariciones del delimitador ($$ o $body$)
+                        if function_delim:
+                            function_delim_count += line.count(function_delim)
+                            # Cuando lo hemos visto al menos 2 veces
+                            # (apertura y cierre) salimos del bloque función
+                            if function_delim_count >= 2:
+                                in_function = False
+                                function_delim = None
+                                function_delim_count = 0
+                        continue
+
+                    # Detectar inicio de COPY
                     if (
                         stripped.upper().startswith("COPY ")
-                        and "FROM stdin" in stripped
+                        and "FROM stdin" in stripped.upper()
                     ):
                         flush_statement()
                         in_copy = True
@@ -271,25 +281,30 @@ class PostgresManager:
                         copy_rows = []
                         continue
 
-                    # ====== BLOQUE CREATE FUNCTION ... $$ ... $$ ======
-                    if in_function:
-                        # seguimos saltando hasta encontrar el delimitador de cierre
-                        if function_end_delim and function_end_delim in stripped:
-                            in_function = False
-                            function_end_delim = None
-                        continue
-
+                    # Detectar inicio de función: CREATE [OR REPLACE] FUNCTION ...
                     if stripped.upper().startswith("CREATE FUNCTION") or stripped.upper().startswith(
                         "CREATE OR REPLACE FUNCTION"
                     ):
                         # Entramos en bloque función → lo ignoramos entero
                         in_function = True
-                        # detectar el delimitador $$...$$ por si viene con nombre ($body$ etc)
+
+                        # Detectar delimitador tipo $$ o $algo$
                         m = re.search(r"\$(\w*)\$", stripped)
                         if m:
-                            function_end_delim = f"${m.group(1)}$"
+                            function_delim = f"${m.group(1)}$"
                         else:
-                            function_end_delim = "$$"
+                            function_delim = "$$"
+
+                        # Contamos cuántas veces aparece ya en esta línea
+                        function_delim_count = stripped.count(function_delim)
+
+                        # Caso raro: función toda en una sola línea (apertura+cierre)
+                        if function_delim_count >= 2:
+                            in_function = False
+                            function_delim = None
+                            function_delim_count = 0
+
+                        # No añadimos la línea al buffer, se ignora
                         continue
 
                     # ====== STATEMENTS NORMALES ======