Radiquum/furaffinity-dl
1
0
Fork 0
mirror of https://github.com/Radiquum/furaffinity-dl.git synced 2025-04-04 23:34:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix System Message extracting and add to file downloader

Browse source
This commit is contained in:
Xerbo 2020-09-23 14:28:16 +01:00
parent 73a6092217
commit 9c8548076f
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
furaffinity-dl.py
View file

@ -91,6 +91,11 @@ def download(path):
response = session.get(page_url)
s = BeautifulSoup(response.text, 'html.parser')
# System messages
if s.find(class_='notice-message') is not None:
message = s.find(class_='notice-message').find('div').find(class_="link-override").text.strip()
raise Exception('System Message', message)
image = s.find(class_='download').find('a').attrs.get('href')
title = s.find(class_='submission-title').find('p').contents[0]
filename = image.split("/")[-1:][0]
@ -141,6 +146,7 @@ def download(path):
return False
else:
print('Skipping "{}", since it\'s already downloaded'.format(data["title"]))
return True
# Write a UTF-8 encoded JSON file for metadata
with open(os.path.join(args.output, '{}.json'.format(filename)), 'w', encoding='utf-8') as f:
@ -165,12 +171,8 @@ while True:
# System messages
if s.find(class_='notice-message') is not None:
message = s.find(class_='notice-message').find('div')
for ele in message:
if ele.name is not None:
ele.decompose()
raise Exception('System Message', message.text.strip())
message = s.find(class_='notice-message').find('div').find(class_="link-override").text.strip()
raise Exception('System Message', message)
# End of gallery
if s.find(id='no-images') is not None: