From 3597c948f381291440bf4228e0b495b9db5b2857 Mon Sep 17 00:00:00 2001
From: Saurabh Bhan <GeekSpin@users.noreply.github.com>
Date: Tue, 28 Jun 2016 12:31:49 +0530
Subject: [PATCH] clean up

---
 wallpy.py | 26 +++++++++++++++-----------
 1 file changed, 15 insertions(+), 11 deletions(-)

diff --git a/wallpy.py b/wallpy.py
index 1b12851..79cddf1 100644
--- a/wallpy.py
+++ b/wallpy.py
@@ -7,6 +7,8 @@
 #                  dated- 26 June 2016                 #
 ########################################################
 
+#!usr/bin/env python3
+
 import os
 import bs4
 import re
@@ -15,17 +17,19 @@ import time
 import tqdm
 
 os.makedirs('Wallhaven', exist_ok=True)
-url = 'https://alpha.wallhaven.cc/latest'
-urlreq = requests.get(url)
-soup = bs4.BeautifulSoup(urlreq.text, 'lxml')
-soupid = soup.findAll('a', {'class': 'preview'})
-res = re.compile(r'\d+')
-imgid = res.findall(str(soupid))
-print('Number of Wallpapers to Download: ' + str(len(imgid)))
-imgext = ['jpg', 'png', 'bmp']
-for i in range(len(imgid)):
-    url = 'http://wallpapers.wallhaven.cc/wallpapers/full/wallhaven-%s.' % imgid[
-        i]
+pgid = int(input('How Many pages you want to Download: '))
+print('Number of Wallpapers to Download: ' + str(24 * pgid))
+for i in range(1, pgid + 1):
+    url = 'https://alpha.wallhaven.cc/latest?page=' + str(i)
+    urlreq = requests.get(url)
+    soup = bs4.BeautifulSoup(urlreq.text, 'lxml')
+    soupid = soup.findAll('a', {'class': 'preview'})
+    res = re.compile(r'\d+')
+    imgid = res.findall(str(soupid))
+    imgext = ['jpg', 'png', 'bmp']
+    for i in range(len(imgid)):
+        url = 'http://wallpapers.wallhaven.cc/wallpapers/full/wallhaven-%s.' % imgid[
+            i]
         for ext in imgext:
             iurl = url + ext
             imgreq = requests.get(iurl)