bulk-downloader-for-reddit/src/downloaders/vreddit.py

import os
import subprocess

from src.downloaders.downloaderUtils import getFile, getExtension

from src.errors import FileNameTooLong
from src.utils import GLOBAL
from src.utils import printToFile as print

class VReddit:
    def __init__(self,directory,post):
        extension = ".mp4"
        if not os.path.exists(directory): os.makedirs(directory)

        filename = GLOBAL.config['filename'].format(**post)+extension
        shortFilename = post['POSTID']+extension

        try:
            FNULL = open(os.devnull, 'w')
            subprocess.call("ffmpeg", stdout=FNULL, stderr=subprocess.STDOUT)
        except:
            getFile(filename,shortFilename,directory,post['CONTENTURL'])
            print("FFMPEG library not found, skipping merging video and audio")
        else:
            videoName = post['POSTID'] + "_video"
            videoURL = post['CONTENTURL']
            audioName = post['POSTID'] + "_audio"
            audioURL = videoURL[:videoURL.rfind('/')] + '/audio'
            
            print(directory,filename,sep="\n")

            getFile(videoName,videoName,directory,videoURL,silent=True)
            getFile(audioName,audioName,directory,audioURL,silent=True)
            try:
                self._mergeAudio(videoName,
                                audioName,
                                filename,
                                shortFilename,
                                directory)
            except KeyboardInterrupt:
                os.remove(directory / filename)
                os.remove(directory / audioName)

                os.rename(directory / videoName, directory / filename)

    @staticmethod
    def _mergeAudio(video,audio,filename,shortFilename,directory):
        
        inputVideo = str(directory / video)
        inputAudio = str(directory / audio)

        FNULL = open(os.devnull, 'w')
        cmd = f"ffmpeg -i {inputAudio} -i {inputVideo} -c:v copy -c:a aac -strict experimental {str(directory / filename)}"
        subprocess.call(cmd.split(), stdout=FNULL, stderr=subprocess.STDOUT)

        os.remove(directory / video)
        os.remove(directory / audio)