from playwright.sync_api import Page

from src.helpers.logger import Logger
from src.locators.album_page.album_header import AlbumPageHeaderLocators
from src.pages.apollo_header_main import ApolloHeader
from src.ui_elements.date_range_day_picker import DateRangeUIElement
from src.ui_elements.dropdown_ui_element import DropDownUIElement

log = Logger()
tab_name = "Album Page Header - "


class AlbumPageHeader(ApolloHeader):

    def __init__(self, page: Page):
        super().__init__(page)
        self.page = page
        self.entity_type = '{0}/album/{1}'
        self.url = f'/{self.entity_type}'
        self.album_title = DropDownUIElement(self.page, AlbumPageHeaderLocators.ALBUM_TITLE,
                                        tab_name + 'album title')
        self.album_tab = DropDownUIElement(self.page, AlbumPageHeaderLocators.ALBUM_TAB,
                                        tab_name + 'album tab')
        self.track_tab = DropDownUIElement(self.page, AlbumPageHeaderLocators.TRACK_TAB,
                                        tab_name + 'track tab')
        self.countries_tab = DropDownUIElement(self.page, AlbumPageHeaderLocators.COUNTRIES_TAB,
                                        tab_name + 'countries tab')
        self.day_picker = DateRangeUIElement(self.page, AlbumPageHeaderLocators.DAY_PICKER,
                                             tab_name + 'day picker')