Rename chunk_base64 as attribute name

greenplum-db · Sep 19, 2023 · 4061bce · 4061bce
1 parent 3d99a52
commit 4061bce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/greenplumpython/experimental/file.py b/greenplumpython/experimental/file.py
@@ -49,22 +49,22 @@ def _from_files(_, files: list[str], parser: NormalFunction, db: gp.Database) ->
             tmp_archive.add(pathlib.Path(file_path))
     with psycopg2.connect(db._conn.dsn, options="-c gp_session_role=utility") as util_conn:
         with util_conn.cursor() as cursor:
-            cursor.execute(f"CREATE TEMP TABLE {tmp_archive_name} (id serial, chunk_base64 text);")
+            cursor.execute(f"CREATE TEMP TABLE {tmp_archive_name} (id serial, text_base64 text);")
             with open(tmp_archive_path, "rb") as tmp_archive:
                 while True:
                     chunk = tmp_archive.read(_CHUNK_SIZE)
                     if len(chunk) == 0:
                         break
                     chunk_base64 = base64.b64encode(chunk)
                     cursor.copy_expert(
-                        f"COPY {tmp_archive_name} (chunk_base64) FROM STDIN",
+                        f"COPY {tmp_archive_name} (text_base64) FROM STDIN",
                         io.BytesIO(chunk_base64),
                     )
             util_conn.commit()
             cursor.execute(_dump_file_chunk._serialize(db))  # type: ignore reportUnknownArgumentType
             cursor.execute(
                 f"""
-                SELECT {_dump_file_chunk._qualified_name_str}('{tmp_archive_name}', chunk_base64)
+                SELECT {_dump_file_chunk._qualified_name_str}('{tmp_archive_name}', text_base64)
                 FROM "{tmp_archive_name}"
                 ORDER BY id;
                 """