split_video

alisx · alisx · commit 9e943e5dadbf · 2021-10-16T14:08:46.000+08:00
diff --git a/taiyangxue/README.md b/taiyangxue/README.md
@@ -1,5 +1,6 @@
 # Python 代码实例
 
+- [split-video](https://github.com/JustDoPython/python-examples/tree/master/taiyangxue/split-video) : 神器 ffmpeg —— 实现短视频批量化操作
 - [pypandoc](https://github.com/JustDoPython/python-examples/tree/master/taiyangxue/pypandoc) : 神器 Pypandoc —— 实现电子书自由
 - [python-thread](https://github.com/JustDoPython/python-examples/tree/master/taiyangxue/python-thread) : 这么一搞，再也不怕线程打架了
 - [python-op2](https://github.com/JustDoPython/python-examples/tree/master/taiyangxue/python-op2) : 只需一招，Python 将系统秒变在线版！
diff --git a/taiyangxue/split-video/split_video.py b/taiyangxue/split-video/split_video.py
@@ -0,0 +1,68 @@
+import math
+import os
+
+# 获取视频时长
+import subprocess as sp
+def get_video_duration(filename):
+
+    cmd = "ffprobe -v error -show_entries format=duration -of default=noprint_wrappers=1:nokey=1 -i %s" % filename
+    print(cmd)
+    p = sp.Popen(cmd, stdout=sp.PIPE, stderr=sp.PIPE)
+    p.wait()
+    # p.stdout
+    strout, strerr = p.communicate() # 去掉最后的回车
+    print(strout, strerr)
+    ret = strout.decode("utf-8").strip() # 去掉最后的回车
+    print(ret)
+    return ret
+
+
+# 将视频拆分成一分钟的小段
+def cut_video(filename, outfile, start, length=90):
+    cmd = "ffmpeg -i %s -ss %d -t %d -c copy %s" % (filename, start, length, outfile)
+    p = sp.Popen(cmd, shell=True)
+    p.wait()
+    return
+
+# 计算分段
+def split_video(filename, outdir, length=90, overwrite=True):
+
+    duration = math.floor(float(get_video_duration(filename)))
+    part = math.floor(duration / length)
+
+    basenames = os.path.basename(filename).split('.')
+
+    mainname = basenames[0]
+    extname = basenames[1]
+    start = 0
+    partindex = 1
+    for i in range(0, part):
+        outname = os.path.join(outdir, ''.join([mainname, "_", str(partindex), ".", extname]))
+        if os.path.exists(outname):
+            if overwrite:
+                os.remove(outname)
+            else:
+                print("文件已生成，跳过")
+                continue
+
+        cut_video(filename, outname, start, length)
+        start += length
+        partindex += 1
+        pass
+    return partindex
+
+
+# 获取文件
+def main(dir):
+    outdir = os.path.join(dir, "output")
+    if not os.path.exists(outdir):
+        os.mkdir(outdir)
+    for fname in os.listdir(dir):
+        fname = os.path.join(dir, os.path.join(dir, fname))
+        if os.path.isfile(fname):
+            split_video(fname, outdir)
+            
+if __name__ == '__main__':
+    main(r"D:\Project\marketing\videos")
+
+
