felixm/ptoos-xray
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Exit with warning if epub is annotated (resolves #4)

Browse Source
This commit is contained in:
Felix Martin
2022-10-24 20:36:31 -04:00
parent d224776a9a
commit 9c200a1246
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/epub.py
View File

@@ -11,12 +11,12 @@ from rich.progress import track
from rich.console import Console from rich.console import Console
POKEMON_ID_PREFIX = "pokemon-id-" POKEMON_ID_PREFIX = "pokemon-id-"
POKEDEX_UID = "np_pokedex"
def create_pokedex_chapter(pokemon: List[Pokemon]) -> epub.EpubHtml: def create_pokedex_chapter(pokemon: List[Pokemon]) -> epub.EpubHtml:
POKEDEX_TITLE = "Pokedex" POKEDEX_TITLE = "Pokedex"
POKEDEX_FILE = "content/np_pokedex.xhtml" POKEDEX_FILE = "content/np_pokedex.xhtml"
POKEDEX_UID = "np_pokedex"
chapter = epub.EpubHtml( chapter = epub.EpubHtml(
title=POKEDEX_TITLE, file_name=POKEDEX_FILE, uid=POKEDEX_UID title=POKEDEX_TITLE, file_name=POKEDEX_FILE, uid=POKEDEX_UID
) )
@@ -95,6 +95,11 @@ def patch(epub_filename: str, pokemon: List[Pokemon]):
if isinstance(b, epub.EpubHtml) if isinstance(b, epub.EpubHtml)
if b.id.startswith("np_") if b.id.startswith("np_")
] ]
if [c for c in chapters if c.id == POKEDEX_UID]:
logging.warning(f"It looks like '{epub_filename}' already has a Pokedex.")
sys.exit(1)
for c in track(chapters, description="Add Pokemon links to chapters"): for c in track(chapters, description="Add Pokemon links to chapters"):
patch_chapter(c, pokemon_lookup) patch_chapter(c, pokemon_lookup)