apiRequests.py

# This scripts handles the retrival part of the api

################ IMPORTS ####################

from pypexels import PyPexels
import requests
import os

##############################################

global search_urls
search_urls = list()

############# main ###########################

# function that retrieves list of all urls from api from a giving string
def execute_search(key, string, pages, format, progress_function):

    if (len(search_urls) > 0):
        search_urls.clear()

    pexels_instance = PyPexels(api_key=key)
    search_results = pexels_instance.search(query=string, per_page=pages)

    progress_function(0)

    while True:
        for image in search_results.entries:
            search_urls.append(image.src.get(format))
            print(image.src.get(format))
        if not search_results.has_next:
            break
        search_results = search_results.get_next_page()

    progress_function('finished')  


#function that downloads images on a certain path
def download_images(urls, path_, download_amount, progress_function):
    path = path_
    if (len(path.split('./')) == 1):
        path = './' + path
    if (path[len(path)-1] != '/'):
        path = path + '/'
    print(path)
    if not os.path.exists(path):
        os.makedirs(path)
    for url in urls:
        
        count = (urls.index(url) + 1)
        segment = (len(url.split('/')) - 1)
        img_name = path + url.split('/')[segment].split('?')[0]

        if (count <= download_amount):
            porcentage = int((count*100)/(download_amount))
            progress_function(porcentage, count, download_amount)
            img = requests.get(url).content
            if not os.path.exists(img_name):
                with open(img_name, 'wb') as image:
                    image.write(img)
            print(img_name, ' has been downloaded sucessfully!')
    
    print('All downloads completed!')
    



###############################################