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