import json

import requests
import urllib
import os
import sys
from urllib.parse import urlparse,parse_qs


def http_get(api):
    my_cookie = {
    "version":0,
    "name":'appver',
    "value":'1.5.0.75771',
    "port":None,
    # "port_specified":False,
    "domain":'www.mydomain.com',
    # "domain_specified":False,
 

    # "domain_initial_dot":False,
    "path":'/',
    # "path_specified":True,
    "secure":False,
    "expires":None,
    "discard":True,
    "comment":None,
    "comment_url":None,
    "rest":{},
    "rfc2109":False
    }

    s = requests.Session()
    s.headers.update({'Referer': "http://music.163.com/"
 
})
    s.cookies.set(**my_cookie)
    response  = s.get(api)
    json_data = json.loads(response.text)
    return json_data


def download_single(id):
    size = "720" #default 720p
    api = "http://music.163.com/api/mv/detail?id="+str(id)+"&type=mp4"
    json_data = http_get(api)
    if json_data["code"
 
]==200:
        mvurl = json_data["data"]["brs"][size]
        artist = json_data["data"]["artistName"]
        song = json_data["data"]["name"]

        filename = '%s/[%s]%s.mp4' %(artist,size,song)

        if os.path.exists(filename)==False:
            if  os.path.exists(artist)==False:
                os.makedirs(artist)
            def reporthook(blocknum, blocksize, totalsize):
                readsofar = blocknum * blocksize
                if totalsize > 0:
                    percent = readsofar * 1e2 / totalsize
                    s = "\r%5.1f%% %*d / %d" % (
                        percent, len(str(totalsize)), readsofar, totalsize)
                    sys.stderr.write(s)
                    if readsofar >= totalsize: # near the end
                        sys.stderr.write("\n")
                else: # total size is unknown
                    sys.stderr.write("read %d\n" % (readsofar,))
            print("downloading "+filename)
            urllib.request.urlretrieve(mvurl,filename,reporthook)



input = input("input the mv url:").replace("#","")
id = parse_qs(urlparse(input).query)["id"][0]
if "playlist" in input:
   playlist_api = "http://music.163.com/api/playlist/detail?id=%s" %(id)
   json_data = http_get(playlist_api)
   for idx, mv in enumerate(json_data["result"]["tracks"]):
       download_single(mv["mvid"])
       print("downloaded:"+str(idx))
elif "album" in input:
    playlist_api = "http://music.163.com/api/album/%s" %(id)
    json_data = http_get(playlist_api)
    for idx, mv in enumerate(json_data["album"]["songs"]):
        if mv["mvid"] != None and mv["mvid"]!=0 :
            download_single(mv["mvid"])
            print("downloaded:"+str(idx))

    download_single(id)