svg.py - mozsearch

Enable keyboard shortcuts

import os, sys

from wptserve.utils import isomorphic_decode

import importlib

subresource = importlib.import_module("common.security-features.subresource.subresource")

def generate_payload(request, server_data):

    data = (u'{"headers": %(headers)s}') % server_data

    if b"id" in request.GET:

        with request.server.stash.lock:

            request.server.stash.take(request.GET[b"id"])

            request.server.stash.put(request.GET[b"id"], data)

    return u"<svg xmlns='http://www.w3.org/2000/svg'></svg>"

def generate_payload_embedded(request, server_data):

    return subresource.get_template(u"svg.embedded.template") % {

        u"id": isomorphic_decode(request.GET[b"id"]),

        u"property": isomorphic_decode(request.GET[b"property"])}

def generate_report_headers_payload(request, server_data):

    stashed_data = request.server.stash.take(request.GET[b"id"])

    return stashed_data

def main(request, response):

    handler = lambda data: generate_payload(request, data)

    content_type = b'image/svg+xml'

    if b"embedded-svg" in request.GET:

        handler = lambda data: generate_payload_embedded(request, data)

    if b"report-headers" in request.GET:

        handler = lambda data: generate_report_headers_payload(request, data)

        content_type = b'application/json'

    subresource.respond(request,

                        response,

                        payload_generator = handler,

                        content_type = content_type)