def __init__(self, driver): super(AdminDashboard, self).__init__(driver) self.tab_people = widget_bar.Tab(self._driver, locator.WidgetBar.ADMIN_PEOPLE) self.tab_roles = widget_bar.Tab(self._driver, locator.WidgetBar.ADMIN_ROLES) self.tab_events = widget_bar.Tab(self._driver, locator.WidgetBar.ADMIN_EVENTS) self.tab_custom_attributes = widget_bar.Tab( self._driver, locator.WidgetBar.ADMIN_CUSTOM_ATTRIBUTE)
def _get_widget(self, widget_name): """Add attribute, click on relevant tab and return relevant widget.""" attr_name = "tab_" + widget_name setattr( self, attr_name, widget_bar.Tab(self._driver, factory.get_locator_widget(widget_name))) getattr(self, attr_name).click() return factory.get_cls_widget(widget_name)(self._driver, widget_name)