transforming/addtitletransform.py

from typing import Any

from pytube import YouTube

from video import YoutubeVideo
from utils import accepts_types
from transforming.transform import Transform

class AddTitleTransform(Transform):
    """
    Transform a Video object using PyTube. Adds title to YouTube video DTO. 
    It's a concrete Transform.
    """
       
    @accepts_types(YoutubeVideo)
    def apply(self, video: YoutubeVideo) -> YoutubeVideo:
        yt = YouTube(video.url)
        
        video_With_title_params = {
            "channel_name": video.channel_name,
            "url": video.url,
            "title": self._get_video_title(yt),
            "description": video.description,
            "transcription": video.transcription,
            "segments": video.segments
        }
        
        return YoutubeVideo(**video_With_title_params)
    
    def _get_video_title(self, yt: Any) -> str:
            return str(yt.title)