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

commits squash

Browse source 
added:
- filter
- remove not fully downloaded file when exiting with CTRL + C
- ability to download description as a list
- ability to download a specific file

changed
- color output
- remove custom output directory
- making of directory tree based off username, category and rating
- default attributes value changes
- system messages parsing changes
- arguments & help text change - move metadata var to section where metadata == True - add page URL to metadata file
- some minor changes
This commit is contained in:
Radiquum 2022-06-16 07:06:42 +05:00 • committed by Radiquum
parent 1dfa8f6453
commit 04a09f07ad
No known key found for this signature in database
GPG key ID: CB1FC16C710DB347
3 changed files with 245 additions and 179 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
.gitignore vendored
View file

@ -8,6 +8,9 @@ cookies.txt
*.gif *.gif
*.swf *.swf
# code monitoring stuff # Download folder
furaffinity-dl/
# vscode stuff
.vscode .vscode
ignore ignore

54
README.md
View file

@ -24,47 +24,51 @@ When downloading a folder make sure to put everything after **/folder/**, for ex
```help ```help
usage: furaffinity-dl.py [-h] [--category [CATEGORY]] [--submissions] [--folder FOLDER] [--output OUTPUT] [--cookies COOKIES] usage: furaffinity-dl.py [-h] [--submissions] [--folder FOLDER [FOLDER ...]] [--cookies COOKIES [COOKIES ...]]
[--user-agent [UA]] [--start [START]] [--stop STOP] [--dont-redownload] [--interval INTERVAL] [--rating] [--user-agent USER_AGENT [USER_AGENT ...]] [--start START [START ...]] [--stop STOP [STOP ...]] [--dont-redownload]
[--metadata] [--interval INTERVAL [INTERVAL ...]] [--rating] [--filter] [--metadata] [--download DOWNLOAD] [--json-description]
username [username] [category]
Downloads the entire gallery/scraps/favorites of a furaffinity user, or your submissions Downloads the entire gallery/scraps/folder/favorites of a furaffinity user, or your submissions notifications
positional arguments: positional arguments:
username username of the furaffinity user [required] username username of the furaffinity user
category the category to download, gallery/scraps/favorites [default: gallery]
options: options:
-h, --help show this help message and exit -h, --help show this help message and exit
--category [CATEGORY], -ca [CATEGORY] --submissions download your submissions
the category to download, gallery/scraps/favorites [default: gallery] --folder FOLDER [FOLDER ...]
--submissions, -su download your submissions full path of the furaffinity gallery folder. for instance 123456/Folder-Name-Here
--folder FOLDER full path of the furaffinity folder. for instance 123456/Folder-Name-Here --cookies COOKIES [COOKIES ...], -c COOKIES [COOKIES ...]
--output OUTPUT output directory [default: furaffinity-dl]
--cookies COOKIES, -c COOKIES
path to a NetScape cookies file path to a NetScape cookies file
--user-agent [UA], -u [UA] --user-agent USER_AGENT [USER_AGENT ...]
Your browser's useragent, may be required, depending on your luck Your browser's useragent, may be required, depending on your luck
--start [START], -s [START] --start START [START ...], -s START [START ...]
page number to start from page number to start from
--stop STOP, -S STOP Page number to stop on. For favorites pages, specify the full URL after the username (1234567890/next). --stop STOP [STOP ...], -S STOP [STOP ...]
Page number to stop on. Specify the full URL after the username: for favorites pages (1234567890/next) or for submissions pages: (new~123456789@48)
--dont-redownload, -d --dont-redownload, -d
Don't redownload files that have already been downloaded [default: true] Allow to redownload files that have been downloaded already
--interval INTERVAL, -i INTERVAL --interval INTERVAL [INTERVAL ...], -i INTERVAL [INTERVAL ...]
delay between downloading pages in seconds [default: 0] delay between downloading pages in seconds [default: 0]
--rating, -r enable rating separation [default: true] --rating, -r disable rating separation
--metadata, -m enable downloading of metadata [default: false] --filter disable submission filter
--metadata, -m enable downloading of metadata
--download DOWNLOAD download a specific submission /view/12345678/
--json-description download description as a JSON list
Examples: Examples:
python3 furaffinity-dl.py koul koul_gallery python3 furaffinity-dl.py koul -> will download gallery of user koul
python3 furaffinity-dl.py -o koulsArt gallery koul python3 furaffinity-dl.py koul scraps -> will download scraps of user koul
python3 furaffinity-dl.py -o mylasFavs --category favorites mylafox python3 furaffinity-dl.py mylafox favorites -> will download favorites of user mylafox
You can also log in to FurAffinity in a web browser and load cookies to download Age restricted content or Submissions: You can also log in to FurAffinity in a web browser and load cookies to download age restricted content or submissions:
python3 furaffinity-dl.py -c cookies.txt gallery letodoesart python3 furaffinity-dl.py letodoesart -c cookies.txt -> will download gallery of user letodoesart including Mature and Adult submissions
python3 furaffinity-dl.py -c cookies.txt --submissions python3 furaffinity-dl.py --submissions -c cookies.txt -> will download your submissions notifications
DISCLAIMER: It is your own responsibility to check whether batch downloading is allowed by FurAffinity terms of service and to abide by them. DISCLAIMER: It is your own responsibility to check whether batch downloading is allowed by FurAffinity terms of service and to abide by them.
``` ```
You can also log in to download restricted content. To do that, log in to FurAffinity in your web browser, export cookies to a file from your web browser in Netscape format (there are extensions to do that [for Firefox](https://addons.mozilla.org/en-US/firefox/addon/ganbo/) and [for Chrome based browsers](https://chrome.google.com/webstore/detail/get-cookiestxt/bgaddhkoddajcdgocldbbfleckgcbcid?hl=en)), you can then pass them to the script with the `-c` flag, like this (you may also have to provide your user agent): You can also log in to download restricted content. To do that, log in to FurAffinity in your web browser, export cookies to a file from your web browser in Netscape format (there are extensions to do that [for Firefox](https://addons.mozilla.org/en-US/firefox/addon/ganbo/) and [for Chrome based browsers](https://chrome.google.com/webstore/detail/get-cookiestxt/bgaddhkoddajcdgocldbbfleckgcbcid?hl=en)), you can then pass them to the script with the `-c` flag, like this (you may also have to provide your user agent):

363
furaffinity-dl.py
View file

@ -10,195 +10,221 @@ import requests
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from tqdm import tqdm from tqdm import tqdm
# COLORS
WHITE = "\033[1;37m"
RED = "\033[1;91m"
GREEN = "\033[1;92m"
YELLOW = "\033[1;33m"
END = "\033[0m"
# Argument parsing # Argument parsing
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
formatter_class=argparse.RawTextHelpFormatter, formatter_class=argparse.RawTextHelpFormatter,
description="Downloads the entire gallery/scraps/favorites of a furaffinity user, or your submissions", description="Downloads the entire gallery/scraps/folder/favorites of a furaffinity user, or your submissions notifications",
epilog=""" epilog="""
Examples: Examples:
python3 furaffinity-dl.py koul koul_gallery python3 furaffinity-dl.py koul -> will download gallery of user koul
python3 furaffinity-dl.py -o koulsArt gallery koul python3 furaffinity-dl.py koul scraps -> will download scraps of user koul
python3 furaffinity-dl.py -o mylasFavs --category favorites mylafox\n python3 furaffinity-dl.py mylafox favorites -> will download favorites of user mylafox \n
You can also log in to FurAffinity in a web browser and load cookies to download Age restricted content or Submissions: You can also log in to FurAffinity in a web browser and load cookies to download age restricted content or submissions:
python3 furaffinity-dl.py -c cookies.txt gallery letodoesart python3 furaffinity-dl.py letodoesart -c cookies.txt -> will download gallery of user letodoesart including Mature and Adult submissions
python3 furaffinity-dl.py -c cookies.txt --submissions\n python3 furaffinity-dl.py --submissions -c cookies.txt -> will download your submissions notifications \n
DISCLAIMER: It is your own responsibility to check whether batch downloading is allowed by FurAffinity terms of service and to abide by them. DISCLAIMER: It is your own responsibility to check whether batch downloading is allowed by FurAffinity terms of service and to abide by them.
""", """,
) )
parser.add_argument("username", nargs="?", help="username of the furaffinity user")
# General stuff
parser.add_argument( parser.add_argument(
"--category", "category",
"-ca",
type=str,
nargs="?", nargs="?",
help="the category to download, gallery/scraps/favorites [default: gallery]", help="the category to download, gallery/scraps/favorites [default: gallery]",
const=1,
default="gallery", default="gallery",
) )
parser.add_argument( parser.add_argument(
"--submissions", "--submissions", action="store_true", help="download your submissions"
"-su",
action="store_true",
help="download your submissions",
)
parser.add_argument(
"username",
type=str,
help="username of the furaffinity user [required]",
) )
parser.add_argument( parser.add_argument(
"--folder", "--folder",
type=str, nargs="+",
help="full path of the furaffinity folder. for instance 123456/Folder-Name-Here", help="full path of the furaffinity gallery folder. for instance 123456/Folder-Name-Here",
) )
parser.add_argument( parser.add_argument(
"--output", type=str, default="furaffinity-dl", help="output directory [default: furaffinity-dl]" "--cookies", "-c", nargs="+", help="path to a NetScape cookies file"
)
parser.add_argument(
"--cookies",
"-c",
dest="cookies",
type=str,
help="path to a NetScape cookies file",
) )
parser.add_argument( parser.add_argument(
"--user-agent", "--user-agent",
"-u", dest="user_agent",
dest="ua", nargs="+",
type=str, default="Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:101.0) Gecko/20100101 Firefox/101.0",
nargs="?",
default="Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:68.7) Gecko/20100101 Firefox/68.7",
help="Your browser's useragent, may be required, depending on your luck", help="Your browser's useragent, may be required, depending on your luck",
) )
parser.add_argument( parser.add_argument(
"--start", "-s", type=str, default=1, help="page number to start from", nargs="?" "--start", "-s", default=[1], help="page number to start from", nargs="+"
) )
parser.add_argument( parser.add_argument(
"--stop", "--stop",
"-S", "-S",
dest="stop", default=[0],
type=str, nargs="+",
help="Page number to stop on. For favorites pages, specify the full URL after the username (1234567890/next).", help="Page number to stop on. Specify the full URL after the username: for favorites pages (1234567890/next) or for submissions pages: (new~123456789@48)",
) )
parser.add_argument( parser.add_argument(
"--dont-redownload", "--dont-redownload",
"-d", "-d",
action="store_false", action="store_false",
help="Don't redownload files that have already been downloaded [default: true]", help="Allow to redownload files that have been downloaded already",
) )
parser.add_argument( parser.add_argument(
"--interval", "--interval",
"-i", "-i",
dest="interval", type=int,
type=float, default=[0],
default=0, nargs="+",
help="delay between downloading pages in seconds [default: 0]", help="delay between downloading pages in seconds [default: 0]",
) )
parser.add_argument( parser.add_argument(
"--rating", "--rating",
"-r", "-r",
action="store_false", action="store_false",
help="enable rating separation [default: true]", help="disable rating separation",
)
parser.add_argument(
"--filter",
action="store_false",
help="disable submission filter",
) )
parser.add_argument( parser.add_argument(
"--metadata", "--metadata",
"-m", "-m",
action="store_true", action="store_true",
help="enable downloading of metadata [default: false]", help="enable downloading of metadata",
)
parser.add_argument(
"--download",
help="download a specific submission /view/12345678/",
)
parser.add_argument(
"--json-description",
dest="json_description",
action="store_true",
help="download description as a JSON list",
) )
args = parser.parse_args() args = parser.parse_args()
BASE_URL = "https://www.furaffinity.net" BASE_URL = "https://www.furaffinity.net"
categories = {
"gallery": "gallery",
"scraps": "scraps",
"favorites": "favorites",
}
category = categories.get(args.category)
if category is None:
print("please enter a valid category")
exit()
if args.username is None:
print("please enter a FA Username")
exit()
if args.output is None:
print("please enter a output folder")
exit()
username = args.username username = args.username
output = f"{args.output}/{args.username}"
metadata = f"{output}/metadata"
filter = {"YCH Open", "Reminder", "YCH Closed", "Auction"} if args.submissions is False and args.download is None: # check if you are not downloading submissions or a specific post
categories = {
"gallery": "gallery",
"scraps": "scraps",
"favorites": "favorites",
}
category = categories.get(args.category)
if args.username is None:
print(f"{RED}<!> please enter a FA Username{END}")
exit()
if category is None:
print(f"{RED}<!> please enter a valid category gallery/scraps/favorites{END}")
exit()
download_url = f"{BASE_URL}/{category}/{username}"
output = f"furaffinity-dl/{category}/{username}"
# get session
session = requests.session() session = requests.session()
session.headers.update({"User-Agent": args.ua}) session.headers.update({"User-Agent": args.user_agent[0]})
if args.cookies is not None: if args.cookies is not None: # add cookies if present
cookies = cookielib.MozillaCookieJar(args.cookies) cookies = cookielib.MozillaCookieJar(args.cookies[0])
cookies.load() cookies.load()
session.cookies = cookies session.cookies = cookies
# File downloading
def download_file(url, fname, desc): def download_file(url, fname, desc):
r = session.get(url, stream=True) try:
if r.status_code != 200: r = session.get(url, stream=True)
print(f"Got a HTTP {r.status_code} while downloading {fname}; ...skipping") if r.status_code != 200:
return False print(
f"{RED}<!> Got a HTTP {r.status_code} while downloading {fname}; ...skipping{END}"
)
return False
total = int(r.headers.get("Content-Length", 0))
with open(fname, "wb") as file, tqdm(
desc=desc.ljust(60)[:60],
total=total,
miniters=100,
unit="b",
unit_scale=True,
unit_divisor=1024,
) as bar:
for data in r.iter_content(chunk_size=1024):
size = file.write(data)
bar.update(size)
except KeyboardInterrupt:
print(f"{GREEN}<i> Finished downloading{END}")
os.remove(fname)
exit()
total = int(r.headers.get("Content-Length", 0))
with open(fname, "wb") as file, tqdm(
desc=desc.ljust(40)[:40],
total=total,
miniters=100,
unit="b",
unit_scale=True,
unit_divisor=1024,
) as bar:
for data in r.iter_content(chunk_size=1024):
size = file.write(data)
bar.update(size)
return True return True
download_url = f"{BASE_URL}/{category}/{username}" # checks if you are downloading forder or submission
if args.folder is not None: if args.folder is not None:
download_url = f"{BASE_URL}/gallery/{username}/folder/{args.folder}" folder = args.folder[0].split("/")
if args.submissions is True: download_url = f"{BASE_URL}/gallery/{username}/folder/{args.folder[0]}"
output = f"furaffinity-dl/folders/{username}/{folder[1]}"
if args.submissions is True:
download_url = f"{BASE_URL}/msg/submissions" download_url = f"{BASE_URL}/msg/submissions"
def download(path): def download(path):
response = session.get(f"{BASE_URL}{path}") response = session.get(f"{BASE_URL}{path}")
s = BeautifulSoup(response.text, "html.parser") s = BeautifulSoup(response.text, "html.parser")
# System messages # System messages
if s.find(class_="notice-message") is not None: if s.find(class_="notice-message") is not None:
message = ( try:
s.find(class_="notice-message") message = (
.find("div") s.find(class_="notice-message")
.find(class_="link-override") .find("div")
.text.strip() .find(class_="link-override")
) .text.strip()
raise Exception("System Message", message) .replace(".", ". \n")
)
except AttributeError:
message = (
s.find(class_="notice-message")
.find("div", class_="section-body alignleft")
.find("div", class_="redirect-message")
.text.strip()
.replace(".", ". \n")
)
print(f"{YELLOW}<i> System Message: {message}{END}")
exit()
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] + " "
description = s.find(class_="submission-description").text.strip().replace("\r\n", "\n")
author = s.find(class_="submission-id-sub-container")
if args.submissions is True:
global output
output = f"furaffinity-dl/gallery/{author}"
if args.json_description is True:
description = []
filename = image.split("/")[-1:][0] filename = image.split("/")[-1:][0]
data = { data = {
"id": int(path.split("/")[-2:-1][0]), "id": int(path.split("/")[-2:-1][0]),
"filename": filename, "filename": filename,
"author": s.find(class_="submission-id-sub-container") "author": author
.find("a") .find("a")
.find("strong") .find("strong")
.text, .text,
"date": s.find(class_="popup_date").attrs.get("title"), "date": s.find(class_="popup_date").attrs.get("title"),
"title": title, "title": title,
"description": s.find(class_="submission-description") "description": description,
.text.strip() "url": f"{BASE_URL}{path}",
.replace("\r\n", "\n"),
"tags": [], "tags": [],
"category": s.find(class_="info").find(class_="category-name").text, "category": s.find(class_="info").find(class_="category-name").text,
"type": s.find(class_="info").find(class_="type-name").text, "type": s.find(class_="info").find(class_="type-name").text,
@ -210,39 +236,49 @@ def download(path):
"comments": [], "comments": [],
} }
match = re.search( if args.filter is True:
"reminder|auction|YCH Open|YCH Closed|Adopt|pre-order", title, re.IGNORECASE match = re.search(
) "YCH[a-z $-/:-?{-~!\"^_`\[\]]*OPEN|OPEN[a-z $-/:-?{-~!\"^_`\[\]]*YCH|YCH[a-z $-/:-?{-~!\"^_`\[\]]*CLOSE|CLOSE[a-z $-/:-?{-~!\"^_`\[\]]*YCH|YCH[a-z $-/:-?{-~!\"^_`\[\]]*ABLE|AVAIL[a-z $-/:-?{-~!\"^_`\[\]]*YCH|YCH[a-z $-/:-?{-~!\"^_`\[\]]*CLONE|CLONE[a-z $-/:-?{-~!\"^_`\[\]]*YCH|YCH[a-z $-/:-?{-~!\"^_`\[\]]*LIM|LIM[a-z $-/:-?{-~!\"^_`\[\]]*YCH|COM[a-z $-/:-?{-~!\"^_`\[\]]*OPEN|OPEN[a-z $-/:-?{-~!\"^_`\[\]]*COM|COM[a-z $-/:-?{-~!\"^_`\[\]]*CLOSE|CLOSE[a-z $-/:-?{-~!\"^_`\[\]]*COM|FIX[a-z $-/:-?{-~!\"^_`\[\]]*ICE|REM[insder]*\W|\\bREF|Sale$|auction|multislot|stream|adopt",
if match is not None and title == match.string: title,
print(f"post {title} was filtered and will not be downloaded") re.IGNORECASE,
return True )
if match is not None and title == match.string:
# Extact tags print(
try: f"{YELLOW}<i> post {title} was filtered and will not be downloaded - {data.get('url')}{END}"
for tag in s.find(class_="tags-row").findAll(class_="tags"): )
data["tags"].append(tag.find("a").text) return True
except AttributeError:
print(f'post: "{title}", has no tags')
image_url = f"https:{image}" image_url = f"https:{image}"
os.makedirs(output, exist_ok=True) os.makedirs(output, exist_ok=True)
output_path = f"{output}/{filename}" output_path = f"{output}/{filename}"
if args.rating is True: if args.rating is True:
os.makedirs(f'{output}/{data.get("rating")}', exist_ok=True) os.makedirs(f'{output}/{data.get("rating")}', exist_ok=True)
output_path = f'{output}/{data.get("rating")}/{filename}' output_path = f'{output}/{data.get("rating")}/{filename}'
if args.folder is not None:
os.makedirs(f"{output}/{args.folder}", exist_ok=True)
output_path = f"{output}/{args.folder}/{filename}"
if args.rating is True:
os.makedirs(f'{output}/{args.folder}/{data.get("rating")}', exist_ok=True)
output_path = f'{output}/{args.folder}/{data.get("rating")}/{filename}'
if args.dont_redownload is True and os.path.isfile(output_path): if args.dont_redownload is True and os.path.isfile(output_path):
print(f'Skipping "{title}", since it\'s already downloaded') print(f'{YELLOW}<i> Skipping "{title}", since it\'s already downloaded{END}')
else: else:
download_file(image_url, output_path, title) download_file(image_url, output_path, title)
if args.metadata is True: if args.metadata is True:
metadata = f"{output}/metadata"
# Extract description as list
if args.json_description is True:
for desc in s.find("div", class_="submission-description").strings:
description = desc.strip()
data["description"].append(description)
# Extact tags
try:
for tag in s.find(class_="tags-row").findAll(class_="tags"):
data["tags"].append(tag.find("a").text)
except AttributeError:
print(f'{YELLOW}<i> post: "{title}", has no tags{END}')
# Extract comments # Extract comments
for comment in s.findAll(class_="comment_container"): for comment in s.findAll(class_="comment_container"):
temp_ele = comment.find(class_="comment-parent") temp_ele = comment.find(class_="comment-parent")
@ -274,23 +310,30 @@ def download(path):
return True return True
if args.download is not None:
# Main downloading loop output = "furaffinity-dl/downloaded/"
download(args.download)
print(f"{GREEN}<i> File downloaded{END}")
exit()
# Main function
def main(): def main():
page_end = args.stop # check if you are logged in
page_num = args.start page_end = args.stop[0]
page_num = args.start[0]
page_url = f"{download_url}/{page_num}" page_url = f"{download_url}/{page_num}"
response = session.get(page_url) response = session.get(page_url)
s = BeautifulSoup(response.text, "html.parser") s = BeautifulSoup(response.text, "html.parser")
if s.find(class_="loggedin_user_avatar") is not None: if s.find(class_="loggedin_user_avatar") is not None:
account_username = s.find(class_="loggedin_user_avatar").attrs.get("alt") account_username = s.find(class_="loggedin_user_avatar").attrs.get("alt")
print("Logged in as", account_username) print(f"{GREEN}<i> Logged in as: {account_username}{END}")
else: else:
print("Not logged in, NSFW content is inaccessible") print(f"{YELLOW}<i> Not logged in, NSFW content is inaccessible{END}")
# download loop
while True: while True:
if page_end == page_num: if page_end == page_num:
print(f"Reached page {page_end}, stopping.") print(f"{YELLOW}<i> Reached page {page_end}, stopping.{END}")
break break
page_url = f"{download_url}/{page_num}" page_url = f"{download_url}/{page_num}"
@ -305,50 +348,66 @@ def main():
.find("div") .find("div")
.find(class_="link-override") .find(class_="link-override")
.text.strip() .text.strip()
.replace(".", ". \n")
) )
except AttributeError: except AttributeError:
print("You didn't provide cookies to log in") message = (
break s.find(class_="notice-message")
raise Exception("System Message", message) .find("div", class_="section-body alignleft")
.find("div", class_="redirect-message")
.text.strip()
.replace(".", ". \n")
)
print(f"{YELLOW}<i> System Message: {message}{END}")
exit()
# End of gallery # End of gallery
if s.find(id="no-images") is not None: if s.find(id="no-images") is not None:
print("End of gallery") print(f"{GREEN}<i> End of gallery{END}")
break break
# Download all images on the page # Download all images on the page
for img in s.findAll("figure"): for img in s.findAll("figure"):
download(img.find("a").attrs.get("href")) download(img.find("a").attrs.get("href"))
sleep(args.interval) sleep(args.interval[0])
# Download submissions
if args.submissions is True: if args.submissions is True:
next_button = s.find("a", class_="button standard more", text="Next 48") try:
if next_button is None or next_button.parent is None: next_button = s.find("a", class_="button standard more").attrs.get(
next_button = s.find( "href"
"a", class_="button standard more-half", text="Next 48"
) )
if next_button is None or next_button.parent is None: except AttributeError:
print("Unable to find next button") try:
next_button = s.find(
"a", class_="button standard more-half"
).attrs.get("href")
except AttributeError:
print(f"{YELLOW}<!> Unable to find next button{END}")
break break
next_page_link = next_button.attrs["href"] # unlike galleries that are sequentially numbered, submissions use a different scheme.
page_num = next_page_link.split("/")[-2] # the "page_num" is instead: new~[set of numbers]@(12 or 48 or 72) if sorting by new
page_url = BASE_URL + next_page_link
print("Downloading page", page_num, page_url) page_num = next_button.split("/")[-2]
page_url = f"{BASE_URL}{next_button}"
print(f"{WHITE}<i> Downloading page {page_num} - {page_url} {END}")
# Download everything else
elif args.category != "favorites": elif args.category != "favorites":
next_button = s.find("button", class_="button standard", text="Next") next_button = s.find("button", class_="button standard", text="Next")
if next_button is None or next_button.parent is None: if next_button is None or next_button.parent is None:
print("Unable to find next button") print(f"{YELLOW}<!> Unable to find next button{END}")
break break
page_num = next_button.parent.attrs["action"].split("/")[-2] page_num = next_button.parent.attrs["action"].split("/")[-2]
print("Downloading page", page_num, page_url) print(f"{WHITE}<i> Downloading page {page_num} - {page_url} {END}")
# Download favorites
else: else:
next_button = s.find("a", class_="button mobile-button right", text="Next") next_button = s.find("a", class_="button standard right", text="Next")
if next_button is None: if next_button is None:
print("Unable to find next button") print(f"{YELLOW}<!> Unable to find next button{END}")
break break
# unlike galleries that are sequentially numbered, favorites use a different scheme. # unlike galleries that are sequentially numbered, favorites use a different scheme.
@ -358,16 +417,16 @@ def main():
next_fav_num = re.search(r"\d+", next_page_link) next_fav_num = re.search(r"\d+", next_page_link)
if next_fav_num is None: if next_fav_num is None:
print("Failed to parse next favorite link.") print(f"{YELLOW}<!> Failed to parse next favorite link{END}")
break break
page_num = next_fav_num.group(0) + "/next" page_num = next_fav_num.group(0) + "/next"
# parse it into numbers/next # parse it into numbers/next
print("Downloading page", page_num, page_url) print(f"{WHITE}<i> Downloading page {page_num} - {page_url} {END}")
print("Finished downloading") print(f"{GREEN}Finished downloading{END}")
if __name__ == "__main__": if __name__ == "__main__":