get_images.py 文件源码

def save_images(term, count):
    api_key = get_api_key()
    images = search_images(term, count, api_key)
    filenames = []

    if not os.path.exists(SAVE_DIR):
        os.makedirs(SAVE_DIR)

    for i, img in enumerate(images):
        if img['encodingFormat'] == 'unknown':
            continue
        name = "{path}/{filename}.{ext}".format(
            path=SAVE_DIR,
            filename="_".join(term.split()) + str(i),
            ext=img['encodingFormat'])
        try:
            download_image(img['thumbnailUrl'], name)
            filenames.append(name)
        except urllib.error.HTTPError:
            pass

    return filenames