# coding: utf-8

######################################################
#                                                    #
#         ProstoScript для ProstoPleer'a =)          #
#   Code by IT.Tux.Droid <http://it-tux-droid.ru/>   #
#                                                    #
######################################################

import pycurl, StringIO, re, os, progressbar
from optparse import OptionParser

data = StringIO.StringIO()
curl = pycurl.Curl()
curl.setopt(pycurl.WRITEFUNCTION, data.write)

parser = OptionParser()
parser.add_option('-m', '--mode', dest='mode', metavar='MODE', default='0',
                  help='select mode: 0 - create playlist, 1 - download songs, 2 - create playlist and download songs')
parser.add_option('-u', '--url', dest='url', metavar='URL',
                  help='url to playlist or radio: http://prostopleer.com/#/top/msk/nasheradio')
(options, args) = parser.parse_args()

def progress(download_t, download_d, upload_t, upload_d):
        if (download_t != 0) and (download_d != download_t):
                bar.update((int(download_d)*100)/int(download_t))

def error(text):
        z = len(text.decode('utf-8'))
        print ' ????????????????'+'?'*z
        print ' ?       '+text+'       ?'
        print ' ????????????????'+'?'*z

def download_mp3(folder, song, url):
        global bar
        print '? '+song
        if not os.path.exists(folder):
                os.makedirs(folder)
        if not os.path.exists(folder+'/'+song+'.mp3'):
                bar = progressbar.ProgressBar(maxval=100, widgets=<'?   ?', progressbar.Bar(left='<', marker='=', right='>')>).start()
                data.truncate(0)
                curl.setopt(pycurl.URL, 'http://prostopleer.com/download/'+url)
                curl.setopt(curl.PROGRESSFUNCTION, progress)
                curl.setopt(curl.NOPROGRESS, 0)
                try:
                        curl.perform()
                except:
                        print '?   ?<connection error="error">'
                        return False
                bar.finish()
                mp3 = open(folder+'/'+song+'.mp3', 'w')
                mp3.write(data.getvalue())
                mp3.close()
        else:
                print '?   ?<you already="already" have="have" it="it">'

# Create .m3u playlist and/or download songs from your playlist.
# Example: "playlist('list346790Ntwn', 2)".
def playlist(p, m):
        data.truncate(0)
        curl.setopt(pycurl.URL,'http://prostopleer.com/'+p)
        curl.setopt(curl.NOPROGRESS, 1)
        try:
                curl.perform()
        except:
                error('Connection error!')
                return False

        list_name = re.findall("class=\"results\"&gt; &lt;h2&gt;(<^&lt;>*)&lt;/h2&gt;", data.getvalue())<0><20:-2>
        musics = re.findall("file_id=\"(<^\">*)\" track_id=\"(<^\">*)\" singer=\"(<^\">*)\" song=\"(<^\">*)\" link", data.getvalue())

        z = len(list_name.decode('utf-8'))

        if m in <0, 2>:
                m3u = open('playlist <'+list_name+'>.m3u', 'w')
                for music in musics:
                        m3u.write('#EXTINF: 0, '+music<2>+' - '+music<3>+'\nhttp://prostopleer.com/download/'+music<0>+'\n')
                m3u.close()
                print ' ?????????????????????????????????????????????????????'+'?'*z+'?'
                print ' ? Playlist from playlist ('+list_name+') was successfully created ?'
                print ' ?????????????????????????????????????????????????????'+'?'*z+'?'

        if m in <1, 2>:
                print ' ?????????????????????????????????????????????????????'+'?'*z+'?'
                print '??     Start download songs from your playlist ('+list_name+')     ?'
                print '??????????????????????????????????????????????????????'+'?'*z+'?'
                for music in musics:
                        download_mp3('playlist <'+list_name+'>', music<2>+' - '+music<3>, music<0>)
                print '??????????????????????????????????????????????????????'+'?'*z+'?'
                print '?? Songs from playlist ('+list_name+') was successfully downloaded ?'
                print ' ?????????????????????????????????????????????????????'+'?'*z+'?'

# Create .m3u playlist and/or download songs from radio.
# Example: "radio('top/msk/nasheradio', 2)".
def radio(r, f):
        data.truncate(0)
        curl.setopt(pycurl.URL,'http://prostopleer.com/'+r)
        curl.setopt(curl.NOPROGRESS, 1)
        try:
                curl.perform()
        except:
                error('Connection error!')
                return False

        radio_name = re.findall("radio-name\"&gt;(<^&lt;>*)&lt;/span&gt;", data.getvalue())<0>
        musics = re.findall("file_id=\"(<^\">*)\" singer=\"(<^\">*)\" song=\"(<^\">*)\" link", data.getvalue())

        z = len(radio_name.decode('utf-8'))

        if f in <0, 2>:
                m3u = open('radio <'+radio_name+'>.m3u', 'w')
                for music in musics:
                        m3u.write('#EXTINF: 0, '+music<1>+' - '+music<2>+'\nhttp://prostopleer.com/download/'+music<0>+'\n')
                m3u.close()
                print ' ?????????????????????????????????????????????????????'+'?'*z+'?'
                print ' ?  Playlist from radio ('+radio_name+') was successfully created   ?'
                print ' ?????????????????????????????????????????????????????'+'?'*z+'?'

        if f in <1, 2>:
                print ' ?????????????????????????????????????????????????????'+'?'*z+'?'
                print '??         Start download songs from radio ('+radio_name+')         ?'
                print '??????????????????????????????????????????????????????'+'?'*z+'?'
                for music in musics:
                        download_mp3('radio <'+radio_name+'>', music<1>+' - '+music<2>, music<0>)
                print '??????????????????????????????????????????????????????'+'?'*z+'?'
                print '??  Songs from radio ('+radio_name+') was successfully downloaded   ?'
                print ' ?????????????????????????????????????????????????????'+'?'*z+'?'

if options.mode in <'0', '1', '2'>:
        if options.url != None:
                u = re.findall('(list.*)', options.url)
                if u:
                        playlist(u<0>, int(options.mode))
                else:
                        u = re.findall('(top/.*)', options.url)
                        if u:
                                radio(u<0>, int(options.mode))
                        else:
                                error('Error URL!')
        else:
                error('Enter URL!')
else:
        error('Error MODE!')

curl.close()
</you></connection>