def fetch_images_from_pexels()

in distilvit/pexels.py [0:0]

• 17 lines of code
• 5 McCabe index (conditional complexity)

def fetch_images_from_pexels(api_key, per_page, query, total_images=1000, start_page=1):
    headers = {"Authorization": api_key}
    page = start_page
    num = 0
    while num < total_images:
        response = requests.get(
            PEXELS_API_URL + query,
            headers=headers,
            params={"per_page": per_page, "page": page},
        )
        response_json = response.json()

        if not response_json["photos"]:
            break  # Exit if no more photos are available

        for image in response_json["photos"]:
            yield image
            num += 1

        page += 1