71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
# coding: utf-8
|
|
|
|
import time
|
|
import logging
|
|
import os
|
|
from selenium.webdriver.support.wait import WebDriverWait
|
|
import selenium.webdriver.support.expected_conditions as EC
|
|
from selenium.common.exceptions import TimeoutException
|
|
import undetected_chromedriver as uc
|
|
from pathlib import Path
|
|
|
|
|
|
logging.basicConfig(level=10)
|
|
logger = logging.getLogger('test')
|
|
|
|
def main():
|
|
for k,v in os.environ.items():
|
|
logger.info("%s = %s" % (k,v))
|
|
logger.info('==== END ENV ==== ')
|
|
tmp = Path('/tmp').resolve()
|
|
for item in tmp.rglob('**'):
|
|
logger.info('found %s ' % item)
|
|
|
|
if item.is_dir():
|
|
if 'chrome-' in item.name:
|
|
logger.info('adding %s to PATH' % str(item))
|
|
logger.info('current PATH:' % str(os.environ.get('PATH')))
|
|
path_list = os.environ['PATH'].split(os.pathsep)
|
|
path_list.insert(0, str(item))
|
|
os.environ['PATH'] = os.pathsep.join(path_list)
|
|
logger.info('new PATH:' % str(os.environ.get('PATH')))
|
|
break
|
|
|
|
|
|
|
|
driver = uc.Chrome(headless=True)
|
|
logging.getLogger().setLevel(20)
|
|
|
|
driver.get('https://www.nowsecure.nl')
|
|
|
|
print(driver.current_url)
|
|
|
|
try:
|
|
WebDriverWait(driver,10).until(EC.visibility_of_element_located(("css selector", "body")))
|
|
except TimeoutException:
|
|
pass
|
|
print(driver.current_url)
|
|
try:
|
|
WebDriverWait(driver,10).until(EC.text_to_be_present_in_element(("css selector", "main h1"), "OH YEAH, you passed!"))
|
|
except TimeoutError:
|
|
logging.getLogger().setLevel(20)
|
|
driver.reconnect()
|
|
print(driver.current_url)
|
|
logger.info('trying to save a screenshot via imgur')
|
|
# driver.reconnect()
|
|
driver.save_screenshot('/tmp/screenshot.jpg')
|
|
driver.get('https://imgur.com/upload')
|
|
driver.find_element('css selector', 'input').send_keys('/tmp/screenshot.jpg')
|
|
|
|
time.sleep(2)
|
|
logger.info('A SCREENSHOT IS SAVED ON %s' % driver.current_url)
|
|
time.sleep(5)
|
|
driver.quit()
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|