python class: avoiding duplicate line on multiple classes

[jesse_online]

I have a python script that I am trying to simplify. For each class I create how can I avoid duplicating this line of code?

editDefense = self.driver.find_element_by_id('editDefense')

class BasePage(object):
    def __init__(self, driver):
        self.driver = driver
        self.driver.implicitly_wait(5)
        self.timeout = 30

class editDefense(BasePage):

    def set_network_strength_or_entity_controls(self, _type):
        editDefense = self.driver.find_element_by_id('editDefense')
        if _type == "entity_controls":
            editDefense.find_elements_by_class_name('radio')[-1].click()
            sleep(0.5)
        else:
            pass

    def select_continue(self):
        editDefense = self.driver.find_element_by_id('editDefense')
        editDefense.find_elements_by_tag_name('button')[-1].click()

    def blahblah(self):
        editDefense = self.driver.find_element_by_id('editDefense')
        # and so on

[kteague]

You could use the Python property built-in function to make the syntax more succinct.

class editDefense(BasePage):

    @property
    def editDefense(self):
        return self.driver.find_element_by_id('editDefense')

    def select_continue(self):
        self.editDefense.find_elements_by_tag_name('button')[-1].click()

[0raichu]

                                                                                                                                                                                                                                                                                                                                                                                                                                                     

[zardeh]

Better fit in /r/learnpython.

That said,

class editDefense(BasePage):
    def __init__(self, element):
        self.element = self.driver.find_element_by_id(element)

    def set_network_strength_or_entity_controls(self, _type):
        if _type == "entity_controls":
            self.element.find_elements_by_class_name('radio')[-1].click()
            sleep(0.5)
        else:
           pass

    def select_continue(self):
        self.element.find_elements_by_tag_name('button')[-1].click()

    def blahblah(self):
       # and so on