Backup_Repos/leech
1
0
Fork 0
mirror of https://github.com/kemayo/leech synced 2025-12-06 08:22:56 +01:00
Code Issues Projects Releases Wiki Activity

Add a null check to RR author note placement

Browse source
This commit is contained in:
Alex Raubach 2018-09-17 20:22:13 -04:00
parent cf62faf5dd
commit a9dfdb5dd3
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
sites/royalroad.py
View file

@ -55,8 +55,9 @@ class RoyalRoad(Site):
author_note = soup.find_all('div', class_='author-note-portlet')
if len(author_note) is 1:
# The first child div is either the chapter content or an author note
if 'author-note-portlet' in soup.find('div', class_='portlet-body').find('div', recursive=False)['class']:
# The first child div of portlet-body is either the chapter content or an author note
first_div = soup.find('div', class_='portlet-body').find('div', recursive=False)
if first_div and ('author-note-portlet' in first_div['class']):
content = author_note[0].prettify() + '<hr/>' + content
else: # The author note must be after the chapter content
content = content + '<hr/>' + author_note[0].prettify()