netease_lyric.py

# /usr/bin/env python
# coding: utf-8
#------------------------------------------
import json
import requests
import re
import os
from io import BytesIO
from PIL import Image, ImageDraw, ImageFont
import optparse
import random
#------------------------------------------

class NetEase():

    cookies_filename = "netease_cookies.json"

    def __init__(self):
        self.headers = {
            'Host': 'music.163.com',
            'Connection': 'keep-alive',
            'Content-Type': "application/x-www-form-urlencoded; charset=UTF-8",
            'Referer': 'http://music.163.com/',
            "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_2) AppleWebKit/537.36"
                          " (KHTML, like Gecko) Chrome/33.0.1750.152 Safari/537.36"
        }
        self.cookies = dict(appver="1.2.1", os="osx")

    def show_progress(self, response):
        content = bytes()
        total_size = response.headers.get('content-length')
        if total_size is None:
            content = response.content
            return content
        else:
            total_size = int(total_size)
            bytes_so_far = 0

            for chunk in response.iter_content():
                content += chunk
                bytes_so_far += len(chunk)
                progress = round(bytes_so_far * 1.0 / total_size * 100)
                self.signal_load_progress.emit(progress)
            return content

    def http_request(self, method, action, query=None, urlencoded=None, callback=None, timeout=1):
        headers={
            'Host': 'music.163.com',
            'Connection': 'keep-alive',
            'Content-Type': "application/x-www-form-urlencoded; charset=UTF-8",
            'Referer': 'http://music.163.com/',
            "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_2) AppleWebKit/537.36"
                          " (KHTML, like Gecko) Chrome/33.0.1750.152 Safari/537.36"
        }
        cookies = dict(appver="1.2.1", os="osx")
        res = None
        if method == "GET":
            res = requests.get(action, headers=headers, cookies=cookies, timeout=timeout)
        elif method == "POST":
            res = requests.post(action, query, headers=self.headers, cookies=self.cookies, timeout=timeout)
        elif method == "POST_UPDATE":
            res = requests.post(action, query, headers=self.headers, cookies=self.cookies, timeout=timeout)
            self.cookies.update(res.cookies.get_dict())
            self.save_cookies()
        content = self.show_progress(res)
        content_str = content.decode('utf-8')
        content_dict = json.loads(content_str)
        return content_dict

    def user_playlist(self, pid):
        action = 'http://music.163.com/api/playlist/detail?id=' + str(pid)
        res_data = self.http_request('GET', action)
        return res_data

    def song_detail(self, sid):
        action = 'http://music.163.com/api/song/detail?ids=%5B' + str(sid) + '%5D'
        res_data = self.http_request('GET', action)
        return res_data['songs'][0]

    def get_lyric_by_musicid(self, mid):
        # 此API必须使用POST方式才能正确返回，否则提示“illegal request”
        url = 'http://music.163.com/api/song/lyric?id=' + str(mid) + '&lv=1&kv=1&tv=-1'
        return self.http_request('POST', url)

    def clean_lyric(self, lrc):
        r = []
        is_empty = False
        for line in lrc.strip().split('\n'):
            line = line.strip()
            if not is_empty:
                r.append(line)
                if line == '':
                    is_empty = True
            else:
                if line != '':
                    r.append(line)
                    is_empty = False
        return '\n'.join(r)


class Playlist():
    def __init__(self, pid):
        self.id = pid
        self.playlist_name = ''
        self.song_name = []
        self.song_id = []
        self.song_img = []
        self.song_lrc = []

    def get_lrc(self, random_line):
        pid = self.id
        netease = NetEase()
        playlist = netease.user_playlist(pid)
        self.playlist_name = playlist['result']['name']
        self.playlist = playlist['result']['tracks']
        print(u'歌单名：《%s》，歌曲数：%d' % (self.playlist_name, len(self.playlist)))
        
        for song in self.playlist:
            self.song_name.append(song["name"])
            self.song_id.append(song["id"])
            self.song_img.append(song["album"]["blurPicUrl"])

        for sid in self.song_id:
            lrc = netease.get_lyric_by_musicid(sid)
            if 'lrc' in lrc and 'lyric' in lrc['lrc'] and lrc['lrc']['lyric'] != '':
                lrc = lrc['lrc']['lyric']
                pat = re.compile(r'\[.*\]')
                lrc = re.sub(pat, "", lrc)
                lrc = lrc.strip()
            else:
                lrc = u'纯音乐，无歌词'

            lrc = netease.clean_lyric(lrc)
            if random_line != 0:
                lrc_arr = lrc.split('\n')
                n = len(lrc_arr)
                if n > random_line:
                    i = random.randint(0, n - random_line)
                    lrc = '\n'.join(lrc_arr[i:i+random_line])

            self.song_lrc.append(lrc)

    def create_img(self, pic_style):
        img = Img(self.playlist_name + '/')
        save_func = None
        if pic_style == 1:
            save_func = img.save
        elif pic_style == 2:
            save_func = img.save2
        elif pic_style == 3:
            save_func = img.save3

        for s_name, s_lrc, s_img in zip(self.song_name, self.song_lrc, self.song_img):
            save_func(s_name, s_lrc, s_img)


class Song(object):
    def __init__(self, uid):
        self.id = uid
        self.song_lrc = ''
        self.song_id = uid
        self.song_name = ''
        self.song_img = ''

    def get_lrc(self, random_line):
        uid = self.id
        netease = NetEase()
        song = netease.song_detail(uid)
        self.song_name = song['name'].strip()
        self.song_id = uid
        self.song_img = song["album"]["blurPicUrl"]

        lrc = netease.get_lyric_by_musicid(uid)
        if 'lrc' in lrc and 'lyric' in lrc['lrc'] and lrc['lrc']['lyric'] != '':
            lrc = lrc['lrc']['lyric']
            pat = re.compile(r'\[.*\]')
            lrc = re.sub(pat, "", lrc)
            lrc = lrc.strip()
        else:
            lrc = u'纯音乐，无歌词'
        self.song_lrc = netease.clean_lyric(lrc)

        if random_line != 0:
            lrc = self.song_lrc.split('\n')
            n = len(lrc)
            if n > random_line:
                i = random.randint(0, n - random_line)
                self.song_lrc = '\n'.join(lrc[i:i+random_line])

    def show_lrc(self):
        lrcs = self.song_lrc.split('\n')
        n = len(lrcs)
        for i in range(n):
            print(('%'+str(len(str(n)))+'d %s') % (i+1, lrcs[i]))

    def create_img(self, pic_style):
        if pic_style == 1:
            Img().save(self.song_name, self.song_lrc, self.song_img)
        elif pic_style == 2:
            Img().save2(self.song_name, self.song_lrc, self.song_img)
        elif pic_style == 3:
            Img().save3(self.song_name, self.song_lrc, self.song_img)


class Img():

    def __init__(self, save_dir=None):
        self.save_dir = save_dir

        self.font_family = 'res/STHeiti_Light.ttc'
        self.font_size = 30 # 字体大小
        self.line_space = 30 # 行间隔大小
        self.word_space = 5
        self.share_img_width = 640
        self.padding = 50
        self.song_name_space = 50
        self.banner_space = 60
        self.text_color = '#767676'
        # self.netease_banner = unicode('来自网易云音乐•歌词分享', "utf-8")
        self.netease_banner = u'来自网易云音乐•歌词分享'
        self.netease_banner_color = '#D3D7D9'
        self.netease_banner_size = 20
        self.netease_icon = 'res/netease_icon.png'
        self.icon_width = 25

        self.style2_margin = 50
        self.style2_padding = 30
        self.style2_line_width = 2
        self.style2_quote_width = 30
        self.quote_icon = 'res/quote.png'

        if self.save_dir is not None:
            try:
                os.mkdir(self.save_dir)
            except:
                pass

    def save(self, name, lrc, img_url):
        lyric_font = ImageFont.truetype(self.font_family, self.font_size)
        banner_font = ImageFont.truetype(self.font_family, self.netease_banner_size)
        lyric_w, lyric_h = ImageDraw.Draw(Image.new(mode='RGB', size=(1, 1))).textsize(lrc, font=lyric_font, spacing=self.line_space)

        padding = self.padding
        w = self.share_img_width
        
        album_img = None
        if img_url.startswith('http'):
            raw_img = requests.get(img_url)
            album_img = Image.open(BytesIO(raw_img.content))
        else:
            album_img = Image.open(img_url)
        
        iw, ih = album_img.size
        album_h = ih *  w // iw

        h = album_h + padding + lyric_h + self.song_name_space + \
            self.font_size + self.banner_space + self.netease_banner_size + padding

        resized_album = album_img.resize((w, album_h), resample=3)
        icon = Image.open(self.netease_icon).resize((self.icon_width, self.icon_width), resample=3)

        out_img = Image.new(mode='RGB', size=(w, h), color=(255, 255, 255))
        draw = ImageDraw.Draw(out_img)

        # 添加封面
        out_img.paste(resized_album, (0, 0))
        
        # 添加文字
        draw.text((padding, album_h + padding), lrc, font=lyric_font, fill=self.text_color, spacing=self.line_space)
        
        # Python中字符串类型分为byte string 和 unicode string两种，'——'为中文标点byte string，需转换为unicode string
        y_song_name = album_h + padding + lyric_h + self.song_name_space
        # song_name = unicode('—— 「', "utf-8") + name + unicode('」', "utf-8")
        song_name = u'—— 「' + name + u'」'
        sw, sh = draw.textsize(song_name, font=lyric_font)
        draw.text((w - padding - sw, y_song_name), song_name, font=lyric_font, fill=self.text_color)
        
        # 添加网易标签
        y_netease_banner = h - padding - self.netease_banner_size
        out_img.paste(icon, (padding, y_netease_banner - 2))
        draw.text((padding + self.icon_width + 5, y_netease_banner), self.netease_banner, font=banner_font, fill=self.netease_banner_color)
        
        img_save_path = ''
        if self.save_dir is not None:
            img_save_path = self.save_dir
        out_img.save(img_save_path + name + '.png')

    def save2(self, name, lrc, img_url):
        lyric_font = ImageFont.truetype(self.font_family, self.font_size)
        banner_font = ImageFont.truetype(self.font_family, self.netease_banner_size)
        lyric_w, lyric_h = ImageDraw.Draw(Image.new(mode='RGB', size=(1, 1))).textsize(lrc, font=lyric_font, spacing=self.line_space)

        margin = self.style2_margin
        padding = self.style2_padding
        w = self.share_img_width
        h = margin + padding + lyric_h + self.song_name_space + \
            self.font_size + self.banner_space + self.netease_banner_size + padding + margin

        icon = Image.open(self.netease_icon).resize((self.icon_width, self.icon_width), resample=3)
        quote = Image.open(self.quote_icon).resize((self.style2_quote_width, self.style2_quote_width), resample=3)

        out_img = Image.new(mode='RGB', size=(w, h), color=(255, 255, 255))
        draw = ImageDraw.Draw(out_img)

        def draw_rectangle(draw, rect, width):
            for i in range(width):
                draw.rectangle((rect[0] + i, rect[1] + i, rect[2] - i, rect[3] - i), outline=self.netease_banner_color)


        # 画边框
        rect_h = padding + lyric_h + self.song_name_space + self.font_size + self.banner_space
        draw_rectangle(draw, (margin, margin, w - margin, margin + rect_h ), 2)
        out_img.paste(quote, (margin - self.style2_quote_width // 2, margin + self.style2_quote_width // 2))
        quote = quote.rotate(180)
        out_img.paste(quote, (w - margin - self.style2_quote_width // 2, margin + rect_h - self.style2_quote_width - self.style2_quote_width // 2))
        
        # 添加文字
        draw.text((margin + padding, margin + padding), lrc, font=lyric_font, fill=self.text_color, spacing=self.line_space)
        
        y_song_name = margin + padding + lyric_h + self.song_name_space
        # song_name = unicode('—— 「', "utf-8") + name + unicode('」', "utf-8")
        song_name = u'—— 「' + name + u'」'
        sw, sh = draw.textsize(song_name, font=lyric_font)
        draw.text((w - margin - padding - sw, y_song_name), song_name, font=lyric_font, fill=self.text_color)
        
        # 添加网易标签
        y_netease_banner = h - padding - self.netease_banner_size
        out_img.paste(icon, (margin, y_netease_banner - 2))
        draw.text((margin + self.icon_width + 5, y_netease_banner), self.netease_banner, font=banner_font, fill=self.netease_banner_color)
        
        img_save_path = ''
        if self.save_dir is not None:
            img_save_path = self.save_dir
        out_img.save(img_save_path + name + '.png')

    def save3(self, name, lrc, img_url):
        lyric_font = ImageFont.truetype(self.font_family, self.font_size)
        banner_font = ImageFont.truetype(self.font_family, self.netease_banner_size)
        lyric_w, lyric_h = ImageDraw.Draw(Image.new(mode='RGB', size=(1, 1))).textsize(lrc, font=lyric_font, spacing=self.line_space)

        margin = self.style2_margin
        padding = self.style2_padding
        w = self.share_img_width

        album_img = None
        if img_url.startswith('http'):
            raw_img = requests.get(img_url)
            album_img = Image.open(BytesIO(raw_img.content))
        else:
            album_img = Image.open(img_url)

        iw, ih = album_img.size
        album_h = ih *  w // iw

        h = album_h + margin + padding + lyric_h + self.song_name_space + \
            self.font_size + self.banner_space + self.netease_banner_size + padding + margin

        resized_album = album_img.resize((w, album_h), resample=3)
        icon = Image.open(self.netease_icon).resize((self.icon_width, self.icon_width), resample=3)
        quote = Image.open(self.quote_icon).resize((self.style2_quote_width, self.style2_quote_width), resample=3)

        out_img = Image.new(mode='RGB', size=(w, h), color=(255, 255, 255))
        draw = ImageDraw.Draw(out_img)

        def draw_rectangle(draw, rect, width):
            for i in range(width):
                draw.rectangle((rect[0] + i, rect[1] + i, rect[2] - i, rect[3] - i), outline=self.netease_banner_color)

        # 添加封面
        out_img.paste(resized_album, (0, 0))

        # 画边框
        rect_h = padding + lyric_h + self.song_name_space + self.font_size + self.banner_space
        draw_rectangle(draw, (margin, album_h + margin, w - margin, album_h + margin + rect_h ), 2)
        out_img.paste(quote, (margin - self.style2_quote_width // 2, album_h + margin + self.style2_quote_width // 2))
        quote = quote.rotate(180)
        out_img.paste(quote, (w - margin - self.style2_quote_width // 2, album_h + margin + rect_h - self.style2_quote_width - self.style2_quote_width // 2))
        
        # 添加文字
        draw.text((margin + padding, album_h + margin + padding), lrc, font=lyric_font, fill=self.text_color, spacing=self.line_space)
        
        y_song_name = album_h + margin + padding + lyric_h + self.song_name_space
        # song_name = unicode('—— 「', "utf-8") + name + unicode('」', "utf-8")
        song_name = u'—— 「' + name + u'」'
        sw, sh = draw.textsize(song_name, font=lyric_font)
        draw.text((w - margin - padding - sw, y_song_name), song_name, font=lyric_font, fill=self.text_color)
        
        # 添加网易标签
        y_netease_banner = h - padding - self.netease_banner_size
        out_img.paste(icon, (margin, y_netease_banner - 2))
        draw.text((margin + self.icon_width + 5, y_netease_banner), self.netease_banner, font=banner_font, fill=self.netease_banner_color)
        
        img_save_path = ''
        if self.save_dir is not None:
            img_save_path = self.save_dir
        out_img.save(img_save_path + name + '.png')

def unicode_str(s):
    try:
        t = unicode(s, 'utf-8')
        return t
    except:
        return s

def main():
    parser = optparse.OptionParser('usage: [--sid <song id> | --pid <playlist id>] -t <pic style> -r <random_line>\n\t-i <your image> -w <some text> -n <name> -l -p <line_range>')
    parser.add_option('--sid', dest='sid', type='string', help='song id')
    parser.add_option('--pid', dest='pid', type='string', help='playlist id')
    parser.add_option('-t', dest='pic_style', type='int', help='1: has album image, 2: lyric only, 3: combine 1 & 2')
    parser.add_option('-r', dest='random_line', type='int', help='number of random lyric lines')
    parser.add_option('-p', dest='line_range', type='string', help='range of lyric lines')
    parser.add_option('-i', dest='img_file', type='string', help='your own image')
    parser.add_option('-w', dest='text', type='string', help='some text')
    parser.add_option('-n', dest='name', type='string', help='name')
    parser.add_option('-l', dest="show_lyrics", action="store_true", default=False, 
                        help="only show the lyrics with line number to stdout")
    (options, args) = parser.parse_args()
    sid = options.sid
    pid = options.pid
    pic_style = options.pic_style
    line_range = options.line_range
    random_line = options.random_line
    img_file = options.img_file
    text = options.text
    name = options.name

    if pic_style is None:
        pic_style = 1
    if random_line is None:
        random_line = 0

    if sid is not None:
        song = Song(sid)
        song.get_lrc(random_line)

        if options.show_lyrics:
            song.show_lrc()
        else:
            if line_range is not None:
                lrcs = song.song_lrc.split('\n')
                tmp_lrcs = []
                for i in line_range.split(','):
                    if '-' in i:
                        a, b = i.split('-')
                        tmp_lrcs += lrcs[int(a)-1:int(b)]
                    else:
                        tmp_lrcs.append(lrcs[int(i)-1])
                song.song_lrc = '\n'.join(tmp_lrcs)

            song.create_img(pic_style)

    elif pid is not None:
        playlist = Playlist(pid)
        playlist.get_lrc(random_line)
        playlist.create_img(pic_style)
    elif text is not None:
        text = '\n'.join([line.strip() for line in text.replace('\\n','\n').split('\n')])
        text = unicode_str(text)
        
        if img_file is None:
            img_file = 'res/dog.png'
        if name is None:
            name = u'Anonymous'
        else:
            name = unicode_str(name)

        save_func = None
        if pic_style == 1:
            save_func = Img().save
        elif pic_style == 2:
            save_func = Img().save2
        elif pic_style == 3:
            save_func = Img().save3
        save_func(name, text, img_file)

if __name__ == '__main__':
    main()