Si j'ai plusieurs classe comme la classe Page, classe Page1, classe Page2 qui ont besoin d'utiliser la même classe Element. Afin d'obtenir self.driver à utiliser sur la classe ELement, je vais devoir dupliquer beaucoup de code de class Ele(e)
à class itemWrap
sur chaque class Page
pour passer self.driver
en entrée. Quelle est la meilleure façon pour moi de construire le code suivant?Quelle est la bonne façon de construire la classe, objet en python
# elements.py
class Element(object):
def addItem(self, owner=None):
for _i in owner:
options = self.findAvaItem()
i = options.index(_i)
self.ele.find_element_by_xpath('{}/{}'.format(self.l, i)).click()
def delItem(self, owner=None):
for _i in owner:
options = self.findSelecItem()
i = options.index(_i)
self.ele.find_element_by_xpath('{}/{}'.format(self.l, i)).click()
def findAvaItem(self):
_o = self.driver.find_element_by_css_selector('css==div.menu.transition.visible').text
return _o.split('\n')
def findSelecItem(self):
_o = self.driver.find_element_by_css_selector('css==a.ui.label')
return [i.text for i in _o]
# BasePage.py
class BasePage(object):
def __init__(self, driver):
self.driver = driver
# Page.py
from elements import Element as e
class Ele(e):
def __init__(self, driver, _t):
self.driver = driver
self.l = 'element l variable'
self.t = _t
class itemWrap(object):
def __init__(self, driver, _type):
self.ele = Ele(driver, _type)
self.driver = driver
self.t = _type
def sel_item(self, _value):
self.ele.addItem(_value)
def desel_item(self, _value):
self.ele.delItem(_value)
class Page(BasePage):
def inputItem(self, _type, _value):
itemWrap(self.driver, _type).sel_item(_value)
def removeItem(self, _type, _value):
itemWrap(self.driver, _type).desel_item(_value)
# Test.py
from Page import Page
from selenium import webdriver as WebDriver
class Test(object):
@classmethod
def setup_class(cls):
if hasattr(super(Test, cls), 'setup_class'):
super(Test, cls).setup_class()
cls.driver = WebDriver(desired_capabilities=chrome, command_executor='127.0.0.1')
def check_function(self):
self.Page.inputItem('typeA', ['item A1', 'item A2', 'item A3'])
self.Page.removeItem('typeA', ['item A1', 'item A2', 'item A3'])
self.Page.inputItem('typeB', ['item B1', 'item B2'])
self.Page.removeItem('typeB', ['item B1', 'item B2'])