61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
from selenium import webdriver
|
|
from selenium.webdriver.chrome.options import Options
|
|
from PIL import Image
|
|
import time, os
|
|
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
|
|
import yaml
|
|
import datetime
|
|
|
|
# enable browser logging
|
|
d = DesiredCapabilities.CHROME
|
|
d['loggingPrefs'] = {'browser': 'ALL'}
|
|
|
|
image_path = os.environ.get("SCREEN_IMAGE_PATH", default='/tmp')
|
|
config_file = os.environ.get("SCREEN_CONFIG", default="/app/config.yaml")
|
|
width = os.environ.get("SCREEN_WIDTH", default="640")
|
|
height = os.environ.get("SCREEN_HEIGHT", default="480")
|
|
rotate = os.environ.get("SCREEN_ROTATE", default="0")
|
|
wait = os.environ.get("SCREEN_WAIT", default="5")
|
|
|
|
# Open another headless browser with height extracted above
|
|
chrome_options = Options()
|
|
chrome_options.add_argument("--headless")
|
|
chrome_options.add_argument('--no-sandbox')
|
|
chrome_options.add_argument(f"--window-size={width},{height}")
|
|
chrome_options.add_argument("--hide-scrollbars")
|
|
chrome_options.add_argument('user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, '
|
|
'like Gecko) Chrome/68.0.3440.84 Safari/537.36')
|
|
|
|
config = None
|
|
with open(config_file, "r") as stream:
|
|
config = yaml.safe_load(stream)
|
|
|
|
tasks = []
|
|
|
|
for image in config["images"]:
|
|
image["next_execution"] = datetime.datetime.now()
|
|
tasks += [image]
|
|
|
|
while True:
|
|
now = datetime.datetime.now()
|
|
for task in tasks:
|
|
if task["next_execution"] > now:
|
|
continue
|
|
print(f"Running {task['name']}")
|
|
driver = webdriver.Chrome(options=chrome_options, desired_capabilities=d)
|
|
driver.implicitly_wait(10)
|
|
driver.get(task["url"])
|
|
# pause 3 second to let page loads
|
|
time.sleep(int(wait))
|
|
# save screenshot
|
|
file_name = f'/tmp/{task["name"]}.png'
|
|
driver.save_screenshot(file_name)
|
|
# print messages
|
|
for entry in driver.get_log('browser'):
|
|
print(entry)
|
|
image = Image.open(file_name)
|
|
image = image.rotate(int(rotate), expand=True)
|
|
file_name = f'{image_path}/{task["name"]}.png'
|
|
image.save(file_name)
|
|
task["next_execution"] = now + datetime.timedelta(minutes=task["period_minutes"])
|
|
time.sleep(1)
|