PageChecker/main.py at master · qliknln/PageChecker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
from mail import MyMail
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as ec
from time import localtime, strftime, sleep, time
import json, os


class PageChecker(object):
    def __init__(self, url, wait_time, log_file, title, elem_class='', elem_id= ''):
        self.file_str = log_file
        self.url = url
        self.elem_class = elem_class
        self.elem_id = elem_id
        self.wait_time = wait_time
        self.title = title

    def check_page(self):
        driver = webdriver.Chrome()
        driver.implicitly_wait(10)
        driver.get(self.url)
        sleep(3)
        try:
            start = time()
            wait = WebDriverWait(driver, self.wait_time)
            if self.elem_class != '':
                element = wait.until(ec.presence_of_element_located((By.CLASS_NAME, self.elem_class)))
            else:
                element = wait.until(ec.presence_of_element_located((By.ID, self.elem_id)))
            stop = time()
            assert element is not None, 'Page not properly loaded'
            with open(self.file_str, 'a') as f:
                f.write(strftime('%Y-%m-%d %H:%M:%S', localtime()) + ';' + self.url + ';' + self.title + ';Up;' + str(stop-start)+'\r\n')
        except:
            with open(self.file_str, 'a') as f:
                f.write(strftime('%Y-%m-%d %H:%M:%S', localtime()) + ';' + self.url + ';' + self.title + ';Down;' + 'N/A\r\n')
            MyMail.send_mail(self.url + ' Down!', "Could not access the access point/hub for: " + self.title, 'qlikatqlik@qlik.com')
        finally:
            driver.quit()


if __name__ == '__main__':
    conf_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'config.json')
    with open(conf_file) as json_data:
        d = json.load(json_data)
        for obj in d[1]:
            pc = PageChecker(obj["pageUrl"], obj["elementWait"], d[0]['LogFile'], obj['title'], obj["elemClass"], obj["elemId"])
            pc.check_page()
            sleep(5) # Wait 5 before testing next page