from playwright.sync_api import Page

from src.helpers.logger import Logger
from src.locators.charts_digest_page.bubbling_under import ChartsDigestBubblingUnderLocators
from src.pages.apollo_header_main import ApolloHeader
from src.ui_elements.base_ui_element import BaseUIElement


class CDBubblingUnder(ApolloHeader):
    log = Logger()

    def __init__(self, page: Page):
        super().__init__(page)
        self.page = page
        self.entity_type = ''
        self.url = f'/{self.entity_type}'
        self.view_chart_button = BaseUIElement(self.page, ChartsDigestBubblingUnderLocators.VIEW_CHART_BUTTON,
                                         'Charts Digest - Bubbling Under - view chart button')
        self.apple_tab = BaseUIElement(self.page, ChartsDigestBubblingUnderLocators.APPLE_TAB,
                                               'Charts Digest - Bubbling Under - apple tab')