Update app.py

app.py

@@ -91,36 +91,44 @@ class PostgresManager:
 
     def _rewrite_line_for_schema(self, line: str, schema_name: str) -> str:
         """
-        Reescribe referencias a public./pagila. para que apunten al schema de sesión.
-        También ajusta literales 'public.xxx'::regclass.
-        NO toca comentarios (las líneas que lleguen aquí ya no empiezan con '--').
+        Reescribe TODAS las referencias a public.XXX o pagila.XXX:
+        - COPY public.table
+        - ALTER TABLE ONLY public.table
+        - REFERENCES public.table
+        - SET search_path
+        - Literales 'public.table'::regclass
         """
-
-        # 1) Reemplazar public.objeto y pagila.objeto por schema.objeto
-        pattern_schema = re.compile(
+    
+        original = line
+    
+        # 0) Borrar cualquier SET search_path
+        if "search_path" in line.lower():
+            return ""
+    
+        # 1) Reemplazar public.xxx y pagila.xxx en cualquier contexto
+        line = re.sub(
             r'\b(public|pagila)\.("?[A-Za-z_][A-Za-z0-9_]*"?)',
-            flags=re.IGNORECASE,
-        )
-        line = pattern_schema.sub(
             lambda m: f'{schema_name}.{m.group(2)}',
             line,
+            flags=re.IGNORECASE
         )
-
-        # 2) Reescribir literales 'public.objeto'::regclass
-        pattern_regclass = re.compile(
+    
+        # 2) Reemplazar literales 'public.xxx'::regclass
+        line = re.sub(
             r"'public\.([A-Za-z_][A-Za-z0-9_]*)'::regclass",
-            flags=re.IGNORECASE,
-        )
-        line = pattern_regclass.sub(
             lambda m: f"'{schema_name}.{m.group(1)}'::regclass",
             line,
+            flags=re.IGNORECASE
         )
-
-        # 3) Borrar manipulación explícita del search_path en el dump
-        #    (nosotros ya lo fijamos al schema de sesión)
-        if "pg_catalog.set_config('search_path'" in line:
-            return ""
-
+    
+        # 3) Reemplazar SOLO si COPY está antes de reescribir
+        if line.strip().upper().startswith("COPY "):
+            line = line.replace("public.", f"{schema_name}.")
+    
+        # 4) Igual para ALTER TABLE ONLY
+        if "ALTER TABLE ONLY" in line.upper():
+            line = line.replace("public.", f"{schema_name}.")
+    
         return line
 
     def _should_skip_statement(self, stmt: str) -> bool:
@@ -273,8 +281,17 @@ class PostgresManager:
 
             # Detectar inicio de COPY
             if (
-                stripped.upper().startswith("COPY ")
-                and "FROM stdin" in stripped.upper()
+                # Primero reescribimos la línea según el schema
+                line = self._rewrite_line_for_schema(line, schema_name)
+                stripped = line.strip()
+                
+                # SOLO después detectamos COPY
+                if stripped.upper().startswith("COPY ") and "FROM stdin" in stripped.upper():
+                    flush_statement()
+                    in_copy = True
+                    copy_sql = stripped
+                    copy_lines = []
+                    continue
             ):
                 # Ejecutar lo que haya pendiente antes del COPY
                 flush_statement()