Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add some type annotations

Browse Source
This commit is contained in:
Kovid Goyal 2025-02-21 14:46:51 +05:30
parent 08524e760b
commit 42f4676771
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
2 changed files with 27 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/calibre/ebooks/oeb/polish/container.py
View File

@ -1577,7 +1577,7 @@ class AZW3Container(Container):
# }}} # }}}
def get_container(path, log=None, tdir=None, tweak_mode=False): def get_container(path, log=None, tdir=None, tweak_mode=False) -> Container:
if log is None: if log is None:
log = default_log log = default_log
try: try:

42
src/calibre/ebooks/oeb/polish/kepubify.py
View File

@ -22,7 +22,7 @@ from lxml import etree
from calibre.ebooks.metadata import authors_to_string from calibre.ebooks.metadata import authors_to_string
from calibre.ebooks.oeb.base import OEB_DOCS, XHTML, XPath, escape_cdata from calibre.ebooks.oeb.base import OEB_DOCS, XHTML, XPath, escape_cdata
from calibre.ebooks.oeb.parse_utils import barename, merge_multiple_html_heads_and_bodies from calibre.ebooks.oeb.parse_utils import barename, merge_multiple_html_heads_and_bodies
from calibre.ebooks.oeb.polish.container import get_container from calibre.ebooks.oeb.polish.container import Container, get_container
from calibre.ebooks.oeb.polish.cover import find_cover_image, find_cover_image3, find_cover_page from calibre.ebooks.oeb.polish.cover import find_cover_image, find_cover_image3, find_cover_page
from calibre.ebooks.oeb.polish.parsing import parse from calibre.ebooks.oeb.polish.parsing import parse
from calibre.ebooks.oeb.polish.tts import lang_for_elem from calibre.ebooks.oeb.polish.tts import lang_for_elem
@ -254,7 +254,7 @@ def is_probably_a_title_page(root):
return (num_images + num_svgs == 1 and textlen <= 10) or (textlen <= 50 and (num_images + num_svgs) < 1) return (num_images + num_svgs == 1 and textlen <= 10) or (textlen <= 50 and (num_images + num_svgs) < 1)
def add_dummy_title_page(container, cover_image_name): def add_dummy_title_page(container: Container, cover_image_name: str) -> None:
html = f'''\ html = f'''\
<?xml version='1.0' encoding='utf-8'?> <?xml version='1.0' encoding='utf-8'?>
<html xmlns="http://www.w3.org/1999/xhtml" lang="en" xml:lang="en"> <html xmlns="http://www.w3.org/1999/xhtml" lang="en" xml:lang="en">
@ -291,7 +291,7 @@ def add_dummy_title_page(container, cover_image_name):
container.apply_unique_properties(titlepage_name, 'calibre:title-page') container.apply_unique_properties(titlepage_name, 'calibre:title-page')
def remove_dummy_title_page(container): def remove_dummy_title_page(container: Container) -> None:
for name, is_linear in container.spine_names(): for name, is_linear in container.spine_names():
if is_linear: if is_linear:
if DUMMY_TITLE_PAGE_NAME in name: if DUMMY_TITLE_PAGE_NAME in name:
@ -299,7 +299,7 @@ def remove_dummy_title_page(container):
break break
def first_spine_item_is_probably_cover(container) -> bool: def first_spine_item_is_probably_title_page(container: Container) -> bool:
for name, is_linear in container.spine_names: for name, is_linear in container.spine_names:
fname = name.split('/')[-1] fname = name.split('/')[-1]
if is_linear: if is_linear:
@ -310,13 +310,13 @@ def first_spine_item_is_probably_cover(container) -> bool:
return False return False
def kepubify_container(container, max_workers=0): def kepubify_container(container: Container, max_workers=0):
remove_dummy_title_page(container) remove_dummy_title_page(container)
metadata_lang = container.mi.language metadata_lang = container.mi.language
cover_image_name = find_cover_image(container) or find_cover_image3(container) cover_image_name = find_cover_image(container) or find_cover_image3(container)
if cover_image_name: if cover_image_name:
container.apply_unique_properties(cover_image_name, 'cover-image') container.apply_unique_properties(cover_image_name, 'cover-image')
if not find_cover_page(container) and not first_spine_item_is_probably_cover(container): if not find_cover_page(container) and not first_spine_item_is_probably_title_page(container):
add_dummy_title_page(container, cover_image_name) add_dummy_title_page(container, cover_image_name)
names_that_need_work = tuple(name for name, mt in container.mime_map.items() if mt in OEB_DOCS) names_that_need_work = tuple(name for name, mt in container.mime_map.items() if mt in OEB_DOCS)
num_workers = calculate_number_of_workers(names_that_need_work, container, max_workers) num_workers = calculate_number_of_workers(names_that_need_work, container, max_workers)
@ -331,11 +331,24 @@ def kepubify_container(container, max_workers=0):
future.result() future.result()
def kepubify_path(path, outpath='', max_workers=0, allow_overwrite=False):
container = get_container(path, tweak_mode=True)
kepubify_container(container, max_workers=max_workers)
base, ext = os.path.splitext(path)
outpath = outpath or base + '.kepub'
c = 0
while not allow_overwrite and outpath == path:
c += 1
outpath = f'{base} - {c}.kepub'
container.commit(output=outpath)
return outpath
def profile(): def profile():
from calibre.ptempfile import TemporaryDirectory from calibre.ptempfile import TemporaryDirectory
path = sys.argv[-1] path = sys.argv[-1]
with TemporaryDirectory() as tdir, Profiler(): with TemporaryDirectory() as tdir, Profiler():
main(path, max_workers=1) kepubify_path(path, max_workers=1)
def develop(): def develop():
@ -344,21 +357,18 @@ def develop():
from calibre.ptempfile import TemporaryDirectory from calibre.ptempfile import TemporaryDirectory
path = sys.argv[-1] path = sys.argv[-1]
with TemporaryDirectory() as tdir: with TemporaryDirectory() as tdir:
outpath = main(path, max_workers=1) outpath = kepubify_path(path, max_workers=1)
with ZipFile(outpath) as zf: with ZipFile(outpath) as zf:
zf.extractall(tdir) zf.extractall(tdir)
print('Extracted to:', tdir) print('Extracted to:', tdir)
input('Press Enter to quit') input('Press Enter to quit')
def main(path, max_workers=0): def main(args):
container = get_container(path, tweak_mode=True) for path in args[1:]:
kepubify_container(container, max_workers=max_workers) outpath = kepubify_path(path)
base, ext = os.path.splitext(path) print(f'{path} converted and saved as: {outpath}')
outpath = base + '.kepub'
container.commit(output=outpath)
return outpath
if __name__ == '__main__': if __name__ == '__main__':
main(sys.argv[-1]) main(sys.argv)