aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-xepubgrep.py31
-rw-r--r--setup.py2
2 files changed, 31 insertions, 2 deletions
diff --git a/epubgrep.py b/epubgrep.py
index 8afbf54..18ab6e4 100755
--- a/epubgrep.py
+++ b/epubgrep.py
@@ -115,10 +115,33 @@ def _metadata_search(mdata: dict, sre: re.Pattern, fname: str,
out += _colorize_found(tag, res, col)
return out
+def _author_search(mdata: dict, sre: re.Pattern, fname: str,
+ col: bool) -> str:
+ """
+ Search through metadata, not text.
+
+ :param: mdata: complete metadata to search through
+ :param: sre: re.Pattern to search
+ :param: fname: filename of the book
+ :param: col: should we colorize the output
+ """
+ out = ''
+ title = ''
+ authors = mdata.get('authors')
+
+ for auth in authors:
+ res = sre.search(auth)
+ if res:
+ if not title:
+ title = f'{fname}:'
+ out += ' ' + _colorize_found(auth, res, col)
+
+ return title + out
+
def grep_book(filename: str, opts: argparse.Namespace,
re_flags: int) -> Optional[str]:
- assert os.path.isfile(filename), f'{filename} is not EPub file.'
+ assert os.path.isfile(filename), f'{filename} is not a file.'
sought_RE = re.compile('(' + opts.pattern + ')', re_flags)
count = 0
icount = 0
@@ -136,6 +159,9 @@ def grep_book(filename: str, opts: argparse.Namespace,
book = zipfile.ZipFile(filename)
printed_booktitle = False
+ if opts.author:
+ return _author_search(metadata, sought_RE, filename, opts.color)
+
if opts.metadata:
return _metadata_search(metadata, sought_RE, filename, opts.color)
@@ -177,6 +203,9 @@ def main():
parser = argparse.ArgumentParser(description='Grep through EPub book')
parser.add_argument('pattern')
parser.add_argument('files', nargs='+')
+ parser.add_argument('-a', '--author',
+ action='store_true',
+ help="prints titles with given author")
parser.add_argument('-c', '--count',
action='store_true',
help="just counts of found patterns")
diff --git a/setup.py b/setup.py
index 659c517..244815c 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
setup(
name="epubgrep",
- version="0.7.1",
+ version="0.8.0",
description='Grep through EPub files',
author=u'Matěj Cepl',
author_email='mcepl@cepl.eu',