diff --git a/src/fb3_reader/__init__.py b/src/fb3_reader/__init__.py index 06d7405..485f44a 100644 Binary files a/src/fb3_reader/__init__.py and b/src/fb3_reader/__init__.py differ diff --git a/src/fb3_reader/fb3.py b/src/fb3_reader/fb3.py index 4c00326..ca3a937 100644 --- a/src/fb3_reader/fb3.py +++ b/src/fb3_reader/fb3.py @@ -1,12 +1,8 @@ -import sys from zipfile import ZipFile -# from bs4 import BeautifulSoup -# from lxml import etree -def reader(): - files = input("Enter the path to file: ") - with ZipFile(files) as book: - for row in book.read('2/fb3/body.xml').decode('utf-8').splitlines(): - opened = open("test.txt", "a") - opened.write(row) +def reader(path_to_file): + with ZipFile(path_to_file) as book: + for book_content in book.read('2/fb3/body.xml').decode('utf-8').splitlines(): + output_content = open("test.txt", "a") + output_content.write(book_content) diff --git a/src/fb3_reader/fb3.pyi b/src/fb3_reader/fb3.pyi new file mode 100644 index 0000000..c83594e --- /dev/null +++ b/src/fb3_reader/fb3.pyi @@ -0,0 +1 @@ +def reader(path_to_file: str): ... \ No newline at end of file diff --git a/src/main.py b/src/main.py index 223b06f..18c0f82 100644 --- a/src/main.py +++ b/src/main.py @@ -1,4 +1,9 @@ -from fb3_reader import fb3 +from sys import argv +from fb3_reader import fb3, __version__ if __name__ == '__main__': - fb3.reader() + try: + if argv[1] == "--version" or argv[1] == "-v": + print(__version__) + except IndexError: + fb3.reader(input("Enter path to file: "))