Package translate :: Package convert :: Module sub2po
[hide private]
[frames] | no frames]

Source Code for Module translate.convert.sub2po

  1  #!/usr/bin/env python 
  2  # -*- coding: utf-8 -*- 
  3  # 
  4  # Copyright 2008-2009 Zuza Software Foundation 
  5  # 
  6  # This file is part of translate. 
  7  # 
  8  # translate is free software; you can redistribute it and/or modify 
  9  # it under the terms of the GNU General Public License as published by 
 10  # the Free Software Foundation; either version 2 of the License, or 
 11  # (at your option) any later version. 
 12  # 
 13  # translate is distributed in the hope that it will be useful, 
 14  # but WITHOUT ANY WARRANTY; without even the implied warranty of 
 15  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
 16  # GNU General Public License for more details. 
 17  # 
 18  # You should have received a copy of the GNU General Public License 
 19  # along with translate; if not, write to the Free Software 
 20  # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA 
 21   
 22  """convert subtitle files to Gettext PO localization files""" 
 23   
 24  import sys 
 25   
 26  from translate.storage import po 
 27   
 28   
29 -def convert_store(input_store, duplicatestyle="msgctxt"):
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 # try and find a translation of the same name... 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 # if we have a valid po unit, get the translation and add it... 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
70 -def convert_unit(input_unit, commenttype):
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 # escape unicode 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
99 -def main(argv=None):
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