import_po.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Download translations from transifex and regenerate files."""

from __future__ import unicode_literals, print_function
import codecs
from glob import glob
import os

import polib

os.system("tx pull -a")
trans_files = glob(os.path.join('translations', 'nikola.messages', '*.po'))
for fname in trans_files:
    lang = os.path.splitext(os.path.basename(fname))[0].lower()
    outf = os.path.join('nikola', 'data', 'themes', 'base',
                        'messages', 'messages_{0}.py'.format(lang))
    po = polib.pofile(fname)
    lines = """# -*- encoding:utf-8 -*-
from __future__ import unicode_literals

MESSAGES = {""".splitlines()
    lines2 = []
    for entry in po:
        lines2.append('    "{0}": "{1}",'. format(entry.msgid, entry.msgstr))
        ### BACKWARDS COMPATIBILITY PATCH START
        ### TODO: remove in v7
        if entry.msgid in ["Posted:", "Also available in:"]:
            fid = entry.msgid
            fid = fid.replace(':', '')
            fstr = entry.msgstr
            fstr = fstr.replace(':', '').replace('：', '')
            lines2.append('    "{0}": "{1}",'. format(fid, fstr))
        elif entry.msgid == 'More posts about %s':
            fid = entry.msgid
            fid = fid.replace(' %s', '')
            fstr = entry.msgstr
            fstr = fstr.replace(' %s', '').replace('%s', '')
            lines2.append('    "{0}": "{1}",'. format(fid, fstr))
        ### BACKWARDS COMPATIBILITY PATCH END
        ### TODO: remove in v7
    lines.extend(sorted(lines2))
    lines.append("}\n")
    print("Generating:", outf)
    with codecs.open(outf, "wb+", "utf8") as outfile:
        outfile.write('\n'.join(lines))