[Python] 유튜브 영상을 PC에 다운받은 후 영상파일을 90도 rotation 하기 (youtube_dl + moviepy)

import youtube_dl, operator, os, glob
from moviepy.editor import *


# 유튜브 영상을 다운받는 함수입니다.
def yt_download(url): 
    ydl_opts = {'outtmpl': '%(title)s.%(ext)s', 
                'format':' bestvideo+bestaudio'
            }
    with youtube_dl.YoutubeDL(ydl_opts) as ydl:
        ydl.download([url])
        info = ydl.extract_info(url, download=False) # 유튜브에 저장된 파일명을 받아옵니다.
        filename = ydl.prepare_filename(info)
        print(filename)
    return filename


# 확장자를 확인하고, 특수문자를 제거하는 함수입니다.
def check_file(filename): 
    if not os.path.isfile(filename):
        name = ''.join(filename.split('.')[:-1])
        filename = glob.glob(name + '.*')[0]
    name, ext = os.path.splitext(filename)
    name_edit = ''.join([x for x in name if x.isalnum() or x.isspace() or x == '_' or x == '-'])
    new_filename = name_edit + ext
    os.rename(filename, new_filename)
    return new_filename


# 영상을 rotation 하는 함수입니다.
def rotate_video(filename): 
    clip = VideoFileClip(filename)
    clip = clip.rotate(270, resample='bicubic') # 270도 또는 90도 rotation 합니다.
    name, ext = os.path.splitext(filename)
    print(ext)
    clip.write_videofile(name + '_rotate.mp4') # MP4로 설정해야 오류가 가장 적네요.
    return


url = 'URL을 입력하세요'
filename = yt_download(url)
filename = check_file(filename)
rotate_video(filename)

안녕하세요?

유튜브 영상 중에 모바일용으로 회전된 영상을 PC용으로 다시 회전시키는 스크립트에요 ^^

youtube_dl과 moviepy를 사용했구요.

이론적으로는 간단하지만 다음과 같은 문제가 있더군요 ㄷㄷ

1. 유튜브 서버에 저장된 파일과 실제 다운받은 파일의 확장자가 불일치한 경우가 있어요.

youtube_dl에서 merge할 때 확장자를 MKV로 바꿔버리는 경우가 있더군요.

(pytube는 어떨지 모르겠네요 ^^)

https://stackoverflow.com/questions/59987612/python-youtube-dl-it-is-possible-to-get-merging-files-extension

위와 같은 방식으로 해결한 분이 계시던데, 저는 가능하면 어떠한 확장자에 대해서도 대처할 수 있는 방법으로 구현했네요 :)

2. 특수문자가 파일명에 포함되어 있으면 moviepy에서 에러가 발생하는 경우가 있어요 ㅠㅠ

3. interpolation 방식은 bicubic이나 bilinear가 무난하던데 아무래도 완벽하지는 않네요~

일단 MP4, WEBM, MKV파일의 경우를 테스트해봤는데

이외에 다른 경우가 또 있을지도 모르겠네요 ㅜㅜ

그럼 남은 연휴 즐겁게 보내세요!

제가 업무 때문에 경황이 없어서 당분간은 글을 작성하지 못할 것 같네요 ㅠㅠ

항상 감사드려요 :)

+) 만약 youtube_dl이 작동하지 않는다면 pytube로 해결하시면 되어요 ^^