Zenn
📌

GoogleCloudFunctionsでseleniumを利用してscreenshotを保存する

2022/09/05に公開
Python
Cloud Storage
#
gcf
tech

目標

  • GoogleCloudFunctions（GCF）でseleniumを利用する
  • ウェブサイトのscreenshotをGoogleCloudStorage（GCS）に保存する

完成品

https://github.com/dicechick373/gcf-selenium-screenshot

main.py

'YOUR BUCKET NAME'にバケット名を指定

import json
import os
import time
from selenium import webdriver
from google.cloud import storage

def  main(request):
    # chrome_options
    chrome_options = webdriver.ChromeOptions()
    chrome_options.add_argument('--headless')
    chrome_options.add_argument('--disable-gpu')
    chrome_options.add_argument('--no-sandbox')
    chrome_options.add_argument('--hide-scrollbars')
    chrome_options.add_argument('--enable-logging')
    chrome_options.add_argument('--log-level=0')
    chrome_options.add_argument('--v=99')
    chrome_options.add_argument('--single-process')
    chrome_options.add_argument('--ignore-certificate-errors')
    chrome_options.binary_location = os.getcwd() + "/headless-chromium"

    driver = webdriver.Chrome(os.getcwd() + "/chromedriver", chrome_options=chrome_options)

    # set url
    url = 'https://www.yahoo.co.jp/'
    driver.get(url)

    # file name
    file_name = "screenshot.png"

    # save screenshot
    time.sleep(5)
    driver.save_screenshot("/tmp/{}".format(file_name))

    # upload google cloud storage
    storage_client = storage.Client()
    bucket = storage_client.bucket('YOUR BUCKET NAME')
    blob = bucket.blob("img/{}".format(file_name))
    blob.upload_from_filename("/tmp/{}".format(file_name))

    driver.quit()
    return  "screenshot upload finished"

requirements.txt

google-cloud-storage==1.23.0

参考サイト

https://blowup-bbs.com/gcp-cloud-functions-python3/
https://note.com/financedog/n/n9a8d655b83ce

Discussion