703 lines
174 KiB
HTML
703 lines
174 KiB
HTML
<!DOCTYPE html>
|
|
<html lang="en">
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
|
|
<title>Coverage for pyWebLayout/io/readers/epub_reader.py: 70%</title>
|
|
<link rel="icon" sizes="32x32" href="favicon_32_cb_58284776.png">
|
|
<link rel="stylesheet" href="style_cb_6b508a39.css" type="text/css">
|
|
<script src="coverage_html_cb_6fb7b396.js" defer></script>
|
|
</head>
|
|
<body class="pyfile">
|
|
<header>
|
|
<div class="content">
|
|
<h1>
|
|
<span class="text">Coverage for </span><b>pyWebLayout/io/readers/epub_reader.py</b>:
|
|
<span class="pc_cov">70%</span>
|
|
</h1>
|
|
<aside id="help_panel_wrapper">
|
|
<input id="help_panel_state" type="checkbox">
|
|
<label for="help_panel_state">
|
|
<img id="keyboard_icon" src="keybd_closed_cb_ce680311.png" alt="Show/hide keyboard shortcuts">
|
|
</label>
|
|
<div id="help_panel">
|
|
<p class="legend">Shortcuts on this page</p>
|
|
<div class="keyhelp">
|
|
<p>
|
|
<kbd>r</kbd>
|
|
<kbd>m</kbd>
|
|
<kbd>x</kbd>
|
|
<kbd>p</kbd>
|
|
toggle line displays
|
|
</p>
|
|
<p>
|
|
<kbd>j</kbd>
|
|
<kbd>k</kbd>
|
|
next/prev highlighted chunk
|
|
</p>
|
|
<p>
|
|
<kbd>0</kbd> (zero) top of page
|
|
</p>
|
|
<p>
|
|
<kbd>1</kbd> (one) first highlighted chunk
|
|
</p>
|
|
<p>
|
|
<kbd>[</kbd>
|
|
<kbd>]</kbd>
|
|
prev/next file
|
|
</p>
|
|
<p>
|
|
<kbd>u</kbd> up to the index
|
|
</p>
|
|
<p>
|
|
<kbd>?</kbd> show/hide this help
|
|
</p>
|
|
</div>
|
|
</div>
|
|
</aside>
|
|
<h2>
|
|
<span class="text">286 statements </span>
|
|
<button type="button" class="run button_toggle_run" value="run" data-shortcut="r" title="Toggle lines run">210<span class="text"> run</span></button>
|
|
<button type="button" class="mis show_mis button_toggle_mis" value="mis" data-shortcut="m" title="Toggle lines missing">76<span class="text"> missing</span></button>
|
|
<button type="button" class="exc show_exc button_toggle_exc" value="exc" data-shortcut="x" title="Toggle lines excluded">18<span class="text"> excluded</span></button>
|
|
<button type="button" class="par run show_par button_toggle_par" value="par" data-shortcut="p" title="Toggle lines partially run">27<span class="text"> partial</span></button>
|
|
</h2>
|
|
<p class="text">
|
|
<a id="prevFileLink" class="nav" href="z_263f2e628cef8c50___init___py.html">« prev</a>
|
|
<a id="indexLink" class="nav" href="index.html">^ index</a>
|
|
<a id="nextFileLink" class="nav" href="z_263f2e628cef8c50_html_extraction_py.html">» next</a>
|
|
|
|
<a class="nav" href="https://coverage.readthedocs.io/en/7.11.2">coverage.py v7.11.2</a>,
|
|
created at 2025-11-12 12:02 +0000
|
|
</p>
|
|
<aside class="hidden">
|
|
<button type="button" class="button_next_chunk" data-shortcut="j"></button>
|
|
<button type="button" class="button_prev_chunk" data-shortcut="k"></button>
|
|
<button type="button" class="button_top_of_page" data-shortcut="0"></button>
|
|
<button type="button" class="button_first_chunk" data-shortcut="1"></button>
|
|
<button type="button" class="button_prev_file" data-shortcut="["></button>
|
|
<button type="button" class="button_next_file" data-shortcut="]"></button>
|
|
<button type="button" class="button_to_index" data-shortcut="u"></button>
|
|
<button type="button" class="button_show_hide_help" data-shortcut="?"></button>
|
|
</aside>
|
|
</div>
|
|
</header>
|
|
<main id="source">
|
|
<p class="exc show_exc"><span class="n"><a id="t1" href="#t1">1</a></span><span class="t"><span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t2" href="#t2">2</a></span><span class="t"><span class="str">EPUB reader for pyWebLayout.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t3" href="#t3">3</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t4" href="#t4">4</a></span><span class="t"><span class="str">This module provides functionality for reading EPUB documents and converting them</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t5" href="#t5">5</a></span><span class="t"><span class="str">to pyWebLayout's abstract document model.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t6" href="#t6">6</a></span><span class="t"><span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t7" href="#t7">7</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t8" href="#t8">8</a></span><span class="t"><span class="key">import</span> <span class="nam">os</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t9" href="#t9">9</a></span><span class="t"><span class="key">import</span> <span class="nam">zipfile</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t10" href="#t10">10</a></span><span class="t"><span class="key">import</span> <span class="nam">tempfile</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t11" href="#t11">11</a></span><span class="t"><span class="key">from</span> <span class="nam">typing</span> <span class="key">import</span> <span class="nam">Dict</span><span class="op">,</span> <span class="nam">List</span><span class="op">,</span> <span class="nam">Optional</span><span class="op">,</span> <span class="nam">Any</span><span class="op">,</span> <span class="nam">Callable</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t12" href="#t12">12</a></span><span class="t"><span class="key">import</span> <span class="nam">xml</span><span class="op">.</span><span class="nam">etree</span><span class="op">.</span><span class="nam">ElementTree</span> <span class="key">as</span> <span class="nam">ET</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t13" href="#t13">13</a></span><span class="t"><span class="key">import</span> <span class="nam">urllib</span><span class="op">.</span><span class="nam">parse</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t14" href="#t14">14</a></span><span class="t"><span class="key">from</span> <span class="nam">PIL</span> <span class="key">import</span> <span class="nam">Image</span> <span class="key">as</span> <span class="nam">PILImage</span><span class="op">,</span> <span class="nam">ImageOps</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t15" href="#t15">15</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t16" href="#t16">16</a></span><span class="t"><span class="key">from</span> <span class="nam">pyWebLayout</span><span class="op">.</span><span class="nam">abstract</span><span class="op">.</span><span class="nam">document</span> <span class="key">import</span> <span class="nam">Book</span><span class="op">,</span> <span class="nam">Chapter</span><span class="op">,</span> <span class="nam">MetadataType</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t17" href="#t17">17</a></span><span class="t"><span class="key">from</span> <span class="nam">pyWebLayout</span><span class="op">.</span><span class="nam">abstract</span><span class="op">.</span><span class="nam">block</span> <span class="key">import</span> <span class="nam">PageBreak</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t18" href="#t18">18</a></span><span class="t"><span class="key">from</span> <span class="nam">pyWebLayout</span><span class="op">.</span><span class="nam">io</span><span class="op">.</span><span class="nam">readers</span><span class="op">.</span><span class="nam">html_extraction</span> <span class="key">import</span> <span class="nam">parse_html_string</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t19" href="#t19">19</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t20" href="#t20">20</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t21" href="#t21">21</a></span><span class="t"><span class="com"># XML namespaces used in EPUB files</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t22" href="#t22">22</a></span><span class="t"><span class="nam">NAMESPACES</span> <span class="op">=</span> <span class="op">{</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t23" href="#t23">23</a></span><span class="t"> <span class="str">'opf'</span><span class="op">:</span> <span class="str">'http://www.idpf.org/2007/opf'</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t24" href="#t24">24</a></span><span class="t"> <span class="str">'dc'</span><span class="op">:</span> <span class="str">'http://purl.org/dc/elements/1.1/'</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t25" href="#t25">25</a></span><span class="t"> <span class="str">'dcterms'</span><span class="op">:</span> <span class="str">'http://purl.org/dc/terms/'</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t26" href="#t26">26</a></span><span class="t"> <span class="str">'xhtml'</span><span class="op">:</span> <span class="str">'http://www.w3.org/1999/xhtml'</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t27" href="#t27">27</a></span><span class="t"> <span class="str">'ncx'</span><span class="op">:</span> <span class="str">'http://www.daisy.org/z3986/2005/ncx/'</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t28" href="#t28">28</a></span><span class="t"><span class="op">}</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t29" href="#t29">29</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t30" href="#t30">30</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t31" href="#t31">31</a></span><span class="t"><span class="key">def</span> <span class="nam">default_eink_processor</span><span class="op">(</span><span class="nam">img</span><span class="op">:</span> <span class="nam">PILImage</span><span class="op">.</span><span class="nam">Image</span><span class="op">)</span> <span class="op">-></span> <span class="nam">PILImage</span><span class="op">.</span><span class="nam">Image</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t32" href="#t32">32</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t33" href="#t33">33</a></span><span class="t"><span class="str"> Process image for 4-bit e-ink display using PIL only.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t34" href="#t34">34</a></span><span class="t"><span class="str"> Applies histogram equalization and 4-bit quantization.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t35" href="#t35">35</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t36" href="#t36">36</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t37" href="#t37">37</a></span><span class="t"><span class="str"> img: PIL Image to process</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t38" href="#t38">38</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t39" href="#t39">39</a></span><span class="t"><span class="str"> Returns:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t40" href="#t40">40</a></span><span class="t"><span class="str"> Processed PIL Image in L mode (grayscale) with 4-bit quantization</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t41" href="#t41">41</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t42" href="#t42">42</a></span><span class="t"> <span class="com"># Convert to grayscale if needed</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t43" href="#t43">43</a></span><span class="t"> <span class="key">if</span> <span class="nam">img</span><span class="op">.</span><span class="nam">mode</span> <span class="op">!=</span> <span class="str">'L'</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t44" href="#t44">44</a></span><span class="t"> <span class="nam">img</span> <span class="op">=</span> <span class="nam">img</span><span class="op">.</span><span class="nam">convert</span><span class="op">(</span><span class="str">'L'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t45" href="#t45">45</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t46" href="#t46">46</a></span><span class="t"> <span class="com"># Apply histogram equalization for contrast enhancement</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t47" href="#t47">47</a></span><span class="t"> <span class="nam">img</span> <span class="op">=</span> <span class="nam">ImageOps</span><span class="op">.</span><span class="nam">equalize</span><span class="op">(</span><span class="nam">img</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t48" href="#t48">48</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t49" href="#t49">49</a></span><span class="t"> <span class="com"># Quantize to 4-bit (16 grayscale levels: 0, 17, 34, ..., 255)</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t50" href="#t50">50</a></span><span class="t"> <span class="nam">img</span> <span class="op">=</span> <span class="nam">img</span><span class="op">.</span><span class="nam">point</span><span class="op">(</span><span class="key">lambda</span> <span class="nam">x</span><span class="op">:</span> <span class="op">(</span><span class="nam">x</span> <span class="op">//</span> <span class="num">16</span><span class="op">)</span> <span class="op">*</span> <span class="num">17</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t51" href="#t51">51</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t52" href="#t52">52</a></span><span class="t"> <span class="key">return</span> <span class="nam">img</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t53" href="#t53">53</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t54" href="#t54">54</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t55" href="#t55">55</a></span><span class="t"><span class="key">class</span> <span class="nam">EPUBReader</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t56" href="#t56">56</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t57" href="#t57">57</a></span><span class="t"><span class="str"> Reader for EPUB documents.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t58" href="#t58">58</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t59" href="#t59">59</a></span><span class="t"><span class="str"> This class extracts content from EPUB files and converts it to</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t60" href="#t60">60</a></span><span class="t"><span class="str"> pyWebLayout's abstract document model.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t61" href="#t61">61</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t62" href="#t62">62</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t63" href="#t63">63</a></span><span class="t"> <span class="key">def</span> <span class="nam">__init__</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">epub_path</span><span class="op">:</span> <span class="nam">str</span><span class="op">,</span> <span class="nam">image_processor</span><span class="op">:</span> <span class="nam">Optional</span><span class="op">[</span><span class="nam">Callable</span><span class="op">[</span><span class="op">[</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t64" href="#t64">64</a></span><span class="t"> <span class="nam">PILImage</span><span class="op">.</span><span class="nam">Image</span><span class="op">]</span><span class="op">,</span> <span class="nam">PILImage</span><span class="op">.</span><span class="nam">Image</span><span class="op">]</span><span class="op">]</span> <span class="op">=</span> <span class="nam">default_eink_processor</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t65" href="#t65">65</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t66" href="#t66">66</a></span><span class="t"><span class="str"> Initialize an EPUB reader.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t67" href="#t67">67</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t68" href="#t68">68</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t69" href="#t69">69</a></span><span class="t"><span class="str"> epub_path: Path to the EPUB file</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t70" href="#t70">70</a></span><span class="t"><span class="str"> image_processor: Optional function to process images for display optimization.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t71" href="#t71">71</a></span><span class="t"><span class="str"> Defaults to default_eink_processor for 4-bit e-ink displays.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t72" href="#t72">72</a></span><span class="t"><span class="str"> Set to None to disable image processing.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t73" href="#t73">73</a></span><span class="t"><span class="str"> Custom processor should accept and return a PIL Image.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t74" href="#t74">74</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t75" href="#t75">75</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">epub_path</span> <span class="op">=</span> <span class="nam">epub_path</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t76" href="#t76">76</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">image_processor</span> <span class="op">=</span> <span class="nam">image_processor</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t77" href="#t77">77</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span> <span class="op">=</span> <span class="nam">Book</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t78" href="#t78">78</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">temp_dir</span> <span class="op">=</span> <span class="key">None</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t79" href="#t79">79</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">content_dir</span> <span class="op">=</span> <span class="key">None</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t80" href="#t80">80</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span> <span class="op">=</span> <span class="op">{</span><span class="op">}</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t81" href="#t81">81</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">toc</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t82" href="#t82">82</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">spine</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t83" href="#t83">83</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">manifest</span> <span class="op">=</span> <span class="op">{</span><span class="op">}</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t84" href="#t84">84</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">cover_id</span> <span class="op">=</span> <span class="key">None</span> <span class="com"># ID of the cover image in manifest</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t85" href="#t85">85</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t86" href="#t86">86</a></span><span class="t"> <span class="key">def</span> <span class="nam">read</span><span class="op">(</span><span class="nam">self</span><span class="op">)</span> <span class="op">-></span> <span class="nam">Book</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t87" href="#t87">87</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t88" href="#t88">88</a></span><span class="t"><span class="str"> Read the EPUB file and convert it to a Book.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t89" href="#t89">89</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t90" href="#t90">90</a></span><span class="t"><span class="str"> Returns:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t91" href="#t91">91</a></span><span class="t"><span class="str"> Book: The parsed book</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t92" href="#t92">92</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t93" href="#t93">93</a></span><span class="t"> <span class="key">try</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t94" href="#t94">94</a></span><span class="t"> <span class="com"># Extract the EPUB file</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t95" href="#t95">95</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">temp_dir</span> <span class="op">=</span> <span class="nam">tempfile</span><span class="op">.</span><span class="nam">mkdtemp</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t96" href="#t96">96</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_extract_epub</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t97" href="#t97">97</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_parse_package_document</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t98" href="#t98">98</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_parse_toc</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t99" href="#t99">99</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_create_book</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t100" href="#t100">100</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t101" href="#t101">101</a></span><span class="t"> <span class="com"># Add chapters to the book</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t102" href="#t102">102</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_add_chapters</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t103" href="#t103">103</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t104" href="#t104">104</a></span><span class="t"> <span class="com"># Process images for e-ink display optimization</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t105" href="#t105">105</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_process_content_images</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t106" href="#t106">106</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t107" href="#t107">107</a></span><span class="t"> <span class="key">return</span> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t108" href="#t108">108</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t109" href="#t109">109</a></span><span class="t"> <span class="key">finally</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t110" href="#t110">110</a></span><span class="t"> <span class="com"># Clean up temporary files</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t111" href="#t111">111</a></span><span class="t"> <span class="key">if</span> <span class="nam">self</span><span class="op">.</span><span class="nam">temp_dir</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t112" href="#t112">112</a></span><span class="t"> <span class="key">import</span> <span class="nam">shutil</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t113" href="#t113">113</a></span><span class="t"> <span class="nam">shutil</span><span class="op">.</span><span class="nam">rmtree</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">temp_dir</span><span class="op">,</span> <span class="nam">ignore_errors</span><span class="op">=</span><span class="key">True</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t114" href="#t114">114</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t115" href="#t115">115</a></span><span class="t"> <span class="key">def</span> <span class="nam">_extract_epub</span><span class="op">(</span><span class="nam">self</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t116" href="#t116">116</a></span><span class="t"> <span class="str">"""Extract the EPUB file to a temporary directory."""</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t117" href="#t117">117</a></span><span class="t"> <span class="key">with</span> <span class="nam">zipfile</span><span class="op">.</span><span class="nam">ZipFile</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">epub_path</span><span class="op">,</span> <span class="str">'r'</span><span class="op">)</span> <span class="key">as</span> <span class="nam">zip_ref</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t118" href="#t118">118</a></span><span class="t"> <span class="nam">zip_ref</span><span class="op">.</span><span class="nam">extractall</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">temp_dir</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t119" href="#t119">119</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t120" href="#t120">120</a></span><span class="t"> <span class="com"># Find the content directory (typically OEBPS or OPS)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t121" href="#t121">121</a></span><span class="t"> <span class="nam">container_path</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">temp_dir</span><span class="op">,</span> <span class="str">'META-INF'</span><span class="op">,</span> <span class="str">'container.xml'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t122" href="#t122">122</a></span><span class="t"> <span class="key">if</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">exists</span><span class="op">(</span><span class="nam">container_path</span><span class="op">)</span><span class="op">:</span> </span><span class="r"><span class="annotate short">122 ↛ 136</span><span class="annotate long">line 122 didn't jump to line 136 because the condition on line 122 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t123" href="#t123">123</a></span><span class="t"> <span class="nam">tree</span> <span class="op">=</span> <span class="nam">ET</span><span class="op">.</span><span class="nam">parse</span><span class="op">(</span><span class="nam">container_path</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t124" href="#t124">124</a></span><span class="t"> <span class="nam">root</span> <span class="op">=</span> <span class="nam">tree</span><span class="op">.</span><span class="nam">getroot</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t125" href="#t125">125</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t126" href="#t126">126</a></span><span class="t"> <span class="com"># Get the path to the package document (content.opf)</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t127" href="#t127">127</a></span><span class="t"> <span class="key">for</span> <span class="nam">rootfile</span> <span class="key">in</span> <span class="nam">root</span><span class="op">.</span><span class="nam">findall</span><span class="op">(</span> </span><span class="r"><span class="annotate short">127 ↛ 136</span><span class="annotate long">line 127 didn't jump to line 136 because the loop on line 127 didn't complete</span></span></p>
|
|
<p class="par par2 ru2 show_par"><span class="n"><a id="t128" href="#t128">128</a></span><span class="t"> <span class="str">'.//{urn:oasis:names:tc:opendocument:xmlns:container}rootfile'</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t129" href="#t129">129</a></span><span class="t"> <span class="nam">full_path</span> <span class="op">=</span> <span class="nam">rootfile</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'full-path'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t130" href="#t130">130</a></span><span class="t"> <span class="key">if</span> <span class="nam">full_path</span><span class="op">:</span> </span><span class="r"><span class="annotate short">130 ↛ 127</span><span class="annotate long">line 130 didn't jump to line 127 because the condition on line 130 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t131" href="#t131">131</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">content_dir</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">dirname</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t132" href="#t132">132</a></span><span class="t"> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">temp_dir</span><span class="op">,</span> <span class="nam">full_path</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t133" href="#t133">133</a></span><span class="t"> <span class="key">return</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t134" href="#t134">134</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t135" href="#t135">135</a></span><span class="t"> <span class="com"># Fallback: look for common content directories</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t136" href="#t136">136</a></span><span class="t"> <span class="key">for</span> <span class="nam">content_dir</span> <span class="key">in</span> <span class="op">[</span><span class="str">'OEBPS'</span><span class="op">,</span> <span class="str">'OPS'</span><span class="op">,</span> <span class="str">'Content'</span><span class="op">]</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t137" href="#t137">137</a></span><span class="t"> <span class="key">if</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">exists</span><span class="op">(</span><span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">temp_dir</span><span class="op">,</span> <span class="nam">content_dir</span><span class="op">)</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t138" href="#t138">138</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">content_dir</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">temp_dir</span><span class="op">,</span> <span class="nam">content_dir</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t139" href="#t139">139</a></span><span class="t"> <span class="key">return</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t140" href="#t140">140</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t141" href="#t141">141</a></span><span class="t"> <span class="com"># If no content directory found, use the root</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t142" href="#t142">142</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">content_dir</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">temp_dir</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t143" href="#t143">143</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t144" href="#t144">144</a></span><span class="t"> <span class="key">def</span> <span class="nam">_parse_package_document</span><span class="op">(</span><span class="nam">self</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t145" href="#t145">145</a></span><span class="t"> <span class="str">"""Parse the package document (content.opf)."""</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t146" href="#t146">146</a></span><span class="t"> <span class="com"># Find the package document</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t147" href="#t147">147</a></span><span class="t"> <span class="nam">opf_path</span> <span class="op">=</span> <span class="key">None</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t148" href="#t148">148</a></span><span class="t"> <span class="key">for</span> <span class="nam">root</span><span class="op">,</span> <span class="nam">dirs</span><span class="op">,</span> <span class="nam">files</span> <span class="key">in</span> <span class="nam">os</span><span class="op">.</span><span class="nam">walk</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">content_dir</span><span class="op">)</span><span class="op">:</span> </span><span class="r"><span class="annotate short">148 ↛ 156</span><span class="annotate long">line 148 didn't jump to line 156 because the loop on line 148 didn't complete</span></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t149" href="#t149">149</a></span><span class="t"> <span class="key">for</span> <span class="nam">file</span> <span class="key">in</span> <span class="nam">files</span><span class="op">:</span> </span><span class="r"><span class="annotate short">149 ↛ 153</span><span class="annotate long">line 149 didn't jump to line 153 because the loop on line 149 didn't complete</span></span></p>
|
|
<p class="run"><span class="n"><a id="t150" href="#t150">150</a></span><span class="t"> <span class="key">if</span> <span class="nam">file</span><span class="op">.</span><span class="nam">endswith</span><span class="op">(</span><span class="str">'.opf'</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t151" href="#t151">151</a></span><span class="t"> <span class="nam">opf_path</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">root</span><span class="op">,</span> <span class="nam">file</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t152" href="#t152">152</a></span><span class="t"> <span class="key">break</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t153" href="#t153">153</a></span><span class="t"> <span class="key">if</span> <span class="nam">opf_path</span><span class="op">:</span> </span><span class="r"><span class="annotate short">153 ↛ 148</span><span class="annotate long">line 153 didn't jump to line 148 because the condition on line 153 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t154" href="#t154">154</a></span><span class="t"> <span class="key">break</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t155" href="#t155">155</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t156" href="#t156">156</a></span><span class="t"> <span class="key">if</span> <span class="key">not</span> <span class="nam">opf_path</span><span class="op">:</span> </span><span class="r"><span class="annotate short">156 ↛ 157</span><span class="annotate long">line 156 didn't jump to line 157 because the condition on line 156 was never true</span></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t157" href="#t157">157</a></span><span class="t"> <span class="key">raise</span> <span class="nam">ValueError</span><span class="op">(</span><span class="str">"No package document (.opf) found in EPUB"</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t158" href="#t158">158</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t159" href="#t159">159</a></span><span class="t"> <span class="com"># Parse the package document</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t160" href="#t160">160</a></span><span class="t"> <span class="nam">tree</span> <span class="op">=</span> <span class="nam">ET</span><span class="op">.</span><span class="nam">parse</span><span class="op">(</span><span class="nam">opf_path</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t161" href="#t161">161</a></span><span class="t"> <span class="nam">root</span> <span class="op">=</span> <span class="nam">tree</span><span class="op">.</span><span class="nam">getroot</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t162" href="#t162">162</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t163" href="#t163">163</a></span><span class="t"> <span class="com"># Parse metadata</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t164" href="#t164">164</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_parse_metadata</span><span class="op">(</span><span class="nam">root</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t165" href="#t165">165</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t166" href="#t166">166</a></span><span class="t"> <span class="com"># Parse manifest</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t167" href="#t167">167</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_parse_manifest</span><span class="op">(</span><span class="nam">root</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t168" href="#t168">168</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t169" href="#t169">169</a></span><span class="t"> <span class="com"># Parse spine</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t170" href="#t170">170</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_parse_spine</span><span class="op">(</span><span class="nam">root</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t171" href="#t171">171</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t172" href="#t172">172</a></span><span class="t"> <span class="key">def</span> <span class="nam">_parse_metadata</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">root</span><span class="op">:</span> <span class="nam">ET</span><span class="op">.</span><span class="nam">Element</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t173" href="#t173">173</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t174" href="#t174">174</a></span><span class="t"><span class="str"> Parse metadata from the package document.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t175" href="#t175">175</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t176" href="#t176">176</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t177" href="#t177">177</a></span><span class="t"><span class="str"> root: Root element of the package document</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t178" href="#t178">178</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t179" href="#t179">179</a></span><span class="t"> <span class="com"># Find the metadata element</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t180" href="#t180">180</a></span><span class="t"> <span class="nam">metadata_elem</span> <span class="op">=</span> <span class="nam">root</span><span class="op">.</span><span class="nam">find</span><span class="op">(</span><span class="str">'.//{{{0}}}metadata'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'opf'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t181" href="#t181">181</a></span><span class="t"> <span class="key">if</span> <span class="nam">metadata_elem</span> <span class="key">is</span> <span class="key">None</span><span class="op">:</span> </span><span class="r"><span class="annotate short">181 ↛ 182</span><span class="annotate long">line 181 didn't jump to line 182 because the condition on line 181 was never true</span></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t182" href="#t182">182</a></span><span class="t"> <span class="key">return</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t183" href="#t183">183</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t184" href="#t184">184</a></span><span class="t"> <span class="com"># Parse DC metadata</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t185" href="#t185">185</a></span><span class="t"> <span class="key">for</span> <span class="nam">elem</span> <span class="key">in</span> <span class="nam">metadata_elem</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t186" href="#t186">186</a></span><span class="t"> <span class="key">if</span> <span class="nam">elem</span><span class="op">.</span><span class="nam">tag</span><span class="op">.</span><span class="nam">startswith</span><span class="op">(</span><span class="str">'{{{0}}}'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'dc'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t187" href="#t187">187</a></span><span class="t"> <span class="com"># Get the local name (without namespace)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t188" href="#t188">188</a></span><span class="t"> <span class="nam">name</span> <span class="op">=</span> <span class="nam">elem</span><span class="op">.</span><span class="nam">tag</span><span class="op">.</span><span class="nam">split</span><span class="op">(</span><span class="str">'}'</span><span class="op">,</span> <span class="num">1</span><span class="op">)</span><span class="op">[</span><span class="num">1</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t189" href="#t189">189</a></span><span class="t"> <span class="nam">value</span> <span class="op">=</span> <span class="nam">elem</span><span class="op">.</span><span class="nam">text</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t190" href="#t190">190</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t191" href="#t191">191</a></span><span class="t"> <span class="key">if</span> <span class="nam">name</span> <span class="op">==</span> <span class="str">'title'</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t192" href="#t192">192</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'title'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">value</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t193" href="#t193">193</a></span><span class="t"> <span class="key">elif</span> <span class="nam">name</span> <span class="op">==</span> <span class="str">'creator'</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t194" href="#t194">194</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'creator'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">value</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t195" href="#t195">195</a></span><span class="t"> <span class="key">elif</span> <span class="nam">name</span> <span class="op">==</span> <span class="str">'language'</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t196" href="#t196">196</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'language'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">value</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t197" href="#t197">197</a></span><span class="t"> <span class="key">elif</span> <span class="nam">name</span> <span class="op">==</span> <span class="str">'description'</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t198" href="#t198">198</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'description'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">value</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t199" href="#t199">199</a></span><span class="t"> <span class="key">elif</span> <span class="nam">name</span> <span class="op">==</span> <span class="str">'subject'</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t200" href="#t200">200</a></span><span class="t"> <span class="key">if</span> <span class="str">'subjects'</span> <span class="key">not</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t201" href="#t201">201</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'subjects'</span><span class="op">]</span> <span class="op">=</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t202" href="#t202">202</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'subjects'</span><span class="op">]</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="nam">value</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t203" href="#t203">203</a></span><span class="t"> <span class="key">elif</span> <span class="nam">name</span> <span class="op">==</span> <span class="str">'date'</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t204" href="#t204">204</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'date'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">value</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t205" href="#t205">205</a></span><span class="t"> <span class="key">elif</span> <span class="nam">name</span> <span class="op">==</span> <span class="str">'identifier'</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t206" href="#t206">206</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'identifier'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">value</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t207" href="#t207">207</a></span><span class="t"> <span class="key">elif</span> <span class="nam">name</span> <span class="op">==</span> <span class="str">'publisher'</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t208" href="#t208">208</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'publisher'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">value</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t209" href="#t209">209</a></span><span class="t"> <span class="key">else</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t210" href="#t210">210</a></span><span class="t"> <span class="com"># Store other metadata</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t211" href="#t211">211</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="nam">name</span><span class="op">]</span> <span class="op">=</span> <span class="nam">value</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t212" href="#t212">212</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t213" href="#t213">213</a></span><span class="t"> <span class="com"># Parse meta elements for cover reference</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t214" href="#t214">214</a></span><span class="t"> <span class="key">for</span> <span class="nam">meta</span> <span class="key">in</span> <span class="nam">metadata_elem</span><span class="op">.</span><span class="nam">findall</span><span class="op">(</span><span class="str">'.//{{{0}}}meta'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'opf'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t215" href="#t215">215</a></span><span class="t"> <span class="nam">name</span> <span class="op">=</span> <span class="nam">meta</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'name'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t216" href="#t216">216</a></span><span class="t"> <span class="nam">content</span> <span class="op">=</span> <span class="nam">meta</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'content'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t217" href="#t217">217</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t218" href="#t218">218</a></span><span class="t"> <span class="key">if</span> <span class="nam">name</span> <span class="op">==</span> <span class="str">'cover'</span> <span class="key">and</span> <span class="nam">content</span><span class="op">:</span> </span><span class="r"><span class="annotate short">218 ↛ 220</span><span class="annotate long">line 218 didn't jump to line 220 because the condition on line 218 was never true</span></span></p>
|
|
<p class="pln"><span class="n"><a id="t219" href="#t219">219</a></span><span class="t"> <span class="com"># This is a reference to the cover image in the manifest</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t220" href="#t220">220</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">cover_id</span> <span class="op">=</span> <span class="nam">content</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t221" href="#t221">221</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t222" href="#t222">222</a></span><span class="t"> <span class="key">def</span> <span class="nam">_parse_manifest</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">root</span><span class="op">:</span> <span class="nam">ET</span><span class="op">.</span><span class="nam">Element</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t223" href="#t223">223</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t224" href="#t224">224</a></span><span class="t"><span class="str"> Parse manifest from the package document.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t225" href="#t225">225</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t226" href="#t226">226</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t227" href="#t227">227</a></span><span class="t"><span class="str"> root: Root element of the package document</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t228" href="#t228">228</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t229" href="#t229">229</a></span><span class="t"> <span class="com"># Find the manifest element</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t230" href="#t230">230</a></span><span class="t"> <span class="nam">manifest_elem</span> <span class="op">=</span> <span class="nam">root</span><span class="op">.</span><span class="nam">find</span><span class="op">(</span><span class="str">'.//{{{0}}}manifest'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'opf'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t231" href="#t231">231</a></span><span class="t"> <span class="key">if</span> <span class="nam">manifest_elem</span> <span class="key">is</span> <span class="key">None</span><span class="op">:</span> </span><span class="r"><span class="annotate short">231 ↛ 232</span><span class="annotate long">line 231 didn't jump to line 232 because the condition on line 231 was never true</span></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t232" href="#t232">232</a></span><span class="t"> <span class="key">return</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t233" href="#t233">233</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t234" href="#t234">234</a></span><span class="t"> <span class="com"># Parse items</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t235" href="#t235">235</a></span><span class="t"> <span class="key">for</span> <span class="nam">item</span> <span class="key">in</span> <span class="nam">manifest_elem</span><span class="op">.</span><span class="nam">findall</span><span class="op">(</span><span class="str">'.//{{{0}}}item'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'opf'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t236" href="#t236">236</a></span><span class="t"> <span class="nam">id</span> <span class="op">=</span> <span class="nam">item</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'id'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t237" href="#t237">237</a></span><span class="t"> <span class="nam">href</span> <span class="op">=</span> <span class="nam">item</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'href'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t238" href="#t238">238</a></span><span class="t"> <span class="nam">media_type</span> <span class="op">=</span> <span class="nam">item</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'media-type'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t239" href="#t239">239</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t240" href="#t240">240</a></span><span class="t"> <span class="key">if</span> <span class="nam">id</span> <span class="key">and</span> <span class="nam">href</span><span class="op">:</span> </span><span class="r"><span class="annotate short">240 ↛ 235</span><span class="annotate long">line 240 didn't jump to line 235 because the condition on line 240 was always true</span></span></p>
|
|
<p class="pln"><span class="n"><a id="t241" href="#t241">241</a></span><span class="t"> <span class="com"># Resolve relative path</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t242" href="#t242">242</a></span><span class="t"> <span class="nam">href</span> <span class="op">=</span> <span class="nam">urllib</span><span class="op">.</span><span class="nam">parse</span><span class="op">.</span><span class="nam">unquote</span><span class="op">(</span><span class="nam">href</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t243" href="#t243">243</a></span><span class="t"> <span class="nam">path</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">normpath</span><span class="op">(</span><span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">content_dir</span><span class="op">,</span> <span class="nam">href</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t244" href="#t244">244</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t245" href="#t245">245</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">manifest</span><span class="op">[</span><span class="nam">id</span><span class="op">]</span> <span class="op">=</span> <span class="op">{</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t246" href="#t246">246</a></span><span class="t"> <span class="str">'href'</span><span class="op">:</span> <span class="nam">href</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t247" href="#t247">247</a></span><span class="t"> <span class="str">'path'</span><span class="op">:</span> <span class="nam">path</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t248" href="#t248">248</a></span><span class="t"> <span class="str">'media_type'</span><span class="op">:</span> <span class="nam">media_type</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t249" href="#t249">249</a></span><span class="t"> <span class="op">}</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t250" href="#t250">250</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t251" href="#t251">251</a></span><span class="t"> <span class="key">def</span> <span class="nam">_parse_spine</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">root</span><span class="op">:</span> <span class="nam">ET</span><span class="op">.</span><span class="nam">Element</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t252" href="#t252">252</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t253" href="#t253">253</a></span><span class="t"><span class="str"> Parse spine from the package document.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t254" href="#t254">254</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t255" href="#t255">255</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t256" href="#t256">256</a></span><span class="t"><span class="str"> root: Root element of the package document</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t257" href="#t257">257</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t258" href="#t258">258</a></span><span class="t"> <span class="com"># Find the spine element</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t259" href="#t259">259</a></span><span class="t"> <span class="nam">spine_elem</span> <span class="op">=</span> <span class="nam">root</span><span class="op">.</span><span class="nam">find</span><span class="op">(</span><span class="str">'.//{{{0}}}spine'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'opf'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t260" href="#t260">260</a></span><span class="t"> <span class="key">if</span> <span class="nam">spine_elem</span> <span class="key">is</span> <span class="key">None</span><span class="op">:</span> </span><span class="r"><span class="annotate short">260 ↛ 261</span><span class="annotate long">line 260 didn't jump to line 261 because the condition on line 260 was never true</span></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t261" href="#t261">261</a></span><span class="t"> <span class="key">return</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t262" href="#t262">262</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t263" href="#t263">263</a></span><span class="t"> <span class="com"># Get the toc attribute (NCX file ID)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t264" href="#t264">264</a></span><span class="t"> <span class="nam">toc_id</span> <span class="op">=</span> <span class="nam">spine_elem</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'toc'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t265" href="#t265">265</a></span><span class="t"> <span class="key">if</span> <span class="nam">toc_id</span> <span class="key">and</span> <span class="nam">toc_id</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">manifest</span><span class="op">:</span> </span><span class="r"><span class="annotate short">265 ↛ 269</span><span class="annotate long">line 265 didn't jump to line 269 because the condition on line 265 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t266" href="#t266">266</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">toc_path</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">manifest</span><span class="op">[</span><span class="nam">toc_id</span><span class="op">]</span><span class="op">[</span><span class="str">'path'</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t267" href="#t267">267</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t268" href="#t268">268</a></span><span class="t"> <span class="com"># Parse itemrefs</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t269" href="#t269">269</a></span><span class="t"> <span class="key">for</span> <span class="nam">itemref</span> <span class="key">in</span> <span class="nam">spine_elem</span><span class="op">.</span><span class="nam">findall</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t270" href="#t270">270</a></span><span class="t"> <span class="str">'.//{{{0}}}itemref'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'opf'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t271" href="#t271">271</a></span><span class="t"> <span class="nam">idref</span> <span class="op">=</span> <span class="nam">itemref</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'idref'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t272" href="#t272">272</a></span><span class="t"> <span class="key">if</span> <span class="nam">idref</span> <span class="key">and</span> <span class="nam">idref</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">manifest</span><span class="op">:</span> </span><span class="r"><span class="annotate short">272 ↛ 269</span><span class="annotate long">line 272 didn't jump to line 269 because the condition on line 272 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t273" href="#t273">273</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">spine</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="nam">idref</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t274" href="#t274">274</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t275" href="#t275">275</a></span><span class="t"> <span class="key">def</span> <span class="nam">_parse_toc</span><span class="op">(</span><span class="nam">self</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t276" href="#t276">276</a></span><span class="t"> <span class="str">"""Parse the table of contents."""</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t277" href="#t277">277</a></span><span class="t"> <span class="key">if</span> <span class="key">not</span> <span class="nam">hasattr</span><span class="op">(</span> </span><span class="r"><span class="annotate short">277 ↛ 282</span><span class="annotate long">line 277 didn't jump to line 282 because the condition on line 277 was never true</span></span></p>
|
|
<p class="par par2 ru2 show_par"><span class="n"><a id="t278" href="#t278">278</a></span><span class="t"> <span class="nam">self</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="par par2 ru2 show_par"><span class="n"><a id="t279" href="#t279">279</a></span><span class="t"> <span class="str">'toc_path'</span><span class="op">)</span> <span class="key">or</span> <span class="key">not</span> <span class="nam">self</span><span class="op">.</span><span class="nam">toc_path</span> <span class="key">or</span> <span class="key">not</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">exists</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="par par2 ru2 show_par"><span class="n"><a id="t280" href="#t280">280</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">toc_path</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t281" href="#t281">281</a></span><span class="t"> <span class="com"># Try to find the toc.ncx file</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t282" href="#t282">282</a></span><span class="t"> <span class="key">for</span> <span class="nam">root</span><span class="op">,</span> <span class="nam">dirs</span><span class="op">,</span> <span class="nam">files</span> <span class="key">in</span> <span class="nam">os</span><span class="op">.</span><span class="nam">walk</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">content_dir</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t283" href="#t283">283</a></span><span class="t"> <span class="key">for</span> <span class="nam">file</span> <span class="key">in</span> <span class="nam">files</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t284" href="#t284">284</a></span><span class="t"> <span class="key">if</span> <span class="nam">file</span><span class="op">.</span><span class="nam">endswith</span><span class="op">(</span><span class="str">'.ncx'</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t285" href="#t285">285</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">toc_path</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">root</span><span class="op">,</span> <span class="nam">file</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t286" href="#t286">286</a></span><span class="t"> <span class="key">break</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t287" href="#t287">287</a></span><span class="t"> <span class="key">if</span> <span class="nam">hasattr</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="str">'toc_path'</span><span class="op">)</span> <span class="key">and</span> <span class="nam">self</span><span class="op">.</span><span class="nam">toc_path</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t288" href="#t288">288</a></span><span class="t"> <span class="key">break</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t289" href="#t289">289</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t290" href="#t290">290</a></span><span class="t"> <span class="key">if</span> <span class="key">not</span> <span class="nam">hasattr</span><span class="op">(</span> </span><span class="r"><span class="annotate short">290 ↛ 295</span><span class="annotate long">line 290 didn't jump to line 295 because the condition on line 290 was never true</span></span></p>
|
|
<p class="par par2 ru2 show_par"><span class="n"><a id="t291" href="#t291">291</a></span><span class="t"> <span class="nam">self</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="par par2 ru2 show_par"><span class="n"><a id="t292" href="#t292">292</a></span><span class="t"> <span class="str">'toc_path'</span><span class="op">)</span> <span class="key">or</span> <span class="key">not</span> <span class="nam">self</span><span class="op">.</span><span class="nam">toc_path</span> <span class="key">or</span> <span class="key">not</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">exists</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="par par2 ru2 show_par"><span class="n"><a id="t293" href="#t293">293</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">toc_path</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t294" href="#t294">294</a></span><span class="t"> <span class="com"># No TOC found</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t295" href="#t295">295</a></span><span class="t"> <span class="key">return</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t296" href="#t296">296</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t297" href="#t297">297</a></span><span class="t"> <span class="com"># Parse the NCX file</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t298" href="#t298">298</a></span><span class="t"> <span class="nam">tree</span> <span class="op">=</span> <span class="nam">ET</span><span class="op">.</span><span class="nam">parse</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">toc_path</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t299" href="#t299">299</a></span><span class="t"> <span class="nam">root</span> <span class="op">=</span> <span class="nam">tree</span><span class="op">.</span><span class="nam">getroot</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t300" href="#t300">300</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t301" href="#t301">301</a></span><span class="t"> <span class="com"># Parse navMap</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t302" href="#t302">302</a></span><span class="t"> <span class="nam">nav_map</span> <span class="op">=</span> <span class="nam">root</span><span class="op">.</span><span class="nam">find</span><span class="op">(</span><span class="str">'.//{{{0}}}navMap'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'ncx'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t303" href="#t303">303</a></span><span class="t"> <span class="key">if</span> <span class="nam">nav_map</span> <span class="key">is</span> <span class="key">None</span><span class="op">:</span> </span><span class="r"><span class="annotate short">303 ↛ 304</span><span class="annotate long">line 303 didn't jump to line 304 because the condition on line 303 was never true</span></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t304" href="#t304">304</a></span><span class="t"> <span class="key">return</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t305" href="#t305">305</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t306" href="#t306">306</a></span><span class="t"> <span class="com"># Parse navPoints</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t307" href="#t307">307</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_parse_nav_points</span><span class="op">(</span><span class="nam">nav_map</span><span class="op">,</span> <span class="op">[</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t308" href="#t308">308</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t309" href="#t309">309</a></span><span class="t"> <span class="key">def</span> <span class="nam">_parse_nav_points</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">parent</span><span class="op">:</span> <span class="nam">ET</span><span class="op">.</span><span class="nam">Element</span><span class="op">,</span> <span class="nam">path</span><span class="op">:</span> <span class="nam">List</span><span class="op">[</span><span class="nam">Dict</span><span class="op">[</span><span class="nam">str</span><span class="op">,</span> <span class="nam">Any</span><span class="op">]</span><span class="op">]</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t310" href="#t310">310</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t311" href="#t311">311</a></span><span class="t"><span class="str"> Recursively parse navPoints from the NCX file.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t312" href="#t312">312</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t313" href="#t313">313</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t314" href="#t314">314</a></span><span class="t"><span class="str"> parent: Parent element containing navPoints</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t315" href="#t315">315</a></span><span class="t"><span class="str"> path: Current path in the TOC hierarchy</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t316" href="#t316">316</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t317" href="#t317">317</a></span><span class="t"> <span class="key">for</span> <span class="nam">nav_point</span> <span class="key">in</span> <span class="nam">parent</span><span class="op">.</span><span class="nam">findall</span><span class="op">(</span><span class="str">'.//{{{0}}}navPoint'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'ncx'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t318" href="#t318">318</a></span><span class="t"> <span class="com"># Get navPoint attributes</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t319" href="#t319">319</a></span><span class="t"> <span class="nam">id</span> <span class="op">=</span> <span class="nam">nav_point</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'id'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t320" href="#t320">320</a></span><span class="t"> <span class="nam">play_order</span> <span class="op">=</span> <span class="nam">nav_point</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'playOrder'</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t321" href="#t321">321</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t322" href="#t322">322</a></span><span class="t"> <span class="com"># Get navLabel</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t323" href="#t323">323</a></span><span class="t"> <span class="nam">nav_label</span> <span class="op">=</span> <span class="nam">nav_point</span><span class="op">.</span><span class="nam">find</span><span class="op">(</span><span class="str">'.//{{{0}}}navLabel'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'ncx'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t324" href="#t324">324</a></span><span class="t"> <span class="nam">text_elem</span> <span class="op">=</span> <span class="nam">nav_label</span><span class="op">.</span><span class="nam">find</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t325" href="#t325">325</a></span><span class="t"> <span class="str">'.//{{{0}}}text'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'ncx'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> <span class="key">if</span> <span class="nam">nav_label</span> <span class="key">else</span> <span class="key">None</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t326" href="#t326">326</a></span><span class="t"> <span class="nam">label</span> <span class="op">=</span> <span class="nam">text_elem</span><span class="op">.</span><span class="nam">text</span> <span class="key">if</span> <span class="nam">text_elem</span> <span class="key">is</span> <span class="key">not</span> <span class="key">None</span> <span class="key">else</span> <span class="str">""</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t327" href="#t327">327</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t328" href="#t328">328</a></span><span class="t"> <span class="com"># Get content</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t329" href="#t329">329</a></span><span class="t"> <span class="nam">content</span> <span class="op">=</span> <span class="nam">nav_point</span><span class="op">.</span><span class="nam">find</span><span class="op">(</span><span class="str">'.//{{{0}}}content'</span><span class="op">.</span><span class="nam">format</span><span class="op">(</span><span class="nam">NAMESPACES</span><span class="op">[</span><span class="str">'ncx'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t330" href="#t330">330</a></span><span class="t"> <span class="nam">src</span> <span class="op">=</span> <span class="nam">content</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'src'</span><span class="op">)</span> <span class="key">if</span> <span class="nam">content</span> <span class="key">is</span> <span class="key">not</span> <span class="key">None</span> <span class="key">else</span> <span class="str">""</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t331" href="#t331">331</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t332" href="#t332">332</a></span><span class="t"> <span class="com"># Create a TOC entry</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t333" href="#t333">333</a></span><span class="t"> <span class="nam">entry</span> <span class="op">=</span> <span class="op">{</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t334" href="#t334">334</a></span><span class="t"> <span class="str">'id'</span><span class="op">:</span> <span class="nam">id</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t335" href="#t335">335</a></span><span class="t"> <span class="str">'label'</span><span class="op">:</span> <span class="nam">label</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t336" href="#t336">336</a></span><span class="t"> <span class="str">'src'</span><span class="op">:</span> <span class="nam">src</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t337" href="#t337">337</a></span><span class="t"> <span class="str">'play_order'</span><span class="op">:</span> <span class="nam">play_order</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t338" href="#t338">338</a></span><span class="t"> <span class="str">'children'</span><span class="op">:</span> <span class="op">[</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t339" href="#t339">339</a></span><span class="t"> <span class="op">}</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t340" href="#t340">340</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t341" href="#t341">341</a></span><span class="t"> <span class="com"># Add to TOC</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t342" href="#t342">342</a></span><span class="t"> <span class="key">if</span> <span class="nam">path</span><span class="op">:</span> </span><span class="r"><span class="annotate short">342 ↛ 343</span><span class="annotate long">line 342 didn't jump to line 343 because the condition on line 342 was never true</span></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t343" href="#t343">343</a></span><span class="t"> <span class="nam">path</span><span class="op">[</span><span class="op">-</span><span class="num">1</span><span class="op">]</span><span class="op">[</span><span class="str">'children'</span><span class="op">]</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="nam">entry</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t344" href="#t344">344</a></span><span class="t"> <span class="key">else</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t345" href="#t345">345</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">toc</span><span class="op">.</span><span class="nam">append</span><span class="op">(</span><span class="nam">entry</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t346" href="#t346">346</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t347" href="#t347">347</a></span><span class="t"> <span class="com"># Parse child navPoints</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t348" href="#t348">348</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_parse_nav_points</span><span class="op">(</span><span class="nam">nav_point</span><span class="op">,</span> <span class="nam">path</span> <span class="op">+</span> <span class="op">[</span><span class="nam">entry</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t349" href="#t349">349</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t350" href="#t350">350</a></span><span class="t"> <span class="key">def</span> <span class="nam">_create_book</span><span class="op">(</span><span class="nam">self</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t351" href="#t351">351</a></span><span class="t"> <span class="str">"""Create a Book object from the parsed metadata."""</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t352" href="#t352">352</a></span><span class="t"> <span class="com"># Set book metadata</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t353" href="#t353">353</a></span><span class="t"> <span class="key">if</span> <span class="str">'title'</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">:</span> </span><span class="r"><span class="annotate short">353 ↛ 356</span><span class="annotate long">line 353 didn't jump to line 356 because the condition on line 353 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t354" href="#t354">354</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">set_title</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'title'</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t355" href="#t355">355</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t356" href="#t356">356</a></span><span class="t"> <span class="key">if</span> <span class="str">'creator'</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t357" href="#t357">357</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">set_metadata</span><span class="op">(</span><span class="nam">MetadataType</span><span class="op">.</span><span class="nam">AUTHOR</span><span class="op">,</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'creator'</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t358" href="#t358">358</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t359" href="#t359">359</a></span><span class="t"> <span class="key">if</span> <span class="str">'language'</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t360" href="#t360">360</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">set_metadata</span><span class="op">(</span><span class="nam">MetadataType</span><span class="op">.</span><span class="nam">LANGUAGE</span><span class="op">,</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'language'</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t361" href="#t361">361</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t362" href="#t362">362</a></span><span class="t"> <span class="key">if</span> <span class="str">'description'</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t363" href="#t363">363</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">set_metadata</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t364" href="#t364">364</a></span><span class="t"> <span class="nam">MetadataType</span><span class="op">.</span><span class="nam">DESCRIPTION</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t365" href="#t365">365</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'description'</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t366" href="#t366">366</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t367" href="#t367">367</a></span><span class="t"> <span class="key">if</span> <span class="str">'subjects'</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t368" href="#t368">368</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">set_metadata</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t369" href="#t369">369</a></span><span class="t"> <span class="nam">MetadataType</span><span class="op">.</span><span class="nam">KEYWORDS</span><span class="op">,</span> <span class="str">', '</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t370" href="#t370">370</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'subjects'</span><span class="op">]</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t371" href="#t371">371</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t372" href="#t372">372</a></span><span class="t"> <span class="key">if</span> <span class="str">'date'</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t373" href="#t373">373</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">set_metadata</span><span class="op">(</span><span class="nam">MetadataType</span><span class="op">.</span><span class="nam">PUBLICATION_DATE</span><span class="op">,</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'date'</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t374" href="#t374">374</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t375" href="#t375">375</a></span><span class="t"> <span class="key">if</span> <span class="str">'identifier'</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">:</span> </span><span class="r"><span class="annotate short">375 ↛ 378</span><span class="annotate long">line 375 didn't jump to line 378 because the condition on line 375 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t376" href="#t376">376</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">set_metadata</span><span class="op">(</span><span class="nam">MetadataType</span><span class="op">.</span><span class="nam">IDENTIFIER</span><span class="op">,</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'identifier'</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t377" href="#t377">377</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t378" href="#t378">378</a></span><span class="t"> <span class="key">if</span> <span class="str">'publisher'</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t379" href="#t379">379</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">set_metadata</span><span class="op">(</span><span class="nam">MetadataType</span><span class="op">.</span><span class="nam">PUBLISHER</span><span class="op">,</span> <span class="nam">self</span><span class="op">.</span><span class="nam">metadata</span><span class="op">[</span><span class="str">'publisher'</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t380" href="#t380">380</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t381" href="#t381">381</a></span><span class="t"> <span class="key">def</span> <span class="nam">_add_cover_chapter</span><span class="op">(</span><span class="nam">self</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t382" href="#t382">382</a></span><span class="t"> <span class="str">"""Add a cover chapter if a cover image is available."""</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t383" href="#t383">383</a></span><span class="t"> <span class="key">if</span> <span class="key">not</span> <span class="nam">self</span><span class="op">.</span><span class="nam">cover_id</span> <span class="key">or</span> <span class="nam">self</span><span class="op">.</span><span class="nam">cover_id</span> <span class="key">not</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">manifest</span><span class="op">:</span> </span><span class="r"><span class="annotate short">383 ↛ 387</span><span class="annotate long">line 383 didn't jump to line 387 because the condition on line 383 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t384" href="#t384">384</a></span><span class="t"> <span class="key">return</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t385" href="#t385">385</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t386" href="#t386">386</a></span><span class="t"> <span class="com"># Get the cover image path from the manifest</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t387" href="#t387">387</a></span><span class="t"> <span class="nam">cover_item</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">manifest</span><span class="op">[</span><span class="nam">self</span><span class="op">.</span><span class="nam">cover_id</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t388" href="#t388">388</a></span><span class="t"> <span class="nam">cover_path</span> <span class="op">=</span> <span class="nam">cover_item</span><span class="op">[</span><span class="str">'path'</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t389" href="#t389">389</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t390" href="#t390">390</a></span><span class="t"> <span class="com"># Check if the file exists</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t391" href="#t391">391</a></span><span class="t"> <span class="key">if</span> <span class="key">not</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">exists</span><span class="op">(</span><span class="nam">cover_path</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t392" href="#t392">392</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="fst">f"</span><span class="fst">Warning: Cover image file not found: </span><span class="op">{</span><span class="nam">cover_path</span><span class="op">}</span><span class="fst">"</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t393" href="#t393">393</a></span><span class="t"> <span class="key">return</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t394" href="#t394">394</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t395" href="#t395">395</a></span><span class="t"> <span class="com"># Create a cover chapter</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t396" href="#t396">396</a></span><span class="t"> <span class="nam">cover_chapter</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">create_chapter</span><span class="op">(</span><span class="str">"Cover"</span><span class="op">,</span> <span class="num">0</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t397" href="#t397">397</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t398" href="#t398">398</a></span><span class="t"> <span class="key">try</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t399" href="#t399">399</a></span><span class="t"> <span class="com"># Create an Image block for the cover</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t400" href="#t400">400</a></span><span class="t"> <span class="key">from</span> <span class="nam">pyWebLayout</span><span class="op">.</span><span class="nam">abstract</span><span class="op">.</span><span class="nam">block</span> <span class="key">import</span> <span class="nam">Image</span> <span class="key">as</span> <span class="nam">AbstractImage</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t401" href="#t401">401</a></span><span class="t"> <span class="key">from</span> <span class="nam">PIL</span> <span class="key">import</span> <span class="nam">Image</span> <span class="key">as</span> <span class="nam">PILImage</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t402" href="#t402">402</a></span><span class="t"> <span class="key">import</span> <span class="nam">io</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t403" href="#t403">403</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t404" href="#t404">404</a></span><span class="t"> <span class="com"># Load the image into memory before the temp directory is cleaned up</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t405" href="#t405">405</a></span><span class="t"> <span class="com"># We need to fully copy the image data to ensure it persists after temp</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t406" href="#t406">406</a></span><span class="t"> <span class="com"># cleanup</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t407" href="#t407">407</a></span><span class="t"> <span class="key">with</span> <span class="nam">open</span><span class="op">(</span><span class="nam">cover_path</span><span class="op">,</span> <span class="str">'rb'</span><span class="op">)</span> <span class="key">as</span> <span class="nam">f</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t408" href="#t408">408</a></span><span class="t"> <span class="nam">image_bytes</span> <span class="op">=</span> <span class="nam">f</span><span class="op">.</span><span class="nam">read</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t409" href="#t409">409</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t410" href="#t410">410</a></span><span class="t"> <span class="com"># Create PIL image from bytes in memory</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t411" href="#t411">411</a></span><span class="t"> <span class="nam">pil_image</span> <span class="op">=</span> <span class="nam">PILImage</span><span class="op">.</span><span class="nam">open</span><span class="op">(</span><span class="nam">io</span><span class="op">.</span><span class="nam">BytesIO</span><span class="op">(</span><span class="nam">image_bytes</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t412" href="#t412">412</a></span><span class="t"> <span class="nam">pil_image</span><span class="op">.</span><span class="nam">load</span><span class="op">(</span><span class="op">)</span> <span class="com"># Force loading into memory</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t413" href="#t413">413</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t414" href="#t414">414</a></span><span class="t"> <span class="com"># Create a copy to ensure all data is in memory</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t415" href="#t415">415</a></span><span class="t"> <span class="nam">pil_image</span> <span class="op">=</span> <span class="nam">pil_image</span><span class="op">.</span><span class="nam">copy</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t416" href="#t416">416</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t417" href="#t417">417</a></span><span class="t"> <span class="com"># Apply image processing if enabled</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t418" href="#t418">418</a></span><span class="t"> <span class="key">if</span> <span class="nam">self</span><span class="op">.</span><span class="nam">image_processor</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t419" href="#t419">419</a></span><span class="t"> <span class="key">try</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t420" href="#t420">420</a></span><span class="t"> <span class="nam">pil_image</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">image_processor</span><span class="op">(</span><span class="nam">pil_image</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t421" href="#t421">421</a></span><span class="t"> <span class="key">except</span> <span class="nam">Exception</span> <span class="key">as</span> <span class="nam">e</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t422" href="#t422">422</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="fst">f"</span><span class="fst">Warning: Image processing failed for cover: </span><span class="op">{</span><span class="nam">str</span><span class="op">(</span><span class="nam">e</span><span class="op">)</span><span class="op">}</span><span class="fst">"</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t423" href="#t423">423</a></span><span class="t"> <span class="com"># Continue with unprocessed image</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t424" href="#t424">424</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t425" href="#t425">425</a></span><span class="t"> <span class="com"># Create an AbstractImage block with the cover image path</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t426" href="#t426">426</a></span><span class="t"> <span class="nam">cover_image</span> <span class="op">=</span> <span class="nam">AbstractImage</span><span class="op">(</span><span class="nam">source</span><span class="op">=</span><span class="nam">cover_path</span><span class="op">,</span> <span class="nam">alt_text</span><span class="op">=</span><span class="str">"Cover Image"</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t427" href="#t427">427</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t428" href="#t428">428</a></span><span class="t"> <span class="com"># Set dimensions from the loaded image</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t429" href="#t429">429</a></span><span class="t"> <span class="nam">cover_image</span><span class="op">.</span><span class="nam">_width</span> <span class="op">=</span> <span class="nam">pil_image</span><span class="op">.</span><span class="nam">width</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t430" href="#t430">430</a></span><span class="t"> <span class="nam">cover_image</span><span class="op">.</span><span class="nam">_height</span> <span class="op">=</span> <span class="nam">pil_image</span><span class="op">.</span><span class="nam">height</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t431" href="#t431">431</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t432" href="#t432">432</a></span><span class="t"> <span class="com"># Store the loaded PIL image in the abstract image so it persists after</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t433" href="#t433">433</a></span><span class="t"> <span class="com"># temp cleanup</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t434" href="#t434">434</a></span><span class="t"> <span class="nam">cover_image</span><span class="op">.</span><span class="nam">_loaded_image</span> <span class="op">=</span> <span class="nam">pil_image</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t435" href="#t435">435</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t436" href="#t436">436</a></span><span class="t"> <span class="com"># Add the image to the cover chapter</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t437" href="#t437">437</a></span><span class="t"> <span class="nam">cover_chapter</span><span class="op">.</span><span class="nam">add_block</span><span class="op">(</span><span class="nam">cover_image</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t438" href="#t438">438</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t439" href="#t439">439</a></span><span class="t"> <span class="key">except</span> <span class="nam">Exception</span> <span class="key">as</span> <span class="nam">e</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t440" href="#t440">440</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="fst">f"</span><span class="fst">Error creating cover chapter: </span><span class="op">{</span><span class="nam">str</span><span class="op">(</span><span class="nam">e</span><span class="op">)</span><span class="op">}</span><span class="fst">"</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t441" href="#t441">441</a></span><span class="t"> <span class="key">import</span> <span class="nam">traceback</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t442" href="#t442">442</a></span><span class="t"> <span class="nam">traceback</span><span class="op">.</span><span class="nam">print_exc</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t443" href="#t443">443</a></span><span class="t"> <span class="com"># If we can't create the cover image, remove the chapter</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t444" href="#t444">444</a></span><span class="t"> <span class="key">if</span> <span class="nam">hasattr</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">,</span> <span class="str">'chapters'</span><span class="op">)</span> <span class="key">and</span> <span class="nam">cover_chapter</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">chapters</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t445" href="#t445">445</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">chapters</span><span class="op">.</span><span class="nam">remove</span><span class="op">(</span><span class="nam">cover_chapter</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t446" href="#t446">446</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t447" href="#t447">447</a></span><span class="t"> <span class="key">def</span> <span class="nam">_process_chapter_images</span><span class="op">(</span><span class="nam">self</span><span class="op">,</span> <span class="nam">chapter</span><span class="op">:</span> <span class="nam">Chapter</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t448" href="#t448">448</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t449" href="#t449">449</a></span><span class="t"><span class="str"> Load and process images in a single chapter.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t450" href="#t450">450</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t451" href="#t451">451</a></span><span class="t"><span class="str"> This method loads images from disk into memory and applies image processing.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t452" href="#t452">452</a></span><span class="t"><span class="str"> Images must be loaded before the temporary EPUB directory is cleaned up.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t453" href="#t453">453</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t454" href="#t454">454</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t455" href="#t455">455</a></span><span class="t"><span class="str"> chapter: The chapter containing images to process</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t456" href="#t456">456</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t457" href="#t457">457</a></span><span class="t"> <span class="key">from</span> <span class="nam">pyWebLayout</span><span class="op">.</span><span class="nam">abstract</span><span class="op">.</span><span class="nam">block</span> <span class="key">import</span> <span class="nam">Image</span> <span class="key">as</span> <span class="nam">AbstractImage</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t458" href="#t458">458</a></span><span class="t"> <span class="key">from</span> <span class="nam">PIL</span> <span class="key">import</span> <span class="nam">Image</span> <span class="key">as</span> <span class="nam">PILImage</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t459" href="#t459">459</a></span><span class="t"> <span class="key">import</span> <span class="nam">io</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t460" href="#t460">460</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t461" href="#t461">461</a></span><span class="t"> <span class="key">for</span> <span class="nam">block</span> <span class="key">in</span> <span class="nam">chapter</span><span class="op">.</span><span class="nam">blocks</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t462" href="#t462">462</a></span><span class="t"> <span class="key">if</span> <span class="nam">isinstance</span><span class="op">(</span><span class="nam">block</span><span class="op">,</span> <span class="nam">AbstractImage</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t463" href="#t463">463</a></span><span class="t"> <span class="com"># Load image into memory if not already loaded</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t464" href="#t464">464</a></span><span class="t"> <span class="key">if</span> <span class="key">not</span> <span class="nam">hasattr</span><span class="op">(</span><span class="nam">block</span><span class="op">,</span> <span class="str">'_loaded_image'</span><span class="op">)</span> <span class="key">or</span> <span class="key">not</span> <span class="nam">block</span><span class="op">.</span><span class="nam">_loaded_image</span><span class="op">:</span> </span><span class="r"><span class="annotate short">464 ↛ 485</span><span class="annotate long">line 464 didn't jump to line 485 because the condition on line 464 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t465" href="#t465">465</a></span><span class="t"> <span class="key">try</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t466" href="#t466">466</a></span><span class="t"> <span class="com"># Load the image from the source path</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t467" href="#t467">467</a></span><span class="t"> <span class="key">if</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">isfile</span><span class="op">(</span><span class="nam">block</span><span class="op">.</span><span class="nam">source</span><span class="op">)</span><span class="op">:</span> </span><span class="r"><span class="annotate short">467 ↛ 485</span><span class="annotate long">line 467 didn't jump to line 485 because the condition on line 467 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t468" href="#t468">468</a></span><span class="t"> <span class="key">with</span> <span class="nam">open</span><span class="op">(</span><span class="nam">block</span><span class="op">.</span><span class="nam">source</span><span class="op">,</span> <span class="str">'rb'</span><span class="op">)</span> <span class="key">as</span> <span class="nam">f</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t469" href="#t469">469</a></span><span class="t"> <span class="nam">image_bytes</span> <span class="op">=</span> <span class="nam">f</span><span class="op">.</span><span class="nam">read</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t470" href="#t470">470</a></span><span class="t"> <span class="com"># Create PIL image from bytes in memory</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t471" href="#t471">471</a></span><span class="t"> <span class="nam">pil_image</span> <span class="op">=</span> <span class="nam">PILImage</span><span class="op">.</span><span class="nam">open</span><span class="op">(</span><span class="nam">io</span><span class="op">.</span><span class="nam">BytesIO</span><span class="op">(</span><span class="nam">image_bytes</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t472" href="#t472">472</a></span><span class="t"> <span class="nam">pil_image</span><span class="op">.</span><span class="nam">load</span><span class="op">(</span><span class="op">)</span> <span class="com"># Force loading into memory</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t473" href="#t473">473</a></span><span class="t"> <span class="nam">block</span><span class="op">.</span><span class="nam">_loaded_image</span> <span class="op">=</span> <span class="nam">pil_image</span><span class="op">.</span><span class="nam">copy</span><span class="op">(</span><span class="op">)</span> <span class="com"># Create a copy to ensure it persists</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t474" href="#t474">474</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t475" href="#t475">475</a></span><span class="t"> <span class="com"># Set width and height on the block from the loaded image</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t476" href="#t476">476</a></span><span class="t"> <span class="com"># This is required for layout calculations</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t477" href="#t477">477</a></span><span class="t"> <span class="nam">block</span><span class="op">.</span><span class="nam">_width</span> <span class="op">=</span> <span class="nam">pil_image</span><span class="op">.</span><span class="nam">width</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t478" href="#t478">478</a></span><span class="t"> <span class="nam">block</span><span class="op">.</span><span class="nam">_height</span> <span class="op">=</span> <span class="nam">pil_image</span><span class="op">.</span><span class="nam">height</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t479" href="#t479">479</a></span><span class="t"> <span class="key">except</span> <span class="nam">Exception</span> <span class="key">as</span> <span class="nam">e</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t480" href="#t480">480</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="fst">f"</span><span class="fst">Warning: Failed to load image '</span><span class="op">{</span><span class="nam">block</span><span class="op">.</span><span class="nam">source</span><span class="op">}</span><span class="fst">': </span><span class="op">{</span><span class="nam">str</span><span class="op">(</span><span class="nam">e</span><span class="op">)</span><span class="op">}</span><span class="fst">"</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t481" href="#t481">481</a></span><span class="t"> <span class="com"># Continue without the image</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t482" href="#t482">482</a></span><span class="t"> <span class="key">continue</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t483" href="#t483">483</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t484" href="#t484">484</a></span><span class="t"> <span class="com"># Apply image processing if enabled and image is loaded</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t485" href="#t485">485</a></span><span class="t"> <span class="key">if</span> <span class="nam">self</span><span class="op">.</span><span class="nam">image_processor</span> <span class="key">and</span> <span class="nam">hasattr</span><span class="op">(</span><span class="nam">block</span><span class="op">,</span> <span class="str">'_loaded_image'</span><span class="op">)</span> <span class="key">and</span> <span class="nam">block</span><span class="op">.</span><span class="nam">_loaded_image</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t486" href="#t486">486</a></span><span class="t"> <span class="key">try</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t487" href="#t487">487</a></span><span class="t"> <span class="nam">block</span><span class="op">.</span><span class="nam">_loaded_image</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">image_processor</span><span class="op">(</span><span class="nam">block</span><span class="op">.</span><span class="nam">_loaded_image</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t488" href="#t488">488</a></span><span class="t"> <span class="key">except</span> <span class="nam">Exception</span> <span class="key">as</span> <span class="nam">e</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t489" href="#t489">489</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="mis mis2 show_mis"><span class="n"><a id="t490" href="#t490">490</a></span><span class="t"> <span class="fst">f"</span><span class="fst">Warning: Image processing failed for image '</span><span class="op">{</span><span class="nam">block</span><span class="op">.</span><span class="nam">alt_text</span><span class="op">}</span><span class="fst">': </span><span class="op">{</span><span class="nam">str</span><span class="op">(</span><span class="nam">e</span><span class="op">)</span><span class="op">}</span><span class="fst">"</span> </span><span class="r"></span></p>
|
|
<p class="mis mis2 show_mis"><span class="n"><a id="t491" href="#t491">491</a></span><span class="t"> <span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t492" href="#t492">492</a></span><span class="t"> <span class="com"># Continue with unprocessed image</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t493" href="#t493">493</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t494" href="#t494">494</a></span><span class="t"> <span class="key">def</span> <span class="nam">_process_content_images</span><span class="op">(</span><span class="nam">self</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t495" href="#t495">495</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t496" href="#t496">496</a></span><span class="t"><span class="str"> Load all images into memory and apply image processing.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t497" href="#t497">497</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t498" href="#t498">498</a></span><span class="t"><span class="str"> This must be called before the temporary EPUB directory is cleaned up,</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t499" href="#t499">499</a></span><span class="t"><span class="str"> to ensure images are loaded from disk into memory.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t500" href="#t500">500</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t501" href="#t501">501</a></span><span class="t"> <span class="key">for</span> <span class="nam">chapter</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">chapters</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t502" href="#t502">502</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_process_chapter_images</span><span class="op">(</span><span class="nam">chapter</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t503" href="#t503">503</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t504" href="#t504">504</a></span><span class="t"> <span class="key">def</span> <span class="nam">_add_chapters</span><span class="op">(</span><span class="nam">self</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t505" href="#t505">505</a></span><span class="t"> <span class="str">"""Add chapters to the book based on the spine and TOC."""</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t506" href="#t506">506</a></span><span class="t"> <span class="com"># Add cover chapter first if available</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t507" href="#t507">507</a></span><span class="t"> <span class="nam">self</span><span class="op">.</span><span class="nam">_add_cover_chapter</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t508" href="#t508">508</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t509" href="#t509">509</a></span><span class="t"> <span class="com"># Create a mapping from src to TOC entry</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t510" href="#t510">510</a></span><span class="t"> <span class="nam">toc_map</span> <span class="op">=</span> <span class="op">{</span><span class="op">}</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t511" href="#t511">511</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t512" href="#t512">512</a></span><span class="t"> <span class="key">def</span> <span class="nam">add_to_toc_map</span><span class="op">(</span><span class="nam">entries</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t513" href="#t513">513</a></span><span class="t"> <span class="key">for</span> <span class="nam">entry</span> <span class="key">in</span> <span class="nam">entries</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t514" href="#t514">514</a></span><span class="t"> <span class="key">if</span> <span class="nam">entry</span><span class="op">[</span><span class="str">'src'</span><span class="op">]</span><span class="op">:</span> </span><span class="r"><span class="annotate short">514 ↛ 521</span><span class="annotate long">line 514 didn't jump to line 521 because the condition on line 514 was always true</span></span></p>
|
|
<p class="pln"><span class="n"><a id="t515" href="#t515">515</a></span><span class="t"> <span class="com"># Extract the path part of the src (remove fragment)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t516" href="#t516">516</a></span><span class="t"> <span class="nam">src_parts</span> <span class="op">=</span> <span class="nam">entry</span><span class="op">[</span><span class="str">'src'</span><span class="op">]</span><span class="op">.</span><span class="nam">split</span><span class="op">(</span><span class="str">'#'</span><span class="op">,</span> <span class="num">1</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t517" href="#t517">517</a></span><span class="t"> <span class="nam">path</span> <span class="op">=</span> <span class="nam">src_parts</span><span class="op">[</span><span class="num">0</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t518" href="#t518">518</a></span><span class="t"> <span class="nam">toc_map</span><span class="op">[</span><span class="nam">path</span><span class="op">]</span> <span class="op">=</span> <span class="nam">entry</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t519" href="#t519">519</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t520" href="#t520">520</a></span><span class="t"> <span class="com"># Process children</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t521" href="#t521">521</a></span><span class="t"> <span class="key">if</span> <span class="nam">entry</span><span class="op">[</span><span class="str">'children'</span><span class="op">]</span><span class="op">:</span> </span><span class="r"><span class="annotate short">521 ↛ 522</span><span class="annotate long">line 521 didn't jump to line 522 because the condition on line 521 was never true</span></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t522" href="#t522">522</a></span><span class="t"> <span class="nam">add_to_toc_map</span><span class="op">(</span><span class="nam">entry</span><span class="op">[</span><span class="str">'children'</span><span class="op">]</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t523" href="#t523">523</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t524" href="#t524">524</a></span><span class="t"> <span class="nam">add_to_toc_map</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">toc</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t525" href="#t525">525</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t526" href="#t526">526</a></span><span class="t"> <span class="com"># Process spine items</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t527" href="#t527">527</a></span><span class="t"> <span class="com"># Start from chapter_index = 1 if cover was added, otherwise 0</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t528" href="#t528">528</a></span><span class="t"> <span class="nam">chapter_index</span> <span class="op">=</span> <span class="num">1</span> <span class="key">if</span> <span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">cover_id</span> <span class="key">and</span> <span class="nam">self</span><span class="op">.</span><span class="nam">cover_id</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">manifest</span><span class="op">)</span> <span class="key">else</span> <span class="num">0</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t529" href="#t529">529</a></span><span class="t"> <span class="key">for</span> <span class="nam">i</span><span class="op">,</span> <span class="nam">idref</span> <span class="key">in</span> <span class="nam">enumerate</span><span class="op">(</span><span class="nam">self</span><span class="op">.</span><span class="nam">spine</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t530" href="#t530">530</a></span><span class="t"> <span class="key">if</span> <span class="nam">idref</span> <span class="key">not</span> <span class="key">in</span> <span class="nam">self</span><span class="op">.</span><span class="nam">manifest</span><span class="op">:</span> </span><span class="r"><span class="annotate short">530 ↛ 531</span><span class="annotate long">line 530 didn't jump to line 531 because the condition on line 530 was never true</span></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t531" href="#t531">531</a></span><span class="t"> <span class="key">continue</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t532" href="#t532">532</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t533" href="#t533">533</a></span><span class="t"> <span class="nam">item</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">manifest</span><span class="op">[</span><span class="nam">idref</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t534" href="#t534">534</a></span><span class="t"> <span class="nam">path</span> <span class="op">=</span> <span class="nam">item</span><span class="op">[</span><span class="str">'path'</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t535" href="#t535">535</a></span><span class="t"> <span class="nam">href</span> <span class="op">=</span> <span class="nam">item</span><span class="op">[</span><span class="str">'href'</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t536" href="#t536">536</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t537" href="#t537">537</a></span><span class="t"> <span class="com"># Skip navigation files</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t538" href="#t538">538</a></span><span class="t"> <span class="key">if</span> <span class="op">(</span><span class="nam">idref</span> <span class="op">==</span> <span class="str">'nav'</span> <span class="key">or</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t539" href="#t539">539</a></span><span class="t"> <span class="nam">item</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'media_type'</span><span class="op">)</span> <span class="op">==</span> <span class="str">'application/xhtml+xml'</span> <span class="key">and</span> </span><span class="r"></span></p>
|
|
<p class="run run2"><span class="n"><a id="t540" href="#t540">540</a></span><span class="t"> <span class="op">(</span><span class="str">'nav'</span> <span class="key">in</span> <span class="nam">href</span><span class="op">.</span><span class="nam">lower</span><span class="op">(</span><span class="op">)</span> <span class="key">or</span> <span class="str">'toc'</span> <span class="key">in</span> <span class="nam">href</span><span class="op">.</span><span class="nam">lower</span><span class="op">(</span><span class="op">)</span><span class="op">)</span><span class="op">)</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t541" href="#t541">541</a></span><span class="t"> <span class="key">continue</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t542" href="#t542">542</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t543" href="#t543">543</a></span><span class="t"> <span class="com"># Check if this item is in the TOC</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t544" href="#t544">544</a></span><span class="t"> <span class="nam">chapter_title</span> <span class="op">=</span> <span class="key">None</span> </span><span class="r"></span></p>
|
|
<p class="par run show_par"><span class="n"><a id="t545" href="#t545">545</a></span><span class="t"> <span class="key">if</span> <span class="nam">href</span> <span class="key">in</span> <span class="nam">toc_map</span><span class="op">:</span> </span><span class="r"><span class="annotate short">545 ↛ 549</span><span class="annotate long">line 545 didn't jump to line 549 because the condition on line 545 was always true</span></span></p>
|
|
<p class="run"><span class="n"><a id="t546" href="#t546">546</a></span><span class="t"> <span class="nam">chapter_title</span> <span class="op">=</span> <span class="nam">toc_map</span><span class="op">[</span><span class="nam">href</span><span class="op">]</span><span class="op">[</span><span class="str">'label'</span><span class="op">]</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t547" href="#t547">547</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t548" href="#t548">548</a></span><span class="t"> <span class="com"># Create a chapter</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t549" href="#t549">549</a></span><span class="t"> <span class="nam">chapter_index</span> <span class="op">+=</span> <span class="num">1</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t550" href="#t550">550</a></span><span class="t"> <span class="nam">chapter</span> <span class="op">=</span> <span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">.</span><span class="nam">create_chapter</span><span class="op">(</span><span class="nam">chapter_title</span><span class="op">,</span> <span class="nam">chapter_index</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t551" href="#t551">551</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t552" href="#t552">552</a></span><span class="t"> <span class="com"># Parse the HTML content</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t553" href="#t553">553</a></span><span class="t"> <span class="key">try</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t554" href="#t554">554</a></span><span class="t"> <span class="com"># Read the HTML file</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t555" href="#t555">555</a></span><span class="t"> <span class="key">with</span> <span class="nam">open</span><span class="op">(</span><span class="nam">path</span><span class="op">,</span> <span class="str">'r'</span><span class="op">,</span> <span class="nam">encoding</span><span class="op">=</span><span class="str">'utf-8'</span><span class="op">)</span> <span class="key">as</span> <span class="nam">f</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t556" href="#t556">556</a></span><span class="t"> <span class="nam">html</span> <span class="op">=</span> <span class="nam">f</span><span class="op">.</span><span class="nam">read</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t557" href="#t557">557</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t558" href="#t558">558</a></span><span class="t"> <span class="com"># Get the directory of the HTML file for resolving relative paths</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t559" href="#t559">559</a></span><span class="t"> <span class="nam">html_dir</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">dirname</span><span class="op">(</span><span class="nam">path</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t560" href="#t560">560</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t561" href="#t561">561</a></span><span class="t"> <span class="com"># Parse HTML and add blocks to chapter, passing base_path for image resolution</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t562" href="#t562">562</a></span><span class="t"> <span class="nam">blocks</span> <span class="op">=</span> <span class="nam">parse_html_string</span><span class="op">(</span><span class="nam">html</span><span class="op">,</span> <span class="nam">document</span><span class="op">=</span><span class="nam">self</span><span class="op">.</span><span class="nam">book</span><span class="op">,</span> <span class="nam">base_path</span><span class="op">=</span><span class="nam">html_dir</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t563" href="#t563">563</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t564" href="#t564">564</a></span><span class="t"> <span class="com"># Copy blocks to the chapter</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t565" href="#t565">565</a></span><span class="t"> <span class="key">for</span> <span class="nam">block</span> <span class="key">in</span> <span class="nam">blocks</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t566" href="#t566">566</a></span><span class="t"> <span class="nam">chapter</span><span class="op">.</span><span class="nam">add_block</span><span class="op">(</span><span class="nam">block</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t567" href="#t567">567</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t568" href="#t568">568</a></span><span class="t"> <span class="com"># Add a PageBreak after the chapter to ensure next chapter starts on new page</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t569" href="#t569">569</a></span><span class="t"> <span class="com"># This helps maintain chapter boundaries during pagination</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t570" href="#t570">570</a></span><span class="t"> <span class="nam">chapter</span><span class="op">.</span><span class="nam">add_block</span><span class="op">(</span><span class="nam">PageBreak</span><span class="op">(</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t571" href="#t571">571</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t572" href="#t572">572</a></span><span class="t"> <span class="key">except</span> <span class="nam">Exception</span> <span class="key">as</span> <span class="nam">e</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t573" href="#t573">573</a></span><span class="t"> <span class="nam">print</span><span class="op">(</span><span class="fst">f"</span><span class="fst">Error parsing chapter </span><span class="op">{</span><span class="nam">i</span> <span class="op">+</span> <span class="num">1</span><span class="op">}</span><span class="fst">: </span><span class="op">{</span><span class="nam">str</span><span class="op">(</span><span class="nam">e</span><span class="op">)</span><span class="op">}</span><span class="fst">"</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t574" href="#t574">574</a></span><span class="t"> <span class="com"># Add an error message block</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t575" href="#t575">575</a></span><span class="t"> <span class="key">from</span> <span class="nam">pyWebLayout</span><span class="op">.</span><span class="nam">abstract</span><span class="op">.</span><span class="nam">block</span> <span class="key">import</span> <span class="nam">Paragraph</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t576" href="#t576">576</a></span><span class="t"> <span class="key">from</span> <span class="nam">pyWebLayout</span><span class="op">.</span><span class="nam">abstract</span><span class="op">.</span><span class="nam">inline</span> <span class="key">import</span> <span class="nam">Word</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t577" href="#t577">577</a></span><span class="t"> <span class="key">from</span> <span class="nam">pyWebLayout</span><span class="op">.</span><span class="nam">style</span> <span class="key">import</span> <span class="nam">Font</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t578" href="#t578">578</a></span><span class="t"> <span class="nam">error_para</span> <span class="op">=</span> <span class="nam">Paragraph</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t579" href="#t579">579</a></span><span class="t"> <span class="com"># Create a default font style for the error message</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t580" href="#t580">580</a></span><span class="t"> <span class="nam">default_font</span> <span class="op">=</span> <span class="nam">Font</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t581" href="#t581">581</a></span><span class="t"> <span class="nam">error_para</span><span class="op">.</span><span class="nam">add_word</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="mis mis2 show_mis"><span class="n"><a id="t582" href="#t582">582</a></span><span class="t"> <span class="nam">Word</span><span class="op">(</span> </span><span class="r"></span></p>
|
|
<p class="mis mis2 show_mis"><span class="n"><a id="t583" href="#t583">583</a></span><span class="t"> <span class="fst">f"</span><span class="fst">Error loading chapter: </span><span class="op">{</span><span class="nam">str</span><span class="op">(</span><span class="nam">e</span><span class="op">)</span><span class="op">}</span><span class="fst">"</span><span class="op">,</span> </span><span class="r"></span></p>
|
|
<p class="mis mis2 show_mis"><span class="n"><a id="t584" href="#t584">584</a></span><span class="t"> <span class="nam">default_font</span> </span><span class="r"></span></p>
|
|
<p class="mis mis2 show_mis"><span class="n"><a id="t585" href="#t585">585</a></span><span class="t"> <span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="mis mis2 show_mis"><span class="n"><a id="t586" href="#t586">586</a></span><span class="t"> <span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t587" href="#t587">587</a></span><span class="t"> <span class="nam">chapter</span><span class="op">.</span><span class="nam">add_block</span><span class="op">(</span><span class="nam">error_para</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t588" href="#t588">588</a></span><span class="t"> <span class="com"># Still add PageBreak even after error</span> </span><span class="r"></span></p>
|
|
<p class="mis show_mis"><span class="n"><a id="t589" href="#t589">589</a></span><span class="t"> <span class="nam">chapter</span><span class="op">.</span><span class="nam">add_block</span><span class="op">(</span><span class="nam">PageBreak</span><span class="op">(</span><span class="op">)</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t590" href="#t590">590</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="pln"><span class="n"><a id="t591" href="#t591">591</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t592" href="#t592">592</a></span><span class="t"><span class="key">def</span> <span class="nam">read_epub</span><span class="op">(</span><span class="nam">epub_path</span><span class="op">:</span> <span class="nam">str</span><span class="op">)</span> <span class="op">-></span> <span class="nam">Book</span><span class="op">:</span> </span><span class="r"></span></p>
|
|
<p class="exc show_exc"><span class="n"><a id="t593" href="#t593">593</a></span><span class="t"> <span class="str">"""</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t594" href="#t594">594</a></span><span class="t"><span class="str"> Read an EPUB file and convert it to a Book.</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t595" href="#t595">595</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t596" href="#t596">596</a></span><span class="t"><span class="str"> Args:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t597" href="#t597">597</a></span><span class="t"><span class="str"> epub_path: Path to the EPUB file</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t598" href="#t598">598</a></span><span class="t"> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t599" href="#t599">599</a></span><span class="t"><span class="str"> Returns:</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t600" href="#t600">600</a></span><span class="t"><span class="str"> Book: The parsed book</span> </span><span class="r"></span></p>
|
|
<p class="exc exc2 show_exc"><span class="n"><a id="t601" href="#t601">601</a></span><span class="t"><span class="str"> """</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t602" href="#t602">602</a></span><span class="t"> <span class="nam">reader</span> <span class="op">=</span> <span class="nam">EPUBReader</span><span class="op">(</span><span class="nam">epub_path</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
<p class="run"><span class="n"><a id="t603" href="#t603">603</a></span><span class="t"> <span class="key">return</span> <span class="nam">reader</span><span class="op">.</span><span class="nam">read</span><span class="op">(</span><span class="op">)</span> </span><span class="r"></span></p>
|
|
</main>
|
|
<footer>
|
|
<div class="content">
|
|
<p>
|
|
<a class="nav" href="z_263f2e628cef8c50___init___py.html">« prev</a>
|
|
<a class="nav" href="index.html">^ index</a>
|
|
<a class="nav" href="z_263f2e628cef8c50_html_extraction_py.html">» next</a>
|
|
|
|
<a class="nav" href="https://coverage.readthedocs.io/en/7.11.2">coverage.py v7.11.2</a>,
|
|
created at 2025-11-12 12:02 +0000
|
|
</p>
|
|
</div>
|
|
</footer>
|
|
</body>
|
|
</html>
|