utils/video.py

import subprocess, os, requests, json
from utils.log import info
from utils.subtitles import subs

def Popen(cmd: list) -> str:
 """Run a command and return the output as a string

 Args:
 cmd (list): The command to run

 Returns:
 str: The output of the command
 """
 return subprocess.Popen(cmd, shell=False, stdout=subprocess.PIPE).stdout.read().strip().decode('utf-8')

class video:
 def __init__(self,id, path):
 self.path = path
 self.url = "https://youtu.be/" + id
 self.video_id = id
 
 # check if directory exists
 if not os.path.exists(self.path.split("/")[-1]):
 os.mkdir(self.path.split("/")[-1])
 
 def download(self):
 if os.path.exists(f"{self.path}.webm"):
 info(f"{self.path}.webm already exists, skipping download")
 return
 info(f"Downloading {self.url}")
 # (
 # Popen(
 # ["yt-dlp", self.url, "-o", self.path ]
 # )
 # )
 os.system(f"yt-dlp {self.url} -o {self.path}")
 
 def getframe(self, timestamp, out=os.curdir):
 filename = out
 if os.path.exists(filename):
 info(f"{filename} already exists, skipping frame")
 return
 
 info(f"Getting frame at {timestamp}")
 (
 Popen(
 [
 "ffmpeg", 
 "-hide_banner",
 "-loglevel", "panic",
 "-ss", timestamp, 
 "-i", f"{self.path}.webm", 
 "-vframes", "1", 
 filename
 ]
 )
 )
 
 def getAudio(self, out="out.mp3"):
 info("Getting audio...")
 (
 Popen(
 [
 "ffmpeg", 
 "-hide_banner",
 "-loglevel", "panic",
 "-i", f"{self.path}.webm", 
 "-vn", 
 "-ar", "44100", 
 "-ac", "2", 
 "-ab", "192K", 
 "-f", "mp3", 
 out
 ]
 )
 )
 
 def getChapters(self, endpoint: str) -> list:
 """return the chapters of the video

 Args:
 endpoint (str): endpoint to communicate to get chapters
 yt.lemnoslife.com recommended
 Returns:
 list: chapters
 """
 res = requests.get(f"{endpoint}")
 chapters = res.json()['items'][0]['chapters']['chapters']
 return chapters
 
 def getSubtitles(self):
 """return the raw subtitles

 Returns:
 list: subtitles directly from youtube
 """
 return json.loads(
 json.dumps(
 subs(self.video_id)
 .getSubsRaw()
 )
 )