Python pytube youtube 영상 다운로드

pytube로 youtube 영상을 다운로드하는 예제 코드이다.

다운로드 시 프로그레스바가 출력된다.

from pytube import YouTube
import sys

def progress_function(chunk, file_handle, bytes_remaining):
    global filesize
    current = ((filesize - bytes_remaining)/filesize)
    percent = ('{0:.1f}').format(current*100)
    progress = int(50*current)
    status = '█' * progress + '-' * (50 - progress)
    sys.stdout.write(' ↳ |{bar}| {percent}%\r'.format(bar=status, percent=percent))
    sys.stdout.flush()

DOWNLOAD_FOLDER = "./"

url = f"https://www.youtube.com/watch?v=XXXXXX"

yt = YouTube(url, on_progress_callback=progress_function)

video = yt.streams.filter(progressive=True, file_extension='mp4').get_highest_resolution()

filesize = video.filesize
print('FileSize : ' + str(round(video.filesize/(1024*1024))) + 'MB')

print("title :", yt.title)
print("length :", yt.length)
print("author :", yt.author)
print("publish date :", yt.publish_date)
print("views :", yt.views)
print("keywords :", yt.keywords)
print("description :", yt.description)

video.download(DOWNLOAD_FOLDER)
print("\r\n")