Cameron/MarxistEbook
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fixed single-url handling and changed argument structure

Browse Source
This commit is contained in:
SexbearLmao
2021-03-28 16:46:45 -05:00
parent c670d34aed
commit 84562fb8b6
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
marxistbook.py
View File

@@ -86,21 +86,23 @@ def download_book(url):
def main(): def main():
parser = ArgumentParser() parser = ArgumentParser()
parser.add_argument('-o', '--output', help='name of output file', dest='output') parser.add_argument('-o', '--output', help='name of output file', dest='output')
parser.add_argument('-i', '--input', help='input urls', dest='input', action='append') #parser.add_argument('-i', '--input', help='input urls', dest='input', action='append')
#parser.add_argument('-e', '--executable', help='directory of calibre executables', dest='exec') #parser.add_argument('-e', '--executable', help='directory of calibre executables', dest='exec')
parser.add_argument('-t', '--title', help='set the title manually', dest='title', default=None) parser.add_argument('-t', '--title', help='set the title manually', dest='title', default=None)
parser.add_argument('-a', '--author', help='set the author manually', dest='author', default=None) parser.add_argument('-a', '--author', help='set the author manually', dest='author', default=None)
parser.add_argument('url', help='urls to download', nargs='+')
args = parser.parse_args() args = parser.parse_args()
inp = args.input #inp = args.input
urls = args.url
outp = args.output or 'output.epub' outp = args.output or 'output.epub'
# exec_dir = args.exec #exec_dir = args.exec
title = args.title title = args.title
author = args.author author = args.author
output_extension = os.path.split(outp)[1] output_extension = os.path.split(outp)[1]
books = [download_book(i) for i in inp] books = [download_book(i) for i in urls]
if len(books) > 1: if len(books) > 1:
temp_name = rand_name() + '.epub' temp_name = rand_name() + '.epub'
merge_args = ['calibre-debug', '--run-plugin', 'EpubMerge', '--', merge_args = ['calibre-debug', '--run-plugin', 'EpubMerge', '--',
@@ -116,6 +118,9 @@ def main():
raise Exception('final merge returned {}'.format(sub.returncode)) raise Exception('final merge returned {}'.format(sub.returncode))
for book in books: for book in books:
os.remove(book) os.remove(book)
else:
#select the single item as the merge result
temp_name = books[0]
if output_extension == '.epub': if output_extension == '.epub':
os.rename(temp_name, outp) os.rename(temp_name, outp)