一个简单的flv操作库

忘记从哪里找到的了.

1. 'Utilities for working with flv files'
3. from __future__ import with_statement
4. import array, struct
6. def duration(header):
7.     'Duration of flv file'
8.     i = header.index('duration') + len('duration')
9.     a = array.array('d')
10.     a.fromstring(header[i + 1:i + 9])
11.     a.byteswap()
12.     return a[0]
14. def tag_value(text, tag, offset, length):
15.     'Value of tag in text'
16.     i = text.index(tag) + len(tag) + offset
17.     return text[i:i + length]
19. def decode_double(s):
20.     'Double value from binary encoded string'
21.     a = array.array('d')
22.     a.fromstring(s)
23.     a.byteswap()
24.     return a[0]
25.     
27. def replace(text, tag, offset, length, new):
28.     'Replace value of tag in text with new (which must string)'
29.     i = text.index(tag) + len(tag)
30.     return text[:i + offset] + new + text[i + offset + length:]
33. def header_body(flv_path):
34.     'Tuple (header, body) of flv file without onMetaData tag'
35.     with open(flv_path, 'rb') as f:
36.         header = f.read(9)
37.         f.read(4)
38.         f.read(1)
39.         data_size = struct.unpack('>I', '/x00' + f.read(3))[0]
40.         f.read(7)
41.         f.read(data_size)
42.         body = f.read()
43.     return header, body
45. def flv_header(f):
46.     'First 9 bytes from file'
47.     return f.read(9)
49. def skip_metadata(f):
50.     'File position must be on previous tag size'
51.     result = ''
52.     f.read(4)
53.     result += f.read(1)
54.     size = f.read(3)
55.     result += size
56.     data_size = struct.unpack('>I', '/x00' + size)[0]
57.     result += f.read(7)
58.     return result + f.read(data_size)
61. def flv_metadata(f):
62.     'File position must be on previous tag size'
63.     return skip_metadata(f)
65. def add_prev_size(tag, size):
66.     'Replace previous tag size with size'
67.     return struct.pack('>I', size) + tag
69. def previous_tag_size(f):
70.     return struct.unpack('>I', f.read(4))
73. def tag_timestamp(tag):
74.     'Timestamp value of in milliseconds'
75.     return struct.unpack('>I', '/x00' + tag[4:7])[0]
78. def replace_timestamp(tag, timestamp):
79.     'Replace timestamp in tag with new value'
80.     packed = struct.pack('>I', timestamp)[1:]
81.     return tag[:4] + packed + tag[7:]
83. def simple_merge(out_path, *paths):
84.     'Merge paths (which must be flv files) and write output to out_path'
85.     parsed = []
86.     files = [open(p, 'rb') for p in paths]
87.     header = flv_header(files[0])
88.     for f in files[1:]: flv_header(f)
89.     metadata = [flv_metadata(f) for f in files]
90.     total = 0
91.     for meta in metadata:
92.         total += duration(meta)
93.     metadata[0] = replace(metadata[0], 'duration', 1, 8, struct.pack('>d', total))
94.     with open(out_path, 'wb') as out:
95.         out.write(header)
96.         prev_size = 0
97.         offset = 0
98.         # uncomment next line if you want save metadata
99.         # out.write(add_prev_size(metadata[0], prev_size))
100.         for f in files:
101.             prev = previous_tag_size(f)
102.             for tag, size in tags(f):
103.                 new_offset = tag_timestamp(tag)
104.                 tag = replace_timestamp(tag, offset + new_offset)
105.                 out.write(add_prev_size(tag, prev_size))
106.                 prev_size = size
107.                 prev = previous_tag_size(f)
108.             offset += new_offset
112. def tags(flv, max_tags = -1):
113.     '''Generator yielding (tag, size) from flv stream. File position must be on
114. tag type'''
115.     tag_type = flv.read(1)
116.     while tag_type and max_tags:
117.         basic_size = flv.read(3)
118.         data_size = struct.unpack('>I', '/x00' + basic_size)[0]
119.         buf = tag_type + basic_size
120.         buf += flv.read(7)
121.         buf += flv.read(data_size)
122.         #print(flv.tell())
123.         yield buf, data_size + 11
124.         tag_type = flv.read(1)
125.         max_tags -= 1
127. import glob, os.path, re, tkFileDialog, tkMessageBox, webbrowser, sys
129. def flv_files(directory):
130.     ''''List of flv files in directory sorted in right order
131. (i.e part2 is before part10)'''
132.     lst = glob.glob(os.path.join(directory, '*.flv'))
133.     def right_cmp(x, y):
134.         xs = int(re.findall(r'/d+', x)[-1])
135.         ys = int(re.findall(r'/d+', y)[-1])
136.         return cmp(xs, ys)
137.     lst.sort(right_cmp)
138.     return lst
141. if __name__ == '__main__':
142.     directory = tkFileDialog.askdirectory()
143.     if directory:
144.         out_file = os.path.join(os.path.dirname(directory),
145.                                 os.path.basename(directory) + '.flv')
146.         lst = flv_files(directory)
147.         if len(lst) == 0:
148.             tkMessageBox.showerror('Error', 'No flv files in %s' % directory)
149.         else:
150.             simple_merge(out_file, *lst)
151.             webbrowser.open(out_file)