54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import shutil
|
|
import sys
|
|
from pathlib import Path
|
|
from time import sleep
|
|
|
|
import requests
|
|
|
|
import mangadlp.utils as utils
|
|
|
|
|
|
# download images
|
|
def download_chapter(
|
|
image_urls: list, chapter_path: str or Path, download_wait: float, verbose: bool
|
|
) -> None:
|
|
total_img = len(image_urls)
|
|
for img_num, img in enumerate(image_urls, 1):
|
|
# set image path
|
|
image_path = Path(f"{chapter_path}/{img_num:03d}")
|
|
# show progress bar if verbose logging is not active
|
|
if verbose:
|
|
print(f"INFO: Downloading image {img_num}/{total_img}")
|
|
else:
|
|
utils.progress_bar(img_num, total_img)
|
|
|
|
counter = 1
|
|
while counter <= 3:
|
|
try:
|
|
r = requests.get(img, stream=True)
|
|
if r.status_code != 200:
|
|
print(f"ERR: Request for image {img} failed, retrying")
|
|
raise ConnectionError
|
|
except KeyboardInterrupt:
|
|
print("ERR: Stopping")
|
|
sys.exit(1)
|
|
except:
|
|
if counter >= 3:
|
|
print("ERR: Maybe the MangaDex Servers are down?")
|
|
raise ConnectionError
|
|
sleep(download_wait)
|
|
counter += 1
|
|
else:
|
|
break
|
|
|
|
# write image
|
|
try:
|
|
with image_path.open("wb") as file:
|
|
r.raw.decode_content = True
|
|
shutil.copyfileobj(r.raw, file)
|
|
except:
|
|
print("ERR: Can't write file")
|
|
raise IOError
|
|
|
|
img_num += 1
|
|
sleep(download_wait)
|