eneller/epub2go.py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: display authors

Browse Source
This commit is contained in:
eneller
2025-02-25 20:09:31 +01:00
parent 90bdf83950
commit f9942a75d3
1 changed files with 28 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
src/epub2go/convert.py
View File

@@ -87,15 +87,33 @@ class GBConvert():
self.create_epub(f'{self.title} - {self.author}.epub') self.create_epub(f'{self.title} - {self.author}.epub')
def get_all_books() -> list: def get_all_books() -> list:
books = get_all_book_tags() response = requests.get(allbooks_url)
d = [] response.raise_for_status()
for book in books: soup = BeautifulSoup(response.content, 'html.parser', from_encoding='utf-8')
book_href = book.get('href') tags = soup.find('dl').findChildren()
if book_href is not None: books = []
for tag in tags:
# is description tag, i.e. contains author name
if tag.name =='dt':
# update author
# special case when author name and Alphabetical list is in same tag
br_tag = tag.find('br')
if br_tag:
book_author = str(br_tag.next_sibling)
# default case, dt only contains author name
else:
book_author = tag.get_text(strip=True)
book_author = ' '.join(book_author.split())
# is details tag, contains book url
elif tag.name == 'dd':
book_tag = tag.a
if book_tag:
book_href = book_tag.get('href')
book_url = urljoin(allbooks_url, book_href) book_url = urljoin(allbooks_url, book_href)
book_title = book.getText().translate(str.maketrans('','', '\n\t')) book_title = ' '.join(book_tag.getText().split())
d.append({'title': book_title, 'url': book_url}) book = {'author': book_author, 'title': book_title, 'url': book_url}
return d books.append(book)
return books
def get_all_book_tags ()-> ResultSet: def get_all_book_tags ()-> ResultSet:
response = requests.get(allbooks_url) response = requests.get(allbooks_url)
@@ -113,8 +131,7 @@ def main():
else: else:
delimiter = ';' delimiter = ';'
# create lines for fzf # create lines for fzf
# TODO display author books = [f"{item['author']} - {item['title']} {delimiter} {item['url']}" for item in get_all_books()]
books = [f"{item['title']} {delimiter} {item['url']}" for item in get_all_books()]
fzf = FzfPrompt() fzf = FzfPrompt()
selection = fzf.prompt(choices=books, fzf_options=r'--exact --with-nth 1 -m -d\;') selection = fzf.prompt(choices=books, fzf_options=r'--exact --with-nth 1 -m -d\;')
books = [item.split(';')[1].strip() for item in selection] books = [item.split(';')[1].strip() for item in selection]