bottle_gettext.py

'''
'''

__author__ = "Frederic Rodrigo"
__version__ = '0.1'
__license__ = 'MIT'

### CUT HERE (see setup.py)

from bottle import request, response
import gettext, inspect


class GettextPlugin(object):
    '''
    '''

    name = 'gettext'
    api  = 2

    def __init__(self, domain, localedir, allowed_languages, keyword='lang'):
        self.domain = domain
        self.localedir = localedir
        self.allowed_languages = allowed_languages
        self.keyword = keyword
        self.cache = {}

    def get_language(self):
        lang = [None]

        # Remove old no longer used cookie 'lang'
        if request.get_cookie('lang'):
            response.delete_cookie('lang', path="/")

        if len(request.script_name) > 3:
            lang = [request.script_name[-3:-1], self.allowed_languages[0]]
            if lang[0] not in self.allowed_languages:
                lang = [None]
            else:
                return (lang, False)

        if not lang[0] and request.get_cookie('lang'):
            lang = [request.get_cookie('lang')]
            if lang[0] not in self.allowed_languages:
                lang = [None]

        if not lang[0] and request.get_header('Accept-Language'):
            lang = request.get_header('Accept-Language')
            lang = lang.split(',')
            lang = [x.split(";")[0] for x in lang]
            lang = [x.split("-")[0] for x in lang]
            lang = [x for x in lang if x in self.allowed_languages]

        if len(lang) > 0 and lang[0]:
            lang.append(self.allowed_languages[0])
            res = []
            for l in lang:
                if not l in res:
                    res.append(l)
            return (res, True)
        else:
            return (self.allowed_languages, True)

    def apply(self, callback, route):
        conf = route.config.get('gettext') or {}
        keyword = conf.get('keyword', self.keyword)

        # Test if the original callback accepts a 'lang' keyword.
        # Ignore it if it does not need a gettext handle.
        args = inspect.getargspec(route.callback)[0]
        if keyword not in args:
            return callback

        def wrapper(*args, **kwargs):
            (language, redirect) = self.get_language()

            if redirect:
                from tools import utils
                from bottle import redirect
                url = request.urlparts
                new_url = []
                new_url.append(url.scheme)
                new_url.append("://")
                new_url.append(utils.website)
                new_url.append("/" + language[0])
                new_url.append(request.fullpath)
                if url.query:
                    new_url.append("?")
                    new_url.append(url.query)
                redirect("".join(new_url))
                return

            # Setup Gettext
            k = ','.join(language)
            if self.cache.has_key(k):
                gt = self.cache[k]
            else:
                gt = gettext.translation(self.domain, localedir=self.localedir, languages=language)
                self.cache[k] = gt
            gt.install(unicode=1)

            # Add the language as a keyword argument.
            kwargs[keyword] = language

            return callback(*args, **kwargs)

        return wrapper

Plugin = GettextPlugin