182 lines
12 KiB
Python
182 lines
12 KiB
Python
import asyncio
|
||
import zipfile
|
||
from bs4 import BeautifulSoup
|
||
from ebooklib import epub
|
||
from lxml import etree
|
||
import pytest
|
||
from src.common.data_model import BookStructure
|
||
from src.preprocessing.epub_cleaner import EpubCleaner
|
||
from src.preprocessing.text_extractor import FineGrainedExtractor
|
||
from src.preprocessing.navigation import prepare_navigation, destination, chinese_number
|
||
from src.assembly.restoration_engine import RestorationEngine
|
||
from src.assembly.backfiller import BackfillEngine
|
||
from src.assembly.builder import BilingualBuilder
|
||
from src.qa.epub_audit import audit_epub, compare_audits
|
||
|
||
|
||
def fixture(tmp_path):
|
||
book = epub.EpubBook(); book.set_identifier('titles'); book.set_title('Title QA'); book.set_language('en')
|
||
chapter = epub.EpubHtml(uid='ch', file_name='Text/ch.xhtml', title='Roots')
|
||
chapter.content = '''<html><head></head><body><h2 id="ch1"><span class="line">1</span>
|
||
<span class="subtitle">ROOTS</span></h2><p id="opening" class="noindent" style="text-indent:0 !important; color:red">Opening prose.</p>
|
||
<p id="caption" class="caption">A caption.</p><blockquote><p id="quote">Quoted text.</p></blockquote>
|
||
<p id="second">Second paragraph.<a epub:type="noteref" href="notes.xhtml#note">1</a></p></body></html>'''
|
||
part = epub.EpubHtml(uid='part', file_name='Text/part.xhtml', title='Part 1')
|
||
part.content = '<html><body><h1 id="part"><span>PART 1</span><br/><em>BOY GENIUS</em></h1></body></html>'
|
||
contents = epub.EpubHtml(uid='contents', file_name='Text/Contents.xhtml', title='Contents')
|
||
contents.content = '''<html><body><h1 id="contents">Contents</h1><ol><li><a href="part.xhtml#part">Part 1: Boy Genius</a>
|
||
<ol><li><a href="ch.xhtml#ch1"><em>Chapter 1: Roots</em></a></li></ol></li></ol></body></html>'''
|
||
notes = epub.EpubHtml(uid='notes', file_name='Text/notes.xhtml', title='Notes')
|
||
notes.content = '<html><body><h1 id="notes">Notes</h1><aside epub:type="footnote" id="note"><p id="note-text">Note text. <a href="ch.xhtml#second">Back</a></p></aside></body></html>'
|
||
for item in [chapter,part,contents,notes,epub.EpubNav(),epub.EpubNcx()]: book.add_item(item)
|
||
book.toc = [(epub.Section('Part 1: Boy Genius', 'Text/part.xhtml#part'),
|
||
[epub.Link('Text/ch.xhtml#ch1','Chapter 1: Roots','ch1')]),
|
||
epub.Link('Text/notes.xhtml#notes','Notes','notes')]
|
||
book.spine = ['nav',contents,part,chapter,notes]
|
||
source=tmp_path/'source.epub';epub.write_epub(str(source),book)
|
||
work=tmp_path/'source'; structure=BookStructure.load(EpubCleaner(source,work).clean())
|
||
entries=FineGrainedExtractor().extract(structure)
|
||
prepare_navigation(source,structure,entries)
|
||
for e in entries:
|
||
e.translated_text=(e.original_text.replace('ROOTS','根源').replace('PART 1','第一部')
|
||
.replace('BOY GENIUS','天才少年').replace('Contents','目录').replace('Notes','注释')
|
||
.replace('Opening prose.','开篇正文。').replace('Second paragraph.','第二段正文。')
|
||
.replace('A caption.','图注。').replace('Quoted text.','引文。'))
|
||
assert asyncio.run(RestorationEngine().restore_entries(entries))==len(entries)
|
||
return source,work,structure,entries
|
||
|
||
|
||
@pytest.mark.parametrize('mode',['bilingual','target_only'])
|
||
def test_headings_nav_ncx_contents_share_translation_and_keep_links(tmp_path,mode):
|
||
source,work,structure,entries=fixture(tmp_path)
|
||
chapter_titles=[e for e in entries if e.element_id=='ch1']
|
||
assert len(chapter_titles)==1
|
||
assert not any('Chapter 1: Roots' in e.original_text for e in entries)
|
||
asyncio.run(BackfillEngine().backfill(structure,entries,mode))
|
||
# Repeat safely: no duplicate bilingual paragraph or translated TOC label.
|
||
asyncio.run(BackfillEngine().backfill(structure,entries,mode))
|
||
output=tmp_path/(mode+'.epub');BilingualBuilder(work,source).build(structure,output)
|
||
assert not compare_audits(audit_epub(source),audit_epub(output))
|
||
with zipfile.ZipFile(source) as before,zipfile.ZipFile(output) as after:
|
||
prefix=next(n for n in before.namelist() if n.endswith('/nav.xhtml')).removesuffix('nav.xhtml')
|
||
for relative in ['nav.xhtml','Text/Contents.xhtml']:
|
||
name=prefix+relative
|
||
a=BeautifulSoup(before.read(name),'html.parser');b=BeautifulSoup(after.read(name),'html.parser')
|
||
assert [x.get('href') for x in a.find_all('a')]==[x.get('href') for x in b.find_all('a')]
|
||
link=next(x for x in b.find_all('a') if 'ch.xhtml' in x.get('href',''))
|
||
assert '第一章：根源' in link.get_text()
|
||
assert link.get_text().count('根源')==1
|
||
assert link.find_parent('ol').find_parent('li') is not None
|
||
a=etree.fromstring(before.read(prefix+'toc.ncx'));b=etree.fromstring(after.read(prefix+'toc.ncx'))
|
||
assert a.xpath('//*[local-name()="content"]/@src')==b.xpath('//*[local-name()="content"]/@src')
|
||
assert a.xpath('//*[local-name()="navPoint"]/@id')==b.xpath('//*[local-name()="navPoint"]/@id')
|
||
assert '第一章：根源' in ''.join(b.itertext())
|
||
ch=BeautifulSoup(after.read(prefix+'Text/ch.xhtml'),'html.parser')
|
||
title=ch.find('h2',class_='translation')
|
||
assert title.find('span',class_='line').get_text()=='第一章'
|
||
assert title.find('span',class_='subtitle').get_text()=='根源'
|
||
assert title['data-translation-role']=='heading'
|
||
for key in ['opening','second']:
|
||
p=ch.find(attrs={'data-translation-for':key}) if mode=='bilingual' else ch.find(id=key)
|
||
assert p['data-translation-role']=='body'
|
||
assert 'text-indent' not in p.get('style','')
|
||
for key in ['caption','quote']:
|
||
p=ch.find(attrs={'data-translation-for':key}) if mode=='bilingual' else ch.find(id=key)
|
||
assert p['data-translation-role']=='other'
|
||
notes=BeautifulSoup(after.read(prefix+'Text/notes.xhtml'),'html.parser')
|
||
assert notes.find(id='note') and notes.find('a',href='ch.xhtml#second')
|
||
|
||
|
||
def test_uncertain_destination_translates_label_without_overwriting_prose(tmp_path):
|
||
source,work,structure,entries=fixture(tmp_path)
|
||
# A link to prose must not be treated as the chapter heading.
|
||
nav=next(r for r in structure.resources.values() if r.href=='nav.xhtml')
|
||
nav.content=nav.content.replace('Text/ch.xhtml#ch1','Text/ch.xhtml#opening')
|
||
entries=FineGrainedExtractor().extract(structure)
|
||
prepare_navigation(source,structure,entries)
|
||
record=next(r for r in structure.navigation_titles if r['target'].endswith('#opening'))
|
||
assert record['heading_id'] is None
|
||
assert record['binding_note']=='no matching heading'
|
||
entry=next(e for e in entries if e.entry_id==record['entry_id'])
|
||
assert entry.file_path=='__navigation__'
|
||
assert 'Opening prose.' in next(r for r in structure.resources.values() if r.href=='Text/ch.xhtml').content
|
||
|
||
|
||
def test_destination_preserves_canonical_fragment_and_ignores_external():
|
||
assert destination('Nav/toc.xhtml','../Text/ch%20one.xhtml#part%202')=='Text/ch one.xhtml#part 2'
|
||
assert destination('Text/ch.xhtml','#x')=='Text/ch.xhtml#x'
|
||
assert destination('toc.xhtml','https://example.com/') is None
|
||
assert [chinese_number(x) for x in [1,10,11,20,101,1001]]==['一','十','十一','二十','一百零一','一千零一']
|
||
|
||
|
||
@pytest.mark.parametrize('markup',[
|
||
'<h2 id="number">1</h2><h2 id="title">The Bookseller</h2><h2 id="subtitle">The Ren Family: 1937–1968</h2>',
|
||
'<p id="number" class="calibre1">1</p><p id="title" class="calibre2"><em>The Bookseller</em></p><p id="subtitle" class="calibre3">The Ren Family: 1937–1968</p>',
|
||
'<div id="number"><b>CHAPTER 1</b></div><div id="title"><b>The Bookseller</b></div><div id="subtitle">The Ren Family: 1937–1968</div>',
|
||
])
|
||
def test_separate_number_title_subtitle_are_shared_without_merging_dom(tmp_path,markup):
|
||
from src.preprocessing.navigation import navigation_labels
|
||
source,work,structure,entries=fixture(tmp_path)
|
||
chapter=next(r for r in structure.resources.values() if r.href=='Text/ch.xhtml')
|
||
chapter.content='<html><body>'+markup+'<p id="prose">A long unrelated opening paragraph.</p></body></html>'
|
||
nav=next(r for r in structure.resources.values() if r.href=='nav.xhtml')
|
||
nav.content=nav.content.replace('Text/ch.xhtml#ch1','Text/ch.xhtml').replace('Chapter 1: Roots','Chapter 1: The Bookseller | The Ren Family: 1937–1968')
|
||
entries=FineGrainedExtractor().extract(structure);prepare_navigation(source,structure,entries)
|
||
record=next(r for r in structure.navigation_titles if r['target']=='Text/ch.xhtml')
|
||
assert len(record['entry_ids'])==3
|
||
for e in entries:
|
||
if e.element_id in {'number','title','subtitle'}:
|
||
e.translated_text=e.original_text.replace('CHAPTER 1','第1章：').replace('The Bookseller','书商').replace('The Ren Family: 1937–1968','任家：1937—1968')
|
||
asyncio.run(RestorationEngine().restore_entries(entries))
|
||
labels=navigation_labels(structure,entries)
|
||
assert labels['Text/ch.xhtml']=='第一章：书商 任家：1937—1968'
|
||
asyncio.run(BackfillEngine().backfill(structure,entries,'target_only',allow_partial=True))
|
||
ch=BeautifulSoup(chapter.content,'html.parser')
|
||
assert all(ch.find(id=id) for id in ['number','title','subtitle','prose'])
|
||
assert ch.find(id='prose').get_text()=='A long unrelated opening paragraph.'
|
||
|
||
|
||
def test_heading_number_with_spaces_and_colon_normalizes():
|
||
from src.preprocessing.navigation import normalize_heading_html,number_info
|
||
assert normalize_heading_html({'role':'chapter','number':1},'<span>第 1 章：</span><em>根源</em>')=='<span>第一章</span><em>根源</em>'
|
||
assert number_info('Chapter Eleven: Rebels')==('chapter',11)
|
||
assert number_info('Part Two: Coffee')==('part',2)
|
||
|
||
|
||
def test_split_styled_digit_and_separate_separator():
|
||
from src.preprocessing.navigation import normalize_heading_html
|
||
assert normalize_heading_html({'role':'part','number':1},'第<span>1</span>部：<em>天才少年</em>')=='第<span>一</span>部<em>天才少年</em>'
|
||
assert normalize_heading_html({'role':'chapter','number':1},'<span>第1章</span>：<em>根源</em>')=='<span>第一章</span><em>根源</em>'
|
||
|
||
|
||
def test_numbered_titles_starting_with_two_or_i_do_not_lose_title_words():
|
||
from src.preprocessing.navigation import title_identity
|
||
assert title_identity('Chapter 22: Two Indians Are Better than One')==title_identity('22 TWO INDIANS ARE BETTER THAN ONE')
|
||
assert title_identity('Chapter 26: I Hope You Go Bust')==title_identity('26 I HOPE YOU GO BUST')
|
||
|
||
|
||
def test_manifest_reconciliation_updates_title_context(tmp_path):
|
||
from src.common.data_model import ManifestEntry
|
||
from src.translation.manifest_manager import ManifestManager
|
||
manager=ManifestManager(tmp_path/'manifest.json')
|
||
old=ManifestEntry(entry_id='x#h',file_path='x',element_id='h',original_text='Roots',context='section',translated_text='根源')
|
||
manager.add_entries([old])
|
||
new=old.model_copy(update={'context':'navigation-heading: Chapter 1: Roots','translated_text':None})
|
||
manager.add_entries([new])
|
||
assert manager.entries[0].context==new.context
|
||
assert manager.entries[0].translated_text=='根源'
|
||
|
||
|
||
def test_ncx_declared_utf16_is_parsed_without_utf8_assumptions(tmp_path):
|
||
source,work,structure,entries=fixture(tmp_path)
|
||
with zipfile.ZipFile(source) as z:
|
||
items=[(i,z.read(i.filename)) for i in z.infolist()]
|
||
converted=tmp_path/'utf16.epub'
|
||
with zipfile.ZipFile(converted,'w') as z:
|
||
for info,data in items:
|
||
if info.filename.endswith('.ncx'):
|
||
data=etree.tostring(etree.fromstring(data),encoding='utf-16',xml_declaration=True)
|
||
z.writestr(info,data)
|
||
prepare_navigation(converted,structure,entries)
|
||
assert any(r['source_label']=='Chapter 1: Roots' for r in structure.navigation_titles)
|