inkstitch/lib/inx/utils.py

import os
import gettext
from os.path import dirname
from jinja2 import Environment, FileSystemLoader

from ..i18n import translation as default_translation, locale_dir, N_


_top_path = dirname(dirname(dirname(os.path.realpath(__file__))))
inx_path = os.path.join(_top_path, "inx")
template_path = os.path.join(_top_path, "templates")

current_translation = default_translation
current_locale = "en_US"


def build_environment():
    env = Environment(
        loader=FileSystemLoader(template_path),
        autoescape=True,
        extensions=['jinja2.ext.i18n']
    )

    env.install_gettext_translations(current_translation)
    env.globals["locale"] = current_locale

    return env


def write_inx_file(name, contents):
    inx_file_name = "inkstitch_%s_%s.inx" % (name, current_locale)
    with open(os.path.join(inx_path, inx_file_name), 'w') as inx_file:
        print >> inx_file, contents.encode("utf-8")


def iterate_inx_locales():
    global current_translation, current_locale

    locales = sorted(os.listdir(locale_dir))
    for locale in locales:
        translation = gettext.translation("inkstitch", locale_dir, languages=[locale], fallback=True)

        # L10N If you translate this string, that will tell Ink/Stitch to
        # generate menu items for this language in Inkscape's "Extensions"
        # menu.
        magic_string = N_("Generate INX files")
        translated_magic_string = translation.ugettext(magic_string)

        if translated_magic_string != magic_string or locale == "en_US":
            current_translation = translation
            current_locale = locale
            yield locale