transforming/adddescriptiontransform.py

from typing import Any

from pytube import YouTube

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

class AddDescriptionTransform(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_description_params = {
 "channel_name": video.channel_name,
 "url": video.url,
 "title": video.title,
 "description": self._get_video_description(yt),
 "transcription": video.transcription,
 "segments": video.segments
 }
 
 return YoutubeVideo(**video_With_description_params)
 
 def _get_video_description(self, yt: Any) -> str:
 return str(yt.description)