1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 """convert subtitle files to Gettext PO localization files"""
23
24 import sys
25
26 from translate.storage import po
27
28
30 """converts a subtitle file to a .po file..."""
31 output_store = po.pofile()
32 output_header = output_store.init_headers(charset="UTF-8", encoding="8bit")
33 output_header.addnote("extracted from %s" % input_store.filename, "developer")
34
35 for input_unit in input_store.units:
36 output_unit = convert_unit(input_unit, "developer")
37 if output_unit is not None:
38 output_store.addunit(output_unit)
39 output_store.removeduplicates(duplicatestyle)
40 return output_store
41
42
43 -def merge_store(template_store, input_store, blankmsgstr=False, duplicatestyle="msgctxt"):
44 """converts two subtitle files to a .po file..."""
45 output_store = po.pofile()
46 output_header = output_store.init_headers(charset="UTF-8", encoding="8bit")
47 output_header.addnote("extracted from %s, %s" % (template_store.filename, input_store.filename), "developer")
48
49 input_store.makeindex()
50 for template_unit in template_store.units:
51 origpo = convert_unit(template_unit, "developer")
52
53 template_unit_name = "".join(template_unit.getlocations())
54 if template_unit_name in input_store.locationindex:
55 translatedini = input_store.locationindex[template_unit_name]
56 translatedpo = convert_unit(translatedini, "translator")
57 else:
58 translatedpo = None
59
60 if origpo is not None:
61 if translatedpo is not None and not blankmsgstr:
62 origpo.target = translatedpo.source
63 output_store.addunit(origpo)
64 elif translatedpo is not None:
65 print >> sys.stderr, "error converting original subtitle definition %s" % origini.name
66 output_store.removeduplicates(duplicatestyle)
67 return output_store
68
69
71 """Converts a subtitle unit to a .po unit. Returns None if empty
72 or not for translation."""
73 if input_unit is None:
74 return None
75
76 output_unit = po.pounit(encoding="UTF-8")
77 output_unit.addlocation("".join(input_unit.getlocations()))
78 output_unit.addnote(input_unit.getnotes("developer"), "developer")
79 output_unit.source = input_unit.source
80 output_unit.target = ""
81 return output_unit
82
83
84 -def convertsub(input_file, output_file, template_file=None, pot=False, duplicatestyle="msgctxt"):
85 """Reads in L{input_file} using translate.subtitles, converts using L{sub2po}, writes to L{output_file}"""
86 from translate.storage import subtitles
87 input_store = subtitles.SubtitleFile(input_file)
88 if template_file is None:
89 output_store = convert_store(input_store, duplicatestyle=duplicatestyle)
90 else:
91 template_store = subtitles.SubtitleFile(template_file)
92 output_store = merge_store(template_store, input_store, blankmsgstr=pot, duplicatestyle=duplicatestyle)
93 if output_store.isempty():
94 return 0
95 output_file.write(str(output_store))
96 return 1
97
98
100 from translate.convert import convert
101 formats = {
102 "srt": ("po", convertsub), ("srt", "srt"): ("po", convertsub),
103 "sub": ("po", convertsub), ("sub", "sub"): ("po", convertsub),
104 "ssa": ("po", convertsub), ("ssa", "ssa"): ("po", convertsub),
105 "ass": ("po", convertsub), ("ass", "ass"): ("po", convertsub),
106 }
107 parser = convert.ConvertOptionParser(formats, usetemplates=True, usepots=True, description=__doc__)
108 parser.add_duplicates_option()
109 parser.passthrough.append("pot")
110 parser.run(argv)
111
112
113 if __name__ == '__main__':
114 main()
115