from sites.default import DefaultSite
from helper.sites.query import Query
import requests


class MusicTeam(DefaultSite):
    def __init__(self):
        super().__init__()
        self.url = 'https://music-team.cc/music'
        self.query = Query(self, 'GET')\
            .add_parameter('search', '{0}')
        self.structure\
            .set_container_path('div.col-lg-8')\
            .set_item_path('div.card')\
            .set_title_path('div.card-body')\
            .set_url_path('a.leftDownload')
        self.request\
            .add_header('Referer', self.url + '/')\
            .add_header('Origin', self.url)\
            .add_cookie('visited', 'yes')\
            .add_cookie('robot', '1')

    def format_url(self, field):
        link = super().format_url(field)

        request = requests.get(link, cookies=self.request.cookie, allow_redirects=True)

        return request.url