File size: 2,014 Bytes
e71d833 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
import requests
from typing import List
from termcolor import colored
def search_for_stock_videos(query: str, api_key: str, it: int, min_dur: int) -> List[str]:
"""
Searches for stock videos based on a query.
Args:
query (str): The query to search for.
api_key (str): The API key to use.
Returns:
List[str]: A list of stock videos.
"""
# Build headers
headers = {
"Authorization": api_key
}
# Build URL
qurl = f"https://api.pexels.com/videos/search?query={query}&per_page={it}"
# Send the request
r = requests.get(qurl, headers=headers)
# Parse the response
response = r.json()
# Parse each video
raw_urls = []
video_url = []
video_res = 0
try:
# loop through each video in the result
for i in range(it):
#check if video has desired minimum duration
if response["videos"][i]["duration"] < min_dur:
continue
raw_urls = response["videos"][i]["video_files"]
temp_video_url = ""
# loop through each url to determine the best quality
for video in raw_urls:
# Check if video has a valid download link
if ".com/external" in video["link"]:
# Only save the URL with the largest resolution
if (video["width"]*video["height"]) > video_res:
temp_video_url = video["link"]
video_res = video["width"]*video["height"]
# add the url to the return list if it's not empty
if temp_video_url != "":
video_url.append(temp_video_url)
except Exception as e:
print(colored("[-] No Videos found.", "red"))
print(colored(e, "red"))
# Let user know
print(colored(f"\t=> \"{query}\" found {len(video_url)} Videos", "cyan"))
# Return the video url
return video_url
|