Music-Downloader/sites/helper/item.py

import console


class Item:
    def __init__(self, site, title, url, duration='00:00', size=0, artist=None):
        self.site = site
        self.original_url = url
        self.download_url = None
        self.title = title
        self.duration = duration
        self.size = size
        self.artist = artist

        self.url_formatted = False

    def format_original_url(self):
        if not self.url_formatted:
            self.download_url = self.original_url = self.site.format_url(self.original_url)
            console.output('Setting original URL to: {0}'.format(self.original_url), console.DBG_INFO)
            self.url_formatted = True

    def set_download_url(self, url):
        console.output('Setting download url to: {0}'.format(url), console.DBG_INFO)
        self.download_url = url