add lib seaborn

python/util/audio_to_mp3.py

@@ -1,42 +0,0 @@
-from moviepy.editor import *
-import sys
-# print('argv[1]:' + str(sys.argv[1]))
-import os
-from pydub import AudioSegment
-
-
-def mp4_to_mp3(mp4_uri):
-	# Load the mp4 file
-	video = VideoFileClip(mp4_uri)
-
-	# Extract audio from video
-	video.audio.write_audiofile(mp4_uri + ".mp3")
-
-
-def audio_to_wav(wav_uri, out_sample_rate=16000):
-    src = wav_uri
-    dst = wav_uri + ".wav"
-    if os.path.exists(dst): os.remove(dst)
-
-    # convert wav to mp3
-    sound = AudioSegment.from_file(src)
-    sound = sound.set_frame_rate(out_sample_rate)
-    sound.export(dst, format="wav")
-
-    print(' audio_to_wav: ', str(dst))
-    return dst
-
-
-def wav_to_mp3(wav_uri):
-	AudioSegment.from_wav(wav_uri).export(wav_uri + '.mp3', format='mp3')
-
-
-# sys.argv
-if len(sys.argv) > 1:
-	src_uri = sys.argv[1]
-	if src_uri.endswith('.mp4'):
-		res = mp4_to_mp3(src_uri)
-	else:
-		res = wav_to_mp3(src_uri)
-else:
-	print(' usage: python mp4_to_mp3.py /path/to/mp4_file ')

python/util/to_mp3.py

@@ -0,0 +1,76 @@
+from pathlib import Path
+
+from moviepy.editor import *
+import sys
+# print('argv[1]:' + str(sys.argv[1]))
+import os
+from pydub import AudioSegment
+
+
+def mp4_to_mp3(mp4_uri):
+    # Load the mp4 file
+    video = VideoFileClip(mp4_uri)
+
+    # Extract audio from video
+    mp3_uri = mp4_uri + ".mp3"
+    video.audio.write_audiofile(mp3_uri)
+
+    return mp3_uri
+
+
+def audio_to_wav(wav_uri, out_sample_rate=16000):
+    src = wav_uri
+    dst = wav_uri + ".wav"
+    if os.path.exists(dst): os.remove(dst)
+
+    # convert wav to mp3
+    sound = AudioSegment.from_file(src)
+    sound = sound.set_frame_rate(out_sample_rate)
+    sound.export(dst, format="wav")
+
+    print(' audio_to_wav: ', str(dst))
+    return dst
+
+
+def wav_to_mp3(wav_uri):
+    mp3_uri = wav_uri + '.mp3'
+    AudioSegment.from_wav(wav_uri).export(mp3_uri, format='mp3')
+    return mp3_uri
+
+
+# def wav_to_mp3_bitrate(wav_uri, bitrate):
+#     mp3_uri = wav_uri + '.mp3'
+#     AudioSegment.from_wav(wav_uri).export(mp3_uri, format='mp3', bitrate=bitrate)
+#     return mp3_uri
+
+
+# 处理命令行
+if __name__ == "__main__":
+    # 获取命令行参数
+    if len(sys.argv) < 2:
+        print("用法：python to_mp3.py <文件或通配符>")
+        sys.exit(1)
+
+    # 处理文件或通配符参数
+    bitrate = -1
+    if len(sys.argv) > 2:
+        bitrate = sys.argv[2]
+
+    for input_file in sys.argv[1:]:
+        # print("input_file:" + str(input_file))
+        # input_files.extend(Path().glob(input_file))
+        if input_file.lower().endswith(".wav"):
+            res = wav_to_mp3(str(input_file))
+            print(res)
+        elif input_file.lower().endswith(".mp4"):
+            res = mp4_to_mp3(str(input_file))
+            print(res)
+
+    # # 转换每个文件为 MP3 文件
+    # for input_file in input_files:
+    #     if input_file.suffix.lower() == ".wav":
+    #         res = wav_to_mp3(str(input_file))
+    #         print(res)
+    #     elif input_file.suffix.lower() == ".mp4":
+    #         res = mp4_to_mp3(str(input_file))
+    #         print(res)