ENDLESS X

# settings.py

TEMPLATE_CONTEXT_PROCESSORS = (
    'django.core.context_processors.request',
)

# project/common/templatetags/custom_tags.py

from django.conf import settings
from django.template import Library, Node
from django.template.defaulttags import URLNode
from django.utils.encoding import iri_to_uri

from uamobile.nonmobile import NonMobileUserAgent

register = Library()

class MobileURLNode(URLNode):
    def render(self, context):
        url = super(MobileURLNode, self).render(context)
        device = context['request'].device
        # 携帯以外のアクセス処理
        if isinstance(device, NonMobileUserAgent):
            return url
        # cookieが利用可能時処理
        if device.supports_cookie():
            return url
        # session_idの付加
        return url + iri_to_uri(
            '?%s=%s' % (
                settings.SESSION_COOKIE_NAME,
                context['request'].session.session_key
            )
        )

def mobile_url(parser, token):
    bits = token.contents.split(' ', 2)
    if len(bits) < 2:
        raise TemplateSyntaxError, "'%s' takes at least one argument (path to a view)" % bits[0]
    args = []
    kwargs = {}
    if len(bits) > 2:
        for arg in bits[2].split(','):
            if '=' in arg:
                k, v = arg.split('=', 1)
                k = k.strip()
                kwargs[k] = parser.compile_filter(v)
            else:
                args.append(parser.compile_filter(arg))
    return MobileURLNode(bits[1], args, kwargs)
url = register.tag(mobile_url)

Comments