Add minimal setup.py and test0.2.0

5 files changed, 64 insertions, 17 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..61807b4
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,9 @@
+*.egg-info
+*.pyc
+*.pyo
+build/
+dist/
+__pycache__/
+MANIFEST
+.tox/
+.coverage
diff --git a/epubgrep.py b/epubgrep.py
index e3051bf..578c20b 100755
--- a/epubgrep.py
+++ b/epubgrep.py
@@ -8,19 +8,6 @@ from typing import Any, Dict, List, Optional, Tuple
 
 import epub_meta
 
-parser = argparse.ArgumentParser(description='Grep through EPub book')
-parser.add_argument('pattern')
-parser.add_argument('filename')
-parser.add_argument('-i', '--ignore-case',
-                    action='store_true',
-                    help="make search case insensitive")
-args = parser.parse_args()
-
-search_flags = 0
-if args.ignore_case:
-    search_flags |= re.I
-
-
 def get_chapter_title(mdata:List[Dict[str, Any]], fname:str) -> Optional[Tuple[str, int]]:
     found_list = [(x['title'], x['index']) for x in mdata if x['src'] == fname]
     if len(found_list) > 0:
@@ -30,7 +17,8 @@ def get_chapter_title(mdata:List[Dict[str, Any]], fname:str) -> Optional[Tuple[s
         return ('Unknown', 0)
 
 
-def grep_book(filename:str, pattern:str, flags):
+def grep_book(filename:str, pattern:str, flags:int):
+    assert os.path.isfile(filename), "{} is not EPub file.".format(filename)
     sought_RE = re.compile(pattern, flags)
 
     metadata = epub_meta.get_epub_metadata(filename)
@@ -49,6 +37,18 @@ def grep_book(filename:str, pattern:str, flags):
                         printed_title = True
                     print(decoded_line)
 
-assert os.path.isfile(args.filename), "{} is not EPub file.".format(args.filename)
-book_fname = os.path.realpath(args.filename)
-grep_book(book_fname, args.pattern, search_flags)
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='Grep through EPub book')
+    parser.add_argument('pattern')
+    parser.add_argument('filename')
+    parser.add_argument('-i', '--ignore-case',
+                        action='store_true',
+                        help="make search case insensitive")
+    args = parser.parse_args()
+
+    search_flags = 0
+    if args.ignore_case:
+        search_flags |= re.I
+
+    book_fname = os.path.realpath(args.filename)
+    grep_book(book_fname, args.pattern, search_flags)
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..6e5a25a
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+
+from setuptools import setup, find_packages
+
+setup(
+    name="epubgrep",
+    version="0.2.0",
+    description='Grep through EPub files',
+    author=u'Matěj Cepl',
+    author_email='mcepl@cepl.eu',
+    url='https://gitlab.com/mcepl/epubgrep',
+    packages=find_packages(),
+    test_suite='tests',
+    install_requires=['epub_meta'],
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.3",
+        "Programming Language :: Python :: 3.4",
+        "Programming Language :: Python :: 3.5",
+        "Programming Language :: Python :: 3.6",
+        "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.7",
+        "Environment :: Console",
+        "Intended Audience :: Information Technology",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+        "Topic :: Software Development :: Libraries :: Python Modules",
+    ],
+)
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/__init__.py
diff --git a/tests/test_epugrep.py b/tests/test_epugrep.py
new file mode 100644
index 0000000..7db430a
--- /dev/null
+++ b/tests/test_epugrep.py
@@ -0,0 +1,9 @@
+import unittest
+
+import epubgrep
+
+
+class TestEPubGrep(unittest.TestCase):
+    def test_epub_is_file(self):
+        with self.assertRaises(AssertionError):
+            epubgrep.grep_book('foo', 'bar', 0)