voussoir/epubfile
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Let covercomesfirst and normalize take many books via glob.

Browse source
This commit is contained in:
Ethan Dalool 2019-08-18 19:55:41 -07:00
parent 7b4f4cdb1f
commit cf08ce3570
1 changed files with 18 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
epubfile.py
View file

@ -888,6 +888,7 @@ class Epub:
# COMMAND LINE TOOLS # COMMAND LINE TOOLS
################################################################################ ################################################################################
import argparse import argparse
import glob
import html import html
import random import random
import string import string
@ -991,8 +992,7 @@ def addfile_argparse(args):
book.move_nav_to_end() book.move_nav_to_end()
book.save(args.epub) book.save(args.epub)
def covercomesfirst_argparse(args): def covercomesfirst(book):
book = Epub.open(args.epub)
basenames = {i: book.get_filepath(i).basename for i in book.get_images()} basenames = {i: book.get_filepath(i).basename for i in book.get_images()}
if len(basenames) <= 1: if len(basenames) <= 1:
return return
@ -1032,6 +1032,13 @@ def covercomesfirst_argparse(args):
book.save(args.epub) book.save(args.epub)
def covercomesfirst_argparse(args):
epubs = [epub for pattern in args.epubs for epub in glob.glob(pattern)]
for epub in epubs:
print(epub)
book = Epub.open(epub)
covercomesfirst(book)
def holdit_argparse(args): def holdit_argparse(args):
book = Epub.open(args.epub) book = Epub.open(args.epub)
print(book.root_directory.absolute_path) print(book.root_directory.absolute_path)
@ -1104,9 +1111,13 @@ def merge_argparse(args):
return merge(input_filepaths=args.epubs, output_filename=args.output, do_headerfile=args.headerfile) return merge(input_filepaths=args.epubs, output_filename=args.output, do_headerfile=args.headerfile)
def normalize_argparse(args): def normalize_argparse(args):
book = Epub.open(args.epub) epubs = [epub for pattern in args.epubs for epub in glob.glob(pattern)]
book.normalize_directory_structure() for epub in epubs:
book.save(args.epub) print(epub)
book = Epub.open(epub)
book.normalize_directory_structure()
book.move_nav_to_end()
book.save(epub)
@betterhelp.subparser_betterhelp(main_docstring=DOCSTRING, sub_docstrings=SUB_DOCSTRINGS) @betterhelp.subparser_betterhelp(main_docstring=DOCSTRING, sub_docstrings=SUB_DOCSTRINGS)
def main(argv): def main(argv):
@ -1119,7 +1130,7 @@ def main(argv):
p_addfile.set_defaults(func=addfile_argparse) p_addfile.set_defaults(func=addfile_argparse)
p_covercomesfirst = subparsers.add_parser('covercomesfirst') p_covercomesfirst = subparsers.add_parser('covercomesfirst')
p_covercomesfirst.add_argument('epub') p_covercomesfirst.add_argument('epubs', nargs='+', default=[])
p_covercomesfirst.set_defaults(func=covercomesfirst_argparse) p_covercomesfirst.set_defaults(func=covercomesfirst_argparse)
p_holdit = subparsers.add_parser('holdit') p_holdit = subparsers.add_parser('holdit')
@ -1134,7 +1145,7 @@ def main(argv):
p_merge.set_defaults(func=merge_argparse) p_merge.set_defaults(func=merge_argparse)
p_normalize = subparsers.add_parser('normalize') p_normalize = subparsers.add_parser('normalize')
p_normalize.add_argument('epub') p_normalize.add_argument('epubs', nargs='+', default=[])
p_normalize.set_defaults(func=normalize_argparse) p_normalize.set_defaults(func=normalize_argparse)
args = parser.parse_args(argv) args = parser.parse_args(argv)