Update app.py

app.py

@@ -236,7 +236,7 @@ def replace_in_hwpx(hwpx_bytes: bytes, mapping: Dict[str,str]) -> Tuple[bytes, d
     return out_buf.getvalue(), dbg
 
 def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
-    """HWPX에 새 페이지 추가 (템플릿 복제 방식)"""
+    """HWPX 파일들을 섹션 단위로 병합 (COM InsertFile과 유사한 방식)"""
     import time
     
     base_zip = zipfile.ZipFile(io.BytesIO(base_hwpx), "r")
@@ -257,33 +257,43 @@ def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
         zi.flag_bits = 0
         out_zip.writestr(zi, base_zip.read("mimetype"))
     
-    # 섹션 파일들을 따로 처리
-    section_files = {}
+    # 베이스 파일들 처리
+    base_sections = {}
+    base_files = {}
     
-    # 베이스에서 모든 파일 처리
     for filename in base_zip.namelist():
         if filename == "mimetype":
             continue
+        data = base_zip.read(filename)
+        
         if filename.startswith("Contents/section") and filename.endswith(".xml"):
-            section_files[filename] = base_zip.read(filename).decode("utf-8", errors="ignore")
+            base_sections[filename] = data.decode("utf-8", errors="ignore")
         else:
-            zi = zipfile.ZipInfo(filename)
-            zi.compress_type = zipfile.ZIP_DEFLATED
-            zi.external_attr = 0o100666 << 16
-            zi.create_system = 0
-            zi.date_time = now[:6]
-            zi.flag_bits = 0
-            out_zip.writestr(zi, base_zip.read(filename))
+            base_files[filename] = data
+    
+    # 추가 파일의 섹션들 수집
+    add_sections = {}
+    next_section_num = len(base_sections) + 1
     
-    # 추가 파일에서 페이지 추출해서 기존 섹션에 추가
     for filename in add_zip.namelist():
         if filename.startswith("Contents/section") and filename.endswith(".xml"):
-            add_content = add_zip.read(filename).decode("utf-8", errors="ignore")
-            if filename in section_files:
-                section_files[filename] = add_page_to_section(section_files[filename], add_content)
+            # 새로운 섹션 번호로 변경
+            new_filename = f"Contents/section{next_section_num}.xml"
+            add_sections[new_filename] = add_zip.read(filename).decode("utf-8", errors="ignore")
+            next_section_num += 1
     
-    # 수정된 섹션 파일들 저장
-    for filename, content in section_files.items():
+    # 모든 베이스 파일들 복사
+    for filename, data in base_files.items():
+        zi = zipfile.ZipInfo(filename)
+        zi.compress_type = zipfile.ZIP_DEFLATED
+        zi.external_attr = 0o100666 << 16
+        zi.create_system = 0
+        zi.date_time = now[:6]
+        zi.flag_bits = 0
+        out_zip.writestr(zi, data)
+    
+    # 베이스 섹션들 복사
+    for filename, content in base_sections.items():
         zi = zipfile.ZipInfo(filename)
         zi.compress_type = zipfile.ZIP_DEFLATED
         zi.external_attr = 0o100666 << 16
@@ -292,6 +302,29 @@ def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
         zi.flag_bits = 0
         out_zip.writestr(zi, content.encode("utf-8"))
     
+    # 새로운 섹션들 추가
+    for filename, content in add_sections.items():
+        zi = zipfile.ZipInfo(filename)
+        zi.compress_type = zipfile.ZIP_DEFLATED
+        zi.external_attr = 0o100666 << 16
+        zi.create_system = 0
+        zi.date_time = now[:6]
+        zi.flag_bits = 0
+        out_zip.writestr(zi, content.encode("utf-8"))
+    
+    # BodyText 업데이트 (새 섹션 참조 추가)
+    if "Contents/bodytext.xml" in base_files:
+        bodytext = base_files["Contents/bodytext.xml"].decode("utf-8", errors="ignore")
+        updated_bodytext = add_sections_to_bodytext(bodytext, list(add_sections.keys()))
+        
+        zi = zipfile.ZipInfo("Contents/bodytext.xml")
+        zi.compress_type = zipfile.ZIP_DEFLATED
+        zi.external_attr = 0o100666 << 16
+        zi.create_system = 0
+        zi.date_time = now[:6]
+        zi.flag_bits = 0
+        out_zip.writestr(zi, updated_bodytext.encode("utf-8"))
+    
     base_zip.close()
     add_zip.close()
     out_zip.close()
@@ -299,58 +332,33 @@ def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
     
     return out_buf.getvalue()
 
-def add_page_to_section(base_xml: str, add_xml: str) -> str:
-    """섹션에 새 페이지 추가"""
-    # 추가할 XML에서 첫 번째 페이지 추출
-    page_pattern = re.compile(
-        r'<(?P<prefix>[a-zA-Z0-9_]+):page\b[^>]*>.*?</(?P=prefix):page>', 
-        re.DOTALL
-    )
-    
-    page_match = page_pattern.search(add_xml)
-    if not page_match:
-        return base_xml
-    
-    new_page = page_match.group(0)
+def add_sections_to_bodytext(bodytext: str, new_section_files: list) -> str:
+    """BodyText에 새 섹션 참조 추가"""
+    # 마지막 섹션 뒤에 새 섹션들 추가
+    # </hml:body> 태그 앞에 새 섹션 참조 삽입
     
-    # 베이스 XML에서 </prefix:pages> 태그 찾아서 그 앞에 새 페이지 삽입
-    pages_close_pattern = re.compile(r'(</[a-zA-Z0-9_]+:pages>)')
+    section_refs = []
+    for section_file in new_section_files:
+        # section1.xml -> 1 추출
+        section_num = section_file.split("section")[1].split(".xml")[0]
+        section_ref = f'<hml:secDef><hml:secPtr hml:hRef="../Contents/section{section_num}.xml#0"/></hml:secDef>'
+        section_refs.append(section_ref)
     
-    # 페이지 ID 업데이트 (중복 방지)
-    new_page = update_page_id(base_xml, new_page)
+    if section_refs:
+        # </hml:body> 앞에 삽입
+        body_close_pattern = re.compile(r'(</hml:body>)')
+        new_sections_xml = ''.join(section_refs)
+        bodytext = body_close_pattern.sub(new_sections_xml + r'\1', bodytext)
     
-    # 페이지 삽입
-    result = pages_close_pattern.sub(new_page + r'\1', base_xml)
-    
-    return result
+    return bodytext
 
 def update_page_id(base_xml: str, new_page: str) -> str:
-    """페이지 ID를 중복되지 않게 업데이트"""
-    # 기존 페이지들의 ID 추출
-    existing_ids = set()
-    id_pattern = re.compile(r'<[a-zA-Z0-9_]+:page\b[^>]*\bid="([^"]*)"')
-    
-    for match in id_pattern.finditer(base_xml):
-        existing_ids.add(match.group(1))
-    
-    # 새 페이지의 ID 찾기
-    new_page_id_match = id_pattern.search(new_page)
-    if not new_page_id_match:
-        return new_page
-    
-    original_id = new_page_id_match.group(1)
-    
-    # 중복되지 않는 새 ID 생성
-    counter = 2
-    new_id = f"{original_id}_{counter}"
-    while new_id in existing_ids:
-        counter += 1
-        new_id = f"{original_id}_{counter}"
-    
-    # ID 교체
-    updated_page = new_page.replace(f'id="{original_id}"', f'id="{new_id}"')
-    
-    return updated_page
+    """페이지 ID를 중복되지 않게 업데이트 (더 이상 사용하지 않음)"""
+    return new_page
+
+def add_page_to_section(base_xml: str, add_xml: str) -> str:
+    """섹션에 새 페이지 추가 (더 이상 사용하지 않음)"""
+    return base_xml
 
 def merge_section_xml_list(xml_list: list) -> str:
     """여러 섹션 XML을 하나로 병합 (사용하지 않지만 호환성 유지)"""