Update app.py

app.py

@@ -257,17 +257,16 @@ def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
         zi.flag_bits = 0
         out_zip.writestr(zi, base_zip.read("mimetype"))
     
-    # 베이스 HWPX에서 섹션 파일들 읽기
-    base_sections = {}
-    add_sections = {}
+    # 모든 파일을 처리 - 섹션 파일들은 별도 병합
+    section_files = {}
     
+    # 베이스에서 모든 파일 수집
     for filename in base_zip.namelist():
         if filename == "mimetype":
             continue
         if filename.startswith("Contents/section") and filename.endswith(".xml"):
-            base_sections[filename] = base_zip.read(filename).decode("utf-8", errors="ignore")
+            section_files[filename] = [base_zip.read(filename).decode("utf-8", errors="ignore")]
         else:
-            # 다른 파일들은 그대로 복사
             zi = zipfile.ZipInfo(filename)
             zi.compress_type = zipfile.ZIP_DEFLATED
             zi.external_attr = 0o100666 << 16
@@ -276,23 +275,29 @@ def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
             zi.flag_bits = 0
             out_zip.writestr(zi, base_zip.read(filename))
     
-    # 추가할 HWPX에서 섹션 파일들 읽기
+    # 추가 파일에서 섹션만 수집
     for filename in add_zip.namelist():
         if filename.startswith("Contents/section") and filename.endswith(".xml"):
-            add_sections[filename] = add_zip.read(filename).decode("utf-8", errors="ignore")
-    
-    # 섹션 병합 처리
-    merged_sections = merge_sections(base_sections, add_sections)
+            content = add_zip.read(filename).decode("utf-8", errors="ignore")
+            if filename in section_files:
+                section_files[filename].append(content)
+            else:
+                section_files[filename] = [content]
     
-    # 병합된 섹션들을 ZIP에 쓰기
-    for filename, content in merged_sections.items():
+    # 섹션 파일들 병합해서 쓰기
+    for filename, contents in section_files.items():
+        if len(contents) == 1:
+            merged_content = contents[0]
+        else:
+            merged_content = merge_section_xml_list(contents)
+        
         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"))
+        out_zip.writestr(zi, merged_content.encode("utf-8"))
     
     base_zip.close()
     add_zip.close()
@@ -301,6 +306,36 @@ def merge_hwpx_pages(base_hwpx: bytes, additional_hwpx: bytes) -> bytes:
     
     return out_buf.getvalue()
 
+def merge_section_xml_list(xml_list: list) -> str:
+    """여러 섹션 XML을 하나로 병합"""
+    if len(xml_list) <= 1:
+        return xml_list[0] if xml_list else ""
+    
+    base_xml = xml_list[0]
+    
+    for additional_xml in xml_list[1:]:
+        # 각 XML에서 페이지 추출
+        page_pattern = re.compile(
+            r'<(?P<prefix>[a-zA-Z0-9_]+):page\b[^>]*>.*?</(?P=prefix):page>', 
+            re.DOTALL
+        )
+        
+        add_pages = page_pattern.findall(additional_xml)
+        if not add_pages:
+            continue
+        
+        # 페이지 전체 매치
+        page_matches = page_pattern.finditer(additional_xml)
+        pages_to_add = [match.group(0) for match in page_matches]
+        
+        if pages_to_add:
+            # 베이스 XML의 </prefix:pages> 태그 찾아서 그 앞에 삽입
+            pages_close_pattern = re.compile(r'(</[a-zA-Z0-9_]+:pages>)')
+            pages_str = ''.join(pages_to_add)
+            base_xml = pages_close_pattern.sub(pages_str + r'\1', base_xml)
+    
+    return base_xml
+
 def merge_sections(base_sections: dict, add_sections: dict) -> dict:
     """섹션 XML들을 병합"""
     merged = base_sections.copy()
@@ -317,22 +352,25 @@ def merge_sections(base_sections: dict, add_sections: dict) -> dict:
 
 def merge_section_content(base_xml: str, add_xml: str) -> str:
     """단일 섹션 XML 내용을 병합"""
-    # 추가할 XML에서 페이지들 추출
-    page_pattern = re.compile(r'(<(?P<prefix>[a-zA-Z0-9_]+):page[^>]*>.*?</(?P=prefix):page>)', re.DOTALL)
-    add_pages = page_pattern.findall(add_xml)
+    # 추가할 XML에서 페이지들 추출 - 더 정확한 패턴
+    page_pattern = re.compile(
+        r'<(?P<prefix>[a-zA-Z0-9_]+):page\b[^>]*>.*?</(?P=prefix):page>', 
+        re.DOTALL
+    )
     
-    if not add_pages:
+    # 페이지 매칭
+    page_matches = list(page_pattern.finditer(add_xml))
+    if not page_matches:
         return base_xml
     
-    # 베이스 XML의 마지막 페이지 뒤에 새 페이지들 삽입
-    # </hp:pages> 태그 앞에 삽입
-    pages_end_pattern = re.compile(r'(</[a-zA-Z0-9_]+:pages>)')
+    # 추가할 페이지들
+    pages_to_add = [match.group(0) for match in page_matches]
     
-    # 추가할 페이지들을 문자열로 변환
-    pages_to_add = ''.join([page[0] for page in add_pages])
+    # 베이스 XML의 </hp:pages> 또는 </hml:pages> 태그 앞에 삽입
+    pages_end_pattern = re.compile(r'(</[a-zA-Z0-9_]+:pages>)')
+    pages_str = ''.join(pages_to_add)
     
-    # 삽입
-    merged_xml = pages_end_pattern.sub(pages_to_add + r'\1', base_xml)
+    merged_xml = pages_end_pattern.sub(pages_str + r'\1', base_xml)
     
     return merged_xml