onelife, base.py

from selenium import webdriver
from selenium.webdriver.support.select import Select
from selenium.webdriver.common.action_chains import ActionChains
from time import sleep
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys

import time

class BasePage():
enterPath = ()

def __init__(self, driver):
self.driver = driver

def open(self, url=None):
if url is None:
self.driver.get(self.url)
else:
self.driver.get(url)

def location(self, *args):
if args[2]=='':
return self.driver.find_element(getattr(By, args[0]), args[1])
else:
return self.driver.find_elements(getattr(By, args[0]), args[1])[int(args[2])]

def input(self, text, *args):
sleep(0.3)
ele = self.location(*args)
ele.clear()
sleep(0.2)
ele.send_keys(text)
sleep(0.3)
# ele.send_keys(Keys.ENTER)
ele.send_keys(Keys.TAB)

def input_enter(self, text, *args):
sleep(0.2)
ele = self.location(*args)
ele.clear()
sleep(0.2)
ele.send_keys(text)
sleep(0.2)
ele.send_keys(Keys.ENTER)

def click(self, *args):
sleep(0.3)
self.location(*args).click()
sleep(0.3)

def double_click(self, *args):
sleep(0.5)
ele = self.location(*args)
ActionChains(self.driver).double_click(ele).perform()

def max_window(self):
self.driver.maximize_window()

def enterPath(self, *paths):
for path_ in paths:
self.driver.find_element_by_link_text(path_).click()

iframes = self.driver.find_elements_by_xpath('//iframe')
self.driver.switch_to.frame(iframes[1])