video.py

from typing import Dict, List, Optional, Tuple

from pydantic import BaseModel

class YoutubeVideo(BaseModel):
 """This class represent a YouTube video entry
 """
 channel_name: str
 url: str
 title: Optional[str]
 description: Optional[str]
 transcription: Optional[str]
 segments: Optional[List[Dict]] = None
 
 def to_tuple(self) -> Tuple:
 """Convert TranscribedVideo object to a tuple of the type:
 (channel_name, url, title, description, transcription, segments).
 """
 return (self.channel_name, self.url, self.title,
 self.description, self.transcription, self.segments)