Update app.py

app.py

@@ -212,18 +212,15 @@ class PostgresManager:
         - Reescribe public./pagila. -> schema_name.
         - Respeta funciones con $$...$$ (no corta por ';' internos).
         - Ignora statements peligrosos via _should_skip_statement().
-        - Ignora bloques CREATE DOMAIN ... multi-línea.
         """
 
         in_copy = False
         copy_sql = ""
         copy_lines: list[str] = []
 
-        buffer = ""          # statement acumulado
-        in_dollar = False    # estamos dentro de $$...$$ ?
-        dollar_tag = ""      # por ej. "$func$"
-
-        in_domain_block = False  # estamos dentro de un bloque CREATE DOMAIN ?
+        buffer = ""       # statement acumulado
+        in_dollar = False # estamos dentro de $$...$$ ?
+        dollar_tag = ""   # por ej. "$func$"
 
         def flush_statement():
             nonlocal buffer
@@ -248,14 +245,15 @@ class PostgresManager:
             line = raw_line.rstrip("\n")
             stripped = line.strip()
 
-            # Comentarios y líneas vacías (fuera de COPY / DOMAIN)
-            if not in_copy and not in_domain_block:
+            # Comentarios y líneas vacías (fuera de COPY)
+            if not in_copy:
                 if not stripped or stripped.startswith("--"):
                     continue
 
             # ====== BLOQUE COPY ... FROM stdin ======
             if in_copy:
                 if stripped == r"\.":
+                    # fin de COPY
                     data = "\n".join(copy_lines) + "\n"
                     cur.copy_expert(copy_sql, io.StringIO(data))
                     in_copy = False
@@ -265,30 +263,18 @@ class PostgresManager:
                     copy_lines.append(line)
                 continue
 
-            # ====== BLOQUE CREATE DOMAIN (lo ignoramos entero hasta ';') ======
-            if in_domain_block:
-                # Terminamos el bloque cuando encontramos una línea con solo ';'
-                if stripped == ";":
-                    in_domain_block = False
-                # No añadimos nada al buffer
-                continue
-
-            # Detectar inicio de bloque DOMAIN
-            if stripped.upper().startswith("CREATE DOMAIN"):
-                in_domain_block = True
-                continue
-
-            # 👉 Primero reescribimos la línea según el schema de sesión
+            # Reescribimos la línea según el schema de sesión
             line = self._rewrite_line_for_schema(line, schema_name)
-            if not line.strip():
-                continue
             stripped = line.strip()
+            if not stripped:
+                continue
 
-            # Detectar inicio de COPY (YA con la línea reescrita)
+            # Detectar inicio de COPY ahora que la línea ya está reescrita
             if stripped.upper().startswith("COPY ") and "FROM stdin" in stripped.upper():
+                # Ejecutar lo que haya pendiente antes del COPY
                 flush_statement()
                 in_copy = True
-                copy_sql = stripped
+                copy_sql = stripped  # ya reescrita
                 copy_lines = []
                 continue
 
@@ -309,11 +295,9 @@ class PostgresManager:
                     if j < length and line[j] == "$":
                         tag = line[i : j + 1]  # incluye ambos '$'
                         if not in_dollar:
-                            # comenzamos bloque
                             in_dollar = True
                             dollar_tag = tag
                         else:
-                            # posible fin de bloque
                             if tag == dollar_tag:
                                 in_dollar = False
                                 dollar_tag = ""
@@ -322,7 +306,6 @@ class PostgresManager:
 
                 # Fin de statement: ';' fuera de bloque dollar-quoted
                 if ch == ";" and not in_dollar:
-                    # segmento hasta aquí (incluyendo ';')
                     segment = line[start_seg : i + 1]
                     buffer += segment + "\n"
                     flush_statement()