今のお仕事では業務の効率化のための自動化ツールの作成を行っています。
SalesforceからデータをExcelフォーマットに転記してSFのキャプチャをメールに貼って送るという業務が週に1回行われているのですが、この業務の自動化を依頼されました。
RPAで実行できるものはRPAで作成するのですが、RPAでダッシュボードやレポートのみのキャプチャをとるのが難しく、Pythonで実装することになりました。
（Win + Shift + SをSendKeysしてフルサイズのスクリーンショットをとったあとに座標をもとにトリミングしても良かったのですが、端末が変わると意図した場所でトリミングできなかったりしたので断念しました・・・）
実装したのがこちら↓
utils.py
from selenium import webdriver
from selenium.webdriver.common.by import By
def take_screenshot(url, identifier, output_filename):
try:
driver = webdriver.Edge()
driver.get(url)
driver.maximize_window()
time.sleep(1)
# Salesforce login
# Read username and password from file
with open(r'filepath', 'r') as file:
lines = file.readlines()
sfusername = lines[0].strip()
sfpw = lines[1].strip()
username = driver.find_element(By.NAME, 'username')
pw = driver.find_element(By.NAME, 'pw')
username.clear()
pw.clear()
username.send_keys(sfusername)
pw.send_keys(sfpw)
username.submit()
time.sleep(5)
# Determine if the identifier is an ID or a class
element = None
if identifier.startswith('#'):
element = driver.find_element(By.ID, identifier[1:])
# Click the refresh button
refresh = driver.find_element(By.NAME, 'refresh')
refresh.click()
time.sleep(120)
elif identifier.startswith('.'):
element = driver.find_element(By.CLASS_NAME, identifier[1:])
else:
raise ValueError("Identifier must start with '#' for ID or '.' for class")
# Get the location and size of the element
location = element.location
size = element.size
# Take a screenshot of the webpage
driver.save_screenshot('full_page_screenshot.png')
# Crop the screenshot to the specified element
left = location['x']
top = location['y']
right = location['x'] + size['width']
bottom = location['y'] + size['height']
from PIL import Image
# Open the full page screenshot
image = Image.open('full_page_screenshot.png')
# Crop the screenshot to the specified element
element_screenshot = image.crop((left, top, right, bottom))
# Save the screenshot
element_screenshot.save(output_filename)
# Delete the temporary file
os.remove('full_page_screenshot.png')
#driver.quit()
finally:
# Quit the WebDriver
if driver:
driver.quit()
解説
1. HTMLの要素(element)を取得
# Determine if the identifier is an ID or a class
element = None
if identifier.startswith('#'):
element = driver.find_element(By.ID, identifier[1:])
# Click the refresh button
refresh = driver.find_element(By.NAME, 'refresh')
refresh.click()
time.sleep(120)
elif identifier.startswith('.'):
element = driver.find_element(By.CLASS_NAME, identifier[1:])
else:
raise ValueError("Identifier must start with '#' for ID or '.' for class")
2. 要素の位置、サイズを取得
# Get the location and size of the element
location = element.location
size = element.size
3. フルサイズのスクリーンショットをとったあとに上下左右の座標を元にトリミング
# Take a screenshot of the webpage
driver.save_screenshot('full_page_screenshot.png')
# Crop the screenshot to the specified element
left = location['x']
top = location['y']
right = location['x'] + size['width']
bottom = location['y'] + size['height']
from PIL import Image
# Open the full page screenshot
image = Image.open('full_page_screenshot.png')
# Crop the screenshot to the specified element
element_screenshot = image.crop((left, top, right, bottom))
という流れで指定した要素のスクリーンショットを取得しています。
これで毎回安定してスクリーンショットを取得できるようになりました。
要素のidやnameを指定すれば好きな要素のスクリーンショットを取得できるのでとても便利です。
この取得したスクリーンショットをメール本文に埋め込んで送信するというのもPythonで実装したので別の記事でまとめたいと思います。
ソースコードはGitHubに上げています↓
GitHub : sf-screenshot-tool