Added progress_bar option to write_audiofile

Zulko · Mar 3, 2017 · 733d3e2 · 733d3e2
1 parent 6a17cf9
commit 733d3e2
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 14 deletions.
diff --git a/moviepy/audio/AudioClip.py b/moviepy/audio/AudioClip.py
@@ -147,9 +147,10 @@ def max_volume(self, stereo=False, chunksize=50000, progress_bar=False):
 
     @requires_duration
     def write_audiofile(self,filename, fps=44100, nbytes=2,
-                     buffersize=2000, codec=None,
-                     bitrate=None, ffmpeg_params=None,
-                     write_logfile=False, verbose=True):
+                        buffersize=2000, codec=None,
+                        bitrate=None, ffmpeg_params=None,
+                        write_logfile=False, verbose=True,
+                        progress_bar=True):
         """ Writes an audio file from the AudioClip.
 
 
@@ -188,6 +189,8 @@ def write_audiofile(self,filename, fps=44100, nbytes=2,
           If True, displays informations
 
 
+
+
         """
 
         if codec is None:
@@ -200,11 +203,11 @@ def write_audiofile(self,filename, fps=44100, nbytes=2,
                        "write_videofile.")
 
         return ffmpeg_audiowrite(self, filename, fps, nbytes, buffersize,
-                      codec=codec, bitrate=bitrate, write_logfile=write_logfile,
-                      verbose=verbose, ffmpeg_params=ffmpeg_params)
+                                 codec=codec, bitrate=bitrate, write_logfile=write_logfile,
+                                 verbose=verbose, ffmpeg_params=ffmpeg_params,
+                                 progress_bar=progress_bar)
+
 
-###
-#
 # The to_audiofile method is replaced by the more explicit write_audiofile.
 AudioClip.to_audiofile = deprecated_version_of(AudioClip.write_audiofile,
                                                'to_audiofile')

diff --git a/moviepy/audio/io/ffmpeg_audiowriter.py b/moviepy/audio/io/ffmpeg_audiowriter.py
@@ -137,7 +137,7 @@ def close(self):
 def ffmpeg_audiowrite(clip, filename, fps, nbytes, buffersize,
                       codec='libvorbis', bitrate=None,
                       write_logfile = False, verbose=True,
-                      ffmpeg_params=None):
+                      ffmpeg_params=None, progress_bar=True):
     """
     A function that wraps the FFMPEG_AudioWriter to write an AudioClip
     to a file.
@@ -155,11 +155,10 @@ def ffmpeg_audiowrite(clip, filename, fps, nbytes, buffersize,
                                 logfile=logfile,
                                 ffmpeg_params=ffmpeg_params)
 
-
-
     for chunk in clip.iter_chunks(chunksize=buffersize,
-                                  progress_bar=True, quantize=True,
-                                  nbytes= nbytes, fps=fps):
+                                  quantize=True,
+                                  nbytes= nbytes, fps=fps,
+                                  progress_bar=progress_bar):
         writer.write_frames(chunk)
 
     """

diff --git a/moviepy/video/VideoClip.py b/moviepy/video/VideoClip.py
@@ -255,7 +255,8 @@ def write_videofile(self, filename, fps=None, codec=None,
           These will be files ending with '.log' with the name of the
           output file in them.
 
-
+        progress_bar
+          If false, will not display the red progress bar
 
         Examples
         ========
@@ -326,7 +327,7 @@ def write_videofile(self, filename, fps=None, codec=None,
                                        audio_nbytes, audio_bufsize,
                                        audio_codec, bitrate=audio_bitrate,
                                        write_logfile=write_logfile,
-                                       verbose=verbose)
+                                       verbose=verbose, progress_bar=progress_bar)
 
         ffmpeg_write_video(self, filename, fps, codec,
                            bitrate=bitrate,