Merge pull request geekcomputers#112 from yashLadha/master

Script to download Images from google images and Wallpaper Kraft

Author: geekcomputers

Google Image Downloader/__pycache__/create_dir.cpython-35.pyc

@@ -0,0 +1,68 @@
+#!/usr/bin/python3
+
+"""
+Code to directly use in file to
+create directory in home location
+
+Note:- I Have used python package so if you want
+to create in the main directory of your project use
+pardir+"\\"+name in functions
+
+All the folder operations are done on home
+project directory.
+"""
+
+from shutil import copytree
+from shutil import move
+from os import chdir
+from os.path import exists
+from os.path import pardir
+from os import makedirs
+from os import removedirs
+from os import rename
+
+
+# Creates a directory
+def create_directory(name):
+    if exists(pardir+"\\"+name):
+        print('Folder already exists... Cannot Overwrite this')
+    else:
+        makedirs(pardir+"\\"+name)
+
+
+# Deletes a directory
+def delete_directory(name):
+    removedirs(name)
+
+
+# Rename a directory
+def rename_directory(direct, name):
+    rename(direct, name)
+
+
+# Sets the working directory
+def set_working_directory():
+    chdir(pardir)
+
+
+# Backup the folder tree
+def backup_files(name_dir, folder):
+    copytree(pardir, name_dir + ':\\' + folder)
+
+
+# Move folder to specific location
+# Overwrites the file if it already exists
+def move_folder(filename, name_dir, folder):
+    if not exists(name_dir+":\\"+folder):
+        makedirs(name_dir+':\\'+folder)
+    move(filename, name_dir+":\\"+folder+'\\')
+
+
+"""
+For test purpose:
+    1. create_directory("test")
+    2. rename_directory("test","demo")
+    3. delete_directory("demo")
+    4. backup_files('D', 'backup_project')
+    5. move_folder(pardir+'\\'+'test.txt', 'D', 'name')
+"""

Google Image Downloader/create_dir.py

@@ -0,0 +1,66 @@
+"""
+Code to directly use in file to
+create directory in home location
+
+Note:- I Have used python package so if you want
+to create in the main directory of your project use
+pardir+"\\"+name in functions
+
+All the folder operations are done on home
+project directory.
+"""
+
+from shutil import copytree
+from shutil import move
+from os import chdir
+from os.path import exists
+from os.path import pardir
+from os import makedirs
+from os import removedirs
+from os import rename
+
+
+# Creates a directory
+def create_directory(name):
+    if exists(pardir+"\\"+name):
+        print('Folder already exists... Cannot Overwrite this')
+    else:
+        makedirs(pardir+"\\"+name)
+
+
+# Deletes a directory
+def delete_directory(name):
+    removedirs(name)
+
+
+# Rename a directory
+def rename_directory(direct, name):
+    rename(direct, name)
+
+
+# Sets the working directory
+def set_working_directory():
+    chdir(pardir)
+
+
+# Backup the folder tree
+def backup_files(name_dir, folder):
+    copytree(pardir, name_dir + ':\\' + folder)
+
+
+# Move folder to specific location
+# Overwrites the file if it already exists
+def move_folder(filename, name_dir, folder):
+    if not exists(name_dir+":\\"+folder):
+        makedirs(name_dir+':\\'+folder)
+    move(filename, name_dir+":\\"+folder+'\\')
+
+
+"""
+For test purpose:
+    1. create_directory("test")
+    2. rename_directory("test","demo")
+    3. delete_directory("demo")
+    4. backup_files('D', 'backup_project')
+    5. move_folder(pardir+'\\'+'test.txt', 'D', 'name')
+"""

Google Image Downloader/create_dir.py~

@@ -0,0 +1,135 @@
+#!/usr/bin/python3
+
+from os import chdir
+import requests
+from bs4 import BeautifulSoup
+from urllib.request import urlopen, Request
+from urllib.parse import urlencode
+from os import walk
+import json
+from os.path import curdir
+from urllib.request import urlretrieve
+from os.path import pardir
+from create_dir import create_directory
+
+GOOGLE_IMAGE = 'https://www.google.com/search?site=&tbm=isch&source=hp&biw=1873&bih=990&'
+WALLPAPERS_KRAFT = 'https://wallpaperscraft.com/search/keywords?'
+usr_agent = {
+        'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.11 (KHTML, like Gecko) Chrome/23.0.1271.64 Safari/537.11',
+        'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
+        'Accept-Charset': 'ISO-8859-1,utf-8;q=0.7,*;q=0.3',
+        'Accept-Encoding': 'none',
+        'Accept-Language': 'en-US,en;q=0.8',
+        'Connection': 'keep-alive'
+        }
+
+
+def image_grabber(ch):
+
+    # Download images from google images
+    if ch == 1:
+        print('Enter data to download Images: ')
+        data = input()
+        search_query = {'q': data}
+        search = urlencode(search_query)
+        print(search)
+        g = GOOGLE_IMAGE + search
+        request = Request(g, headers=usr_agent)
+        r = urlopen(request).read()
+        sew = BeautifulSoup(r, 'html.parser')
+        images = []
+        # print(sew.prettify())
+        results = sew.findAll("div",{"class":"rg_meta"})
+        for re in results:
+            link, Type = json.loads(re.text)["ou"] , json.loads(re.text)["ity"]
+            images.append(link)
+        counter = 0
+        for re in images:
+            rs = requests.get(re)
+            with open('img' + str(counter) + '.jpg', 'wb') as file:
+                file.write(rs.content)
+                # urlretrieve(re, 'img' + str(count) + '.jpg')
+                counter += 1
+        return True
+
+    elif ch == 2:
+        cont = set()  # Stores the links of images
+        temp = set()  # Refines the links to download images
+
+        print('Enter data to download wallpapers: ')
+        data = input()
+        search_query = {'q': data}
+        search = urlencode(search_query)
+        print(search)
+        g = WALLPAPERS_KRAFT + search
+        request = Request(g, headers=usr_agent)
+        r = urlopen(request).read()
+        sew = BeautifulSoup(r, 'html.parser')
+        count = 0
+        for links in sew.find_all('a'):
+            if 'wallpaperscraft.com/download' in links.get('href'):
+                cont.add(links.get('href'))
+        for re in cont:
+            # print all valid links
+            # print('https://wallpaperscraft.com/image/' + re[31:-10] + '_' + re[-9:] + '.jpg')
+            temp.add('https://wallpaperscraft.com/image/' + re[31:-10] + '_' + re[-9:] + '.jpg')
+
+        # Goes to Each link and downloads high resolution images
+
+        for re in temp:
+            rs = requests.get(re)
+            with open('img' + str(count) + '.jpg', 'wb') as file:
+                file.write(rs.content)
+            # urlretrieve(re, 'img' + str(count) + '.jpg')
+            count += 1
+
+        return True
+
+    elif ch == 3:
+        for folders, subfolder, files in walk(curdir):
+            for folder in subfolder:
+                print(folder)
+        return True
+
+    elif ch == 4:
+        print('Enter the directory to be set: ')
+        data = input()
+        chdir(data + ':\\')
+        print('Enter name for the folder: ')
+        data = input()
+        create_directory(data)
+        return True
+
+    elif ch == 5:
+        print(
+                '''
+-------------------------***Thank You For Using***-------------------------
+            '''
+            )
+        return False
+
+
+run = True
+
+print(
+        '''
+***********[First Creating Folder To Save Your Images}***********
+    '''
+    )
+
+create_directory('Images')
+DEFAULT_DIRECTORY = pardir + '\\Images'
+chdir(DEFAULT_DIRECTORY)
+
+while run:
+    print('''
+-------------------------WELCOME-------------------------
+    1. Search for image
+    2. Download Wallpapers 1080p
+    3. View Images in your directory
+    4. Set directory
+    5. Exit
+-------------------------*******-------------------------
+    ''')
+    choice = input()
+    run = image_grabber(int(choice))