#!/usr/bin/env python3

# -*- coding:utf-8 -*-

from selenium import webdriver

from selenium.webdriver.common.action_chains import ActionChains

import time

class LoginTest(object):

    def __init__(self):

        self.url = ''

        self.uname = ''

        self.pwd = ''

        self.userEle = ''

        self.pwdEle = ''

        self.loginEle = ''

        self.status = ''

        self.statusCode = ''

    def openUrl(self):

        self.driver = webdriver.Firefox()

        self.driver.get(self.url)

        time.sleep(10)

    def findLoginEle(self):

        # id = "user"

        self.userbTn = self.driver.find_element_by_id(self.userEle)

        self.pwdbTn = self.driver.find_element_by_id(self.pwdEle)

        self.loginbTn = self.driver.find_element_by_id(self.loginEle)

        # print(self.userId,self.pwdId,self.loginId)

    def logIn(self):

        self.userbTn.clear()

        self.userbTn.send_keys(self.uname)

        self.pwdbTn.clear()

        self.pwdbTn.send_keys(self.pwd)

        self.loginbTn.click()

        time.sleep(30)

    def checkStatus(self):

        try:

            self.statusbTn = self.driver.find_element_by_id(self.status)

            if self.statusbTn.text == self.statusCode:

                print(self.statusCode)

                return False

        except:

            print("Pass")

            return True

    def quitBrowse(self):

        self.driver.quit()

if __name__ == '__main__':

    test1 = LoginTest()

    userInfo = {

       'uname':['Administrator','USERID'],

        'pwd':['Aa1','PASSW0RD']

    }

    test1.url = 'https://10.245.53.64/designs/imm/index.php'

    for uname,pwd in zip(userInfo['uname'],userInfo['pwd']):

        print(uname,pwd)

        test1.uname = uname

        test1.pwd = pwd

        test1.userEle = 'user'

        test1.pwdEle = 'password'

        test1.loginEle = 'btnLogin_label'

        test1.status = 'gblMessageDlgID_title'

        test1.statusCode = 'Login failed.'

    test1.openUrl()

    test1.findLoginEle()

    test1.logIn()

    if test1.checkStatus() == False:

        test1.quitBrowse()