renatoxsr/m4b
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Atualizar m4b/cli.py

Browse Source
This commit is contained in:
renatoxsr
2026-08-27 15:20:38 -03:00
parent 95f9850186
commit da9646745c
1 changed files with 67 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
m4b/cli.py
+65 -41
View File
@@ -41,20 +41,38 @@ class Cli:
def __post_init__(self): def __post_init__(self):
self.create_logger() self.create_logger()
self.check_tools()
self.get_parser()
self.parse_args()
def check_tools(self, tool):
for tool_name in [f.name for f in dataclasses.fields(self) if isinstance(f.type, Config.Tool)]: for tool_name in [f.name for f in dataclasses.fields(self) if isinstance(f.type, Config.Tool)]:
tool_path = self.gettattr(tool_name) tool_path = self.gettattr(tool_name)
if not shutil.which(tool_path): if not shutil.which(tool_path):
self.error("%s (%s) not found in $PATH", tool_name, tool_path) self.error("%s (%s) not found in $PATH", tool_name, tool_path)
self.exit(1) self.exit(1)
p = argparse.ArgumentParser() def get_parser(self):
p.add_argument("input", type=str, help="input directory") self.parser = argparse.ArgumentParser()
p.add_argument("-o","--output", type=str, help="output directory") self.parser.add_argument("input", type=str, help="input directory or file")
p.add_argument("--debug", dest="loglevel", action="store_const",const="DEBUG", help="Set log level to DEBUG") self.parser.add_argument("-g","--glob", type=str, nargs=1, help="Glob pattern in input folder. See pathlib documentation.")
p.add_argument("--extract", action="store_true", help="Extract file if compressed archive") self.parser.add_argument("-m", "--mode", type=str, choices=['single', 'multi'], help="Input format (single book from epub file or an entire directory tree, or multiple books from each subfolder or epub file")
p.add_argument("--chapters", action="store_true", help="Parse chapters from epub file") self.parser.add_argument("-o","--output", type=str, help="output directory")
self.args = vars(p.parse_args()) self.parser.add_argument("--debug", dest="loglevel", action="store_const",const="DEBUG", help="Set log level to DEBUG")
self.parser.add_argument("--extract", action="store_true", help="Extract file if compressed archive")
self.parser.add_argument("--extract-only-audio", action="store_true", help="Only extract audio from compressed file (default: False)")
self.parser.add_argument("--chapters", action="store_true", help="Parse chapters from epub file")
return self.parser
def parse_args(self, argv: list = None):
if not argv:
argv = sys.argv
self.parsed_args = self.parser.parse_args(argv)
self.args = vars(self.parsed_args)
#for a in self.args: #for a in self.args:
# setattr(self, a, self.args[a]) # setattr(self, a, self.args[a])
@@ -89,9 +107,11 @@ class Cli:
if output_path.suffix == ".m4b": if output_path.suffix == ".m4b":
output_name = output_path.name output_name = output_path.name
output_path = output_path.parent output_path = output_path.parent
self.merge_output = output_path.resolve() / output_name self.merge_output = output_path.resolve() / output_name
return self.args
def extract_input(self, input_path): def extract_input(self, input_path):
self.debug("Input is compressed archive: %s", input_path) self.debug("Input is compressed archive: %s", input_path)
tmp_dir = pathlib.Path(self.tmp_dir).resolve() if self.tmp_dir else input_path.with_name(input_path.name + self.tmp_suffix).resolve() tmp_dir = pathlib.Path(self.tmp_dir).resolve() if self.tmp_dir else input_path.with_name(input_path.name + self.tmp_suffix).resolve()
@@ -101,40 +121,44 @@ class Cli:
self.exit(1) self.exit(1)
self.generate_epub_audio_suffixes() self.generate_epub_audio_suffixes()
with zipfile.ZipFile(input_path) as compressed: with zipfile.ZipFile(input_path) as compressed:
compressed.extractall(tmp_dir) if not self.args["extract-only-audio"]:
compressed.extractall(tmp_dir)
return tmp_dir
# audio_files = [] audio_files = []
# #for file in compressed.namelist(): #for file in compressed.namelist():
# for file_info in compressed.infolist(): for file_info in compressed.infolist():
# filename = pathlib.Path(file_info.filename) filename = pathlib.Path(file_info.filename)
# if not filename.suffix in self.epub_audio_suffixes: if not filename.suffix in self.epub_audio_suffixes:
# self.debug("Ignoring file with unsupported suffix: %s", filename) self.debug("Ignoring file with unsupported suffix: %s", filename)
# continue continue
# with compressed.open(file_info, 'r') as src: with compressed.open(file_info, 'r') as src:
# audio_file = mutagen.File(src) audio_file = mutagen.File(src)
# #guessed_type, _ = mimetypes.guess_file_type(src) #guessed_type, _ = mimetypes.guess_file_type(src)
# guessed_type = audio_file.mime[0] guessed_type = audio_file.mime[0]
# if guessed_type in self.epub_audio_types: if guessed_type in self.epub_audio_types:
# target = tmp_dir / filename.name target = tmp_dir / filename.name
# self.debug("Extracting %s (%s) to %s", filename.name, guessed_type, target) self.debug("Extracting %s (%s) to %s", filename.name, guessed_type, target)
# with open(target, 'wb') as dst: with open(target, 'wb') as dst:
# shutil.copyfileobj(src, dst) shutil.copyfileobj(src, dst)
# #dst.write(src) #dst.write(src)
# audio_files.append(target) audio_files.append(target)
# else: else:
# self.error("Guessed mimetype is not supported: %s", guessed_type) self.error("Guessed mimetype is not supported: %s", guessed_type)
# if len(audio_files)==0: if len(audio_files)==0:
# self.error("No supported audio files in compressed archive: %s", input_path) self.error("No supported audio files in compressed archive: %s", input_path)
# self.exit(1) self.exit(1)
# else: else:
# for file in audio_files: for file in audio_files:
# if not file.exists(): if not file.exists():
# self.error("Error extracting (file does not exist): %s", file) self.error("Error extracting (file does not exist): %s", file)
# if not mutagen.File(file).mime[0] in self.epub_audio_types: if not mutagen.File(file).mime[0] in self.epub_audio_types:
# self.error("Error extracting (wrong mimetype): %s", file) self.error("Error extracting (wrong mimetype): %s", file)
# if not file.parent == tmp_dir: if not file.parent == tmp_dir:
# self.error("Error extracting (wrong target path): %s", file) self.error("Error extracting (wrong target path): %s", file)
# self.info("Verified %s", file) self.info("Verified %s", file)
# Close zipfile
return tmp_dir return tmp_dir