def print_spine() -> int:
"""
Entry point for `se print-spine`
"""
parser = argparse.ArgumentParser(description="Print the <spine> element for the given Standard Ebooks source directory to standard output, for use in that directory’s content.opf.")
parser.add_argument("-i", "--in-place", action="store_true", help="overwrite the <spine> element in content.opf instead of printing to stdout")
parser.add_argument("directories", metavar="DIRECTORY", nargs="+", help="a Standard Ebooks source directory")
args = parser.parse_args()
if not args.in_place and len(args.directories) > 1:
se.print_error("Multiple directories are only allowed with the [bash]--in-place[/] option.")
return se.InvalidArgumentsException.code
for directory in args.directories:
try:
se_epub = SeEpub(directory)
except se.SeException as ex:
se.print_error(ex)
return ex.code
if args.in_place:
se_epub.metadata_xml = regex.sub(r"\s*<spine>.+?</spine>", "\n\t" + "\n\t".join(se_epub.generate_spine().splitlines()), se_epub.metadata_xml, flags=regex.DOTALL)
with open(se_epub.metadata_file_path, "r+", encoding="utf-8") as file:
file.write(se_epub.metadata_xml)
file.truncate()
else:
print(se_epub.generate_spine())
return 0
def build_spine(plain_output: bool) -> int:
"""
Entry point for `se build-spine`
"""
parser = argparse.ArgumentParser(
description=
"Generate the <spine> element for the given Standard Ebooks source directory and write it to the ebook’s metadata file."
)
parser.add_argument(
"-s",
"--stdout",
action="store_true",
help="print to stdout instead of writing to the metadata file")
parser.add_argument("directories",
metavar="DIRECTORY",
nargs="+",
help="a Standard Ebooks source directory")
args = parser.parse_args()
if args.stdout and len(args.directories) > 1:
se.print_error(
"Multiple directories are only allowed without the [bash]--stdout[/] option.",
plain_output=plain_output)
return se.InvalidArgumentsException.code
for directory in args.directories:
try:
se_epub = SeEpub(directory)
if args.stdout:
print(se_epub.generate_spine().to_string())
else:
nodes = se_epub.metadata_dom.xpath("/package/spine")
if nodes:
for node in nodes:
node.replace_with(se_epub.generate_spine())
else:
for node in se_epub.metadata_dom.xpath("/package"):
node.append(se_epub.generate_spine())
with open(se_epub.metadata_file_path, "w",
encoding="utf-8") as file:
file.write(
se.formatting.format_xml(
se_epub.metadata_dom.to_string()))
except se.SeException as ex:
se.print_error(ex)
return ex.code
return 0