1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
import mediabrowser.ffmpeg as ffmpeg
import os
import tempfile
from pprint import pprint
from unittest import TestCase
class TestFfmpeg(TestCase):
def setUp(self):
self.test_mkv = os.path.join(os.path.dirname(__file__),
"data", "test.mkv")
# def test_data(self):
# data = ffmpeg.ffprobe_data(self.test_mkv)
# assert 'streams' in data
# assert len(data['streams']) == 2
#
# assert 'format' in data
# assert 'duration' in data['format']
# assert data['format']['duration'] == '27.472000'
def test_calculate_splittimes(self):
for time in ffmpeg.calculate_splittimes(self.test_mkv, 120):
print(time)
def test_split(self):
_, pos1 = ffmpeg.find_next_keyframe(self.test_mkv, 2, 5)
frame_dur1, pos2 = ffmpeg.find_next_keyframe(self.test_mkv, pos1+2, 5)
frame_dur2, pos3 = ffmpeg.find_next_keyframe(self.test_mkv, pos2+2, 5)
out1 = open(tempfile.mktemp(), "wb")
out2 = open(tempfile.mktemp(), "wb")
pprint(locals())
out1.write(ffmpeg.stream(self.test_mkv, pos1, pos2-pos1-frame_dur1).stdout.read())
out2.write(ffmpeg.stream(self.test_mkv, pos2, pos3-pos2-frame_dur2).stdout.read())
os.remove(out1.name)
os.remove(out2.name)
|
