GitHub/Wallhaven-dl
1
0
Fork 0
mirror of https://github.com/saurabhan/Wallhaven-dl.git synced 2026-01-03 00:18:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added Page support and Progress bar

Browse Source 
Added Number of pages you want to download wallpapers.
Added a Download Progress bar to show the current download Progress.
This commit is contained in:
Saurabh Bhan
2016-06-27 23:07:24 +05:30
committed by GitHub
parent 77ccdaa8df
commit d427d05311
1 changed files with 12 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
wallpy.py
View File

@@ -11,6 +11,8 @@ import os
import bs4
import re
import requests
import time
import tqdm
os.makedirs('Wallhaven', exist_ok=True)
url = 'https://alpha.wallhaven.cc/latest'
@@ -23,16 +25,13 @@ imgext = ['jpg', 'png', 'bmp']
for i in range(len(imgid)):
url = 'http://wallpapers.wallhaven.cc/wallpapers/full/wallhaven-%s.' % imgid[
i]
iurl = url + imgext[0]
imgreq = requests.get(iurl)
er = imgreq.status_code
if er == 404:
iurl == url + imgext[1]
er2 = imgreq.status_code
if er2 == 404:
iurl == url + imgext[2]
print('Downloading: ' + iurl)
imageFile = open(os.path.join('Wallhaven', os.path.basename(iurl)), 'ab')
for chunk in imgreq.iter_content(100000):
imageFile.write(chunk)
imageFile.close()
for ext in imgext:
iurl = url + ext
imgreq = requests.get(iurl)
if imgreq.status_code == 200:
print('Downloading: ' + iurl)
with open(os.path.join('Wallhaven', os.path.basename(iurl)), 'ab') as imageFile:
for chunk in tqdm.tqdm(imgreq.iter_content(1024), total=(int(imgreq.headers['content-length']) / 1024), unit='KB'):
time.sleep(0.01)
imageFile.write(chunk)
break