유튜브 플레이리스트에서 고음질 음원 추출 및 MP3 변환 자동화

파이썬 쓸줄 안다고 가정

Youtube 플레이리스트 생성하고 해당 플레이리스트 공개로 전환해, 링크를 넣으면 다운받으실 수 있습니다.

 

전체코드

import os
import re
import subprocess
from pathlib import Path
from pytubefix import Playlist, YouTube

# 진행 상태 출력
def on_progress(stream, chunk, bytes_remaining):
    total_size = stream.filesize
    bytes_downloaded = total_size - bytes_remaining
    percentage = (bytes_downloaded / total_size) * 100
    print(f"\rDownloading... {percentage:.2f}%", end="")

# 최상의 오디오 itag 찾기
def get_best_audio(yt):
    best_stream = max(
        yt.streams.filter(only_audio=True),
        key=lambda s: int(s.abr.replace('kbps', '')) if s.abr else 0,
        default=None
    )
    return best_stream.itag if best_stream else None

# 파일 이름 정리
def clean_filename(title):
    return re.sub(r'[\\/*?:"<>|]', '', title)

# 오디오 다운로드
def download_audio(url, output_dir, index=None, total=None):
    try:
        yt = YouTube(url, on_progress_callback=on_progress)
        title = clean_filename(yt.title)

        if index and total:
            print(f"\n▶ ({index}/{total}) {title} 다운로드 중...")
        else:
            print(f"\n▶ {title} 다운로드 중...")

        itag = get_best_audio(yt)
        if itag:
            yt.streams.get_by_itag(itag).download(
                output_path=output_dir,
                filename=f"{title}.m4a"
            )
            print(f"\n✅ {title} 다운로드 완료!")
        else:
            print(f"\n❌ 다운로드 실패 (오디오 없음)")
    except Exception as e:
        print(f"\n⚠️ 오류 발생: {e}")

# m4a → mp3 변환 및 원본 삭제
def convert_all_m4a_to_mp3(directory):
    for m4a_file in Path(directory).glob("*.m4a"):
        mp3_file = m4a_file.with_suffix(".mp3")

        if mp3_file.exists():
            print(f"\n⏭ 이미 존재: {mp3_file.name}")
            m4a_file.unlink()
            print(f"🗑 삭제 완료: {m4a_file.name}")
            continue

        print(f"\n🎧 변환 시작: {m4a_file.name}")
        command = [
            "ffmpeg", "-hide_banner", "-y", "-loglevel", "error",
            "-i", str(m4a_file), "-vn", "-ar", "44100", "-ac", "2", "-ab", "192k", "-f", "mp3", str(mp3_file)
        ]
        try:
            subprocess.run(command, check=True)
            print(f"✅ 변환 완료: {mp3_file.name}")
            m4a_file.unlink()
            print(f"🗑 삭제 완료: {m4a_file.name}")
        except subprocess.CalledProcessError:
            print(f"❌ 변환 실패: {m4a_file.name}")

# 플레이리스트 다운로드
def download_playlist(url, output_dir):
    pl = Playlist(url)
    print(f"\n🎵 플레이리스트: {pl.title}")
    print(f"📁 총 {len(pl.video_urls)}개 다운로드 시작!\n")

    for idx, video_url in enumerate(pl.video_urls, start=1):
        download_audio(video_url, output_dir, idx, len(pl.video_urls))

# 실행 함수
def main():
    download_path = Path(input("음악을 저장할 폴더명을 입력: ").strip())
    download_path.mkdir(parents=True, exist_ok=True)

    print("옵션: [1] 플레이리스트 | [2] 단일 영상")
    mode = input("선택 (1/2): ").strip()

    if mode == "1":
        url = input("🎵 플레이리스트 URL 입력: ").strip()
        download_playlist(url, download_path)
    elif mode == "2":
        url = input("🎵 영상 URL 입력: ").strip()
        download_audio(url, download_path)
    else:
        print("❌ 잘못된 입력입니다.")
        return

    convert_all_m4a_to_mp3(download_path)
    print("\n🎉 모든 작업이 완료되었습니다!")

if __name__ == "__main__":
    main()