파이썬으로 mp3 다운로드 프로그램 만들기

https://github.com/Hydragon516/Bugs-Music-Downloader

GitHub - Hydragon516/Bugs-Music-Downloader

 

 

벅스(Bugs) 뮤직 크롤러를 만들어 보았다. 벅스에서 직접 음원을 다운로드할 수 있는 프로그램은 당연히 아니다. 약간의 꼼수로 벅스에서 음원 정보를 크롤링하고 해당 정보의 음원을 유튜브에서 추출하는 방식으로 동작한다. 코드는 아래와 같다.

 

from PyQt5.QtCore import pyqtSignal, pyqtSlot, QThread
from PyQt5.QtWidgets import QLabel, QListWidget, QLineEdit, QDialog, QPushButton, QHBoxLayout, QVBoxLayout, QApplication
from bs4 import BeautifulSoup
import requests
import re
import youtube_dl
import eyed3
import os
import shutil
from selenium import webdriver
from urllib import request

search_title = ""

class MyMainGUI(QDialog):
    def __init__(self, parent=None):
        super().__init__(parent)

        self.search_button = QPushButton("음악 검색")
        self.search_input = QLineEdit(self)
        self.music_list = QListWidget(self)

        self.status_label = QLabel("", self)

        self.youtube_button = QPushButton("음원 다운로드")

        hbox = QHBoxLayout()
        hbox.addStretch(0)
        hbox.addWidget(self.search_input)
        hbox.addWidget(self.search_button)
        hbox.addStretch(0)

        hbox2 = QHBoxLayout()
        hbox2.addWidget(self.music_list)

        hbox3 = QHBoxLayout()
        hbox3.addWidget(self.youtube_button)

        vbox = QVBoxLayout()
        vbox.addStretch(1)
        vbox.addLayout(hbox)
        vbox.addStretch(1)
        vbox.addLayout(hbox2)
        vbox.addStretch(1)
        vbox.addLayout(hbox3)
        vbox.addStretch(1)
        vbox.addWidget(self.status_label)

        self.setLayout(vbox)

        self.setWindowTitle('Bugs Downloader')
        self.setGeometry(300, 300, 500, 200)


class MyMain(MyMainGUI):
    add_sec_signal = pyqtSignal()
    send_instance_singal = pyqtSignal("PyQt_PyObject")

    def __init__(self, parent=None):
        super().__init__(parent)

        self.search_button.clicked.connect(self.search)
        self.youtube_button.clicked.connect(self.download)

        self.search_input.textChanged[str].connect(self.title_update)
        self.music_list.itemClicked.connect(self.chkItemClicked)

        self.th_search = searcher(parent=self)
        self.th_search.updated_list.connect(self.list_update)
        self.th_search.updated_label.connect(self.status_update)

        self.th_download = downloadr(parent=self)
        self.th_download.updated_label.connect(self.status_update)

        self.show()
    
    def title_update(self, input):
        global search_title
        search_title = input
    
    def chkItemClicked(self) :
        global keyword
        keyword = self.music_list.currentItem().text() 

    @pyqtSlot()
    def search(self):
        self.music_list.clear()
        self.th_search.start()

    @pyqtSlot()
    def download(self):
        self.th_download.start()

    @pyqtSlot(str)
    def list_update(self, msg):
        self.music_list.addItem(msg)
    
    @pyqtSlot(str)
    def status_update(self, msg):
        self.status_label.setText(msg)


class searcher(QThread):
    updated_list = pyqtSignal(str)
    updated_label = pyqtSignal(str)

    def __init__(self, parent=None):
        super().__init__()
        self.main = parent

    def __del__(self):
        self.wait()

    def run(self):

        global search_title
        global keyword

        title_list = []
        artist_list = []
        album_list = []
        
        if search_title != "":
            self.updated_label.emit("서버에 접속하는 중...")

            options = webdriver.ChromeOptions()
            options.add_argument('--headless')
            options.add_argument("--disable-gpu")
            options.add_experimental_option('excludeSwitches', ['enable-logging'])
            options.add_argument('--log-level=3')

            driver = webdriver.Chrome('chromedriver', options=options)
            driver.implicitly_wait(5)

            driver.get(url='https://music.bugs.co.kr/search/track?q=' + search_title)

            self.updated_label.emit("제목 정보를 불러오는 중...")

            for indx in range(30):
                try:
                    target = driver.find_element_by_xpath('//*[@id="DEFAULT0"]/table/tbody/tr[{}]/th/p/a'.format(indx + 1))
                    title_list.append(target.get_attribute("title"))
                except:
                    break
            
            self.updated_label.emit("아티스트 정보를 불러오는 중...")
            
            for indx in range(30):
                try:
                    target = driver.find_element_by_xpath('//*[@id="DEFAULT0"]/table/tbody/tr[{}]/td[4]/p/a'.format(indx + 1))
                    artist_list.append(target.get_attribute("title"))
                except:
                    break
            
            self.updated_label.emit("앨범 정보를 불러오는 중...")
            
            for indx in range(30):
                try:
                    target = driver.find_element_by_xpath('//*[@id="DEFAULT0"]/table/tbody/tr[{}]/td[5]/a'.format(indx + 1))
                    album_list.append(target.get_attribute("title"))
                except:
                    break

            try:
                for i in range(len(artist_list)):
                    self.updated_list.emit("%s // %s // %s // %s" % (str(i + 1), title_list[i], artist_list[i], album_list[i]))
            except:
                pass

            driver.close()

            self.updated_label.emit("불러오기 완료!")
        
        else:
            self.updated_label.emit("검색어를 입력하세요")


class downloadr(QThread):
    updated_label = pyqtSignal(str)

    def __init__(self, parent=None):
        super().__init__()
        self.main = parent

    def __del__(self):
        self.wait()

    def run(self):
        global keyword
        global search_title

        self.updated_label.emit("음원 파일을 읽는 중 ...")
        
        target_index = keyword.split(" // ")[0]
        target_title = keyword.split(" // ")[1]
        target_artist = keyword.split(" // ")[2]
        target_album = keyword.split(" // ")[3]

        options = webdriver.ChromeOptions()
        options.add_argument('--headless')
        options.add_argument("--disable-gpu")
        options.add_experimental_option('excludeSwitches', ['enable-logging'])
        options.add_argument('--log-level=3')


        driver = webdriver.Chrome('chromedriver', options=options)
        driver.implicitly_wait(5)

        driver.get(url='https://music.bugs.co.kr/search/track?q=' + search_title)
        target_music_button = driver.find_element_by_xpath('//*[@id="DEFAULT0"]/table/tbody/tr[{}]/td[3]/a'.format(target_index))
        target_music_button.click()
        
        lyrics_url = driver.current_url

        driver.close()

        html = requests.get(lyrics_url)
        soup = BeautifulSoup(html.text, 'html.parser')

        cover = soup.find_all('div', {'class':'photos'})
        cover_link = re.search('src="(.*)"/>', str(cover)).group(1)
        request.urlretrieve(cover_link, "cover.jpg")

        lyric = soup.find_all('div', {'class':'lyricsContainer'})
        lines = str(lyric).split("\n")

        line_state = False
        lyrics = []

        for item in lines:
            if line_state == True:
                if "</xmp></p>" not in item:
                    lyrics.append(item)
            
            if "<p><xmp>" in item:
                lyrics.append(item.replace("<p><xmp>", ""))
                line_state = True
            
            if "</xmp></p>" in item:
                lyrics.append(item.replace("</xmp></p>", ""))
                break

        with open("lyric.txt", 'w') as f:
            for row in lyrics:
                f.write(row)

        f.close()

        surch_keyword = target_title + " " + target_artist + " 음원"

        new_name = target_title + "_" + target_artist + ".mp3"

        url_list = []
        url = 'https://www.youtube.com/results?search_query={}'.format(surch_keyword)

        options = webdriver.ChromeOptions()
        options.add_argument('--headless')
        options.add_argument('--window-size=1024,768')
        options.add_argument("--disable-gpu")

        driver = webdriver.Chrome('./chromedriver.exe', options=options)
        
        driver.get(url)
        soup = BeautifulSoup(driver.page_source, 'html.parser')
        driver.close()

        video_url = soup.select('a#video-title')

        for i in video_url:
            url_list.append('{}{}'.format('https://www.youtube.com',i.get('href')))
        
        ydl_opts = {
            'format': 'bestaudio/best',
            'postprocessors': [{
                'key': 'FFmpegExtractAudio',
                'preferredcodec': 'mp3',
                'preferredquality': '192',
            }],
        }

        self.updated_label.emit("음원 다운로드 중 입니다...")

        with youtube_dl.YoutubeDL(ydl_opts) as ydl:
            ydl.download([url_list[0]])

        self.updated_label.emit("파일 변환 중 입니다...")

        files = os.listdir("./")

        for file in files:
            if ".mp3" in file:
                os.rename(file, new_name)
        
        OpenLyircsFile = open("lyric.txt", 'r') 
        ReadLyirsFile = OpenLyircsFile.read() 

        audiofile = eyed3.load("./" + new_name)
        audiofile.initTag()
        audiofile.tag.artist = target_artist
        audiofile.tag.title = target_title
        audiofile.tag.album = target_album
        audiofile.tag.lyrics.set(ReadLyirsFile)
        audiofile.tag.images.set(3, open('cover.jpg','rb').read(), 'image/jpeg')
        audiofile.tag.save(version=eyed3.id3.ID3_V2_3)

        if not os.path.exists("./변환된 파일"):
            os.makedirs("./변환된 파일")

        shutil.move(new_name, "./변환된 파일/" + new_name)

        self.updated_label.emit("변환 완료!")


if __name__ == "__main__":
    import sys

    app = QApplication(sys.argv)
    form = MyMain()
    app.exec_()

 

코드가 어렵지는 않은데 좀 지저분하다. 사실 벅스가 반응형 홈페이지가 아니라 BeautifulSoup만으로도 크롤링이 가능할지는 모르지만 앨범 커버와 가사 정보를 다운로드하기 위해 selenium을 섞어 사용했다. 참고로 eyed3는 mp3 메타 정보를 수정할 수 있는 파이썬 라이브러리이다.

 

프로그램은 아래 깃헙 링크에서 다운 가능하다.

반응형

https://github.com/Hydragon516/Bugs-Music-Downloader/releases

Releases · Hydragon516/Bugs-Music-Downloader

 

사용방법은 다음과 같다.

• 압축을 해제하고 본인의 크롬 버전과 동일한 최신 크롬 드라이버를 다운로드한다.
• 다운로드한 드라이버는 압축 해제한 폴더에 넣어두면 된다.
• bugs.exe를 실행한다. (실행 시 나오는 검은 터미널 창은 무시)