#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright 2004-2006 Zuza Software Foundation
# 
# This file is part of translate.
#
# translate is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
# 
# translate is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with translate; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#

"""Converts Qt .ts localization files to Gettext .po files
You can convert back to .ts using po2ts"""

from translate.storage import po
from translate.storage import ts
from translate.misc import quote

class ts2po:
  def convertmessage(self, contextname, messagenum, msgid, msgstr, msgcomments, transtype):
    """makes a pounit from the given message"""
    thepo = po.pounit(encoding="UTF-8")
    thepo.sourcecomments.append("#: %s#%d\n" % (contextname, messagenum))
    thepo.msgid = [quote.quotestr(quote.rstripeol(line)) for line in msgid.split("\n")]
    if len(thepo.msgid) > 1:
      thepo.msgid = [quote.quotestr("")] + thepo.msgid
    thepo.msgstr = [quote.quotestr(quote.rstripeol(line)) for line in msgstr.split("\n")]
    if len(msgcomments)>0:
      thepo.othercomments.append("# %s\n" %(msgcomments))
    if transtype == "unfinished" and not thepo.isblankmsgstr():
      thepo.markfuzzy()
    if transtype == "obsolete":
      thepo.visiblecomments.append("#_ OBSOLETE\n")
      # using the fact that -- quote -- "(this is nonsense)"
    return thepo

  def convertfile(self, inputfile):
    """converts a .ts file to .po format"""
    tsfile = ts.QtTsParser(inputfile)
    thepofile = po.pofile()
    headerpo = thepofile.makeheader(charset="UTF-8", encoding="8bit")
    thepofile.units.append(headerpo)
    for contextname, messages in tsfile.iteritems():
      messagenum = 0
      for message in messages:
        messagenum += 1
        source = tsfile.getmessagesource(message)
        translation = tsfile.getmessagetranslation(message)
        comment = tsfile.getmessagecomment(message)
        transtype = tsfile.getmessagetype(message)
        thepo = self.convertmessage(contextname, messagenum, source, translation, comment, transtype)
        thepofile.units.append(thepo)
    return thepofile

def convertts(inputfile, outputfile, templates):
  """reads in stdin using fromfileclass, converts using convertorclass, writes to stdout"""
  convertor = ts2po()
  outputpo = convertor.convertfile(inputfile)
  outputposrc = str(outputpo)
  outputfile.write(outputposrc)
  return 1

def main(argv=None):
  from translate.convert import convert
  formats = {"ts":("po",convertts)}
  parser = convert.ConvertOptionParser(formats, usepots=True, description=__doc__)
  parser.run(argv)