Exemplo n.º 1
0
def render_to_ajax(template_name, context=None, context_instance=None):
    "Render request into JSON object to be handled by AJAX on the server-side"
    if context is None:
        context = {}
    response_format = 'html'
    if 'response_format_tags' not in context:
        context['response_format_tags'] = 'ajax'

    context = preprocess_context_ajax(context)
    content = render_to_string(
        template_name, context, context_instance, response_format)
    content = convert_to_ajax(content, context_instance)
    context['content'] = json.dumps(content)

    notifications = []
    if context_instance and 'request' in context_instance:
        request = context_instance['request']
        maxmsgs = 5
        try:
            for message in list(messages.get_messages(request))[:maxmsgs]:
                msgtext = unicode(message)
                if 'updaterecord:' in msgtext[:13]:
                    try:
                        update_id = int(msgtext.split(':', 1)[1])
                        update = UpdateRecord.objects.get(pk=update_id)
                        message = {'message': update.get_full_message(),
                                   'tags': message.tags}
                        if update.author:
                            if update.record_type == 'manual' or update.record_type == 'share':
                                try:
                                    message[
                                        'image'] = update.author.get_contact().get_picture()
                                except:
                                    pass
                            message['title'] = unicode(update.author)
                        for obj in update.about.all():
                            message['message'] = "({0!s}) {1!s}:<br />{2!s}".format(
                                obj.get_human_type(), unicode(obj), message['message'])
                        notifications.append(message)
                    except:
                        pass
                else:
                    notifications.append({'message': unicode(message),
                                          'tags': message.tags})
        except:
            pass
    context['notifications'] = json.dumps(notifications)

    rendered_string = render_to_string(
        'ajax_base', context, context_instance, response_format='json')

    return rendered_string
Exemplo n.º 2
0
def render_to_response(template_name, context=None, context_instance=None, response_format='html'):
    "Extended render_to_response to support different formats"
    if context is None:
        context = {}
    if not response_format:
        response_format = 'html'

    if response_format not in settings.ANAF_RESPONSE_FORMATS:
        response_format = 'html'

    content_type = settings.ANAF_RESPONSE_FORMATS[response_format]

    if 'pdf' in response_format:
        while True:
            hasher = hashlib.md5()
            hasher.update(str(random.random()))
            filepath = u"pdfs/" + hasher.hexdigest()
            output = settings.MEDIA_ROOT + filepath
            if not os.path.exists(output + ".pdf"):
                break

        while True:
            hasher = hashlib.md5()
            hasher.update(str(random.random()))
            filepath = hasher.hexdigest() + ".html"
            source = getattr(settings, 'WKCWD', './') + filepath
            if not os.path.exists(source):
                break

        page_size = "A4"
        orientation = "portrait"

        rendered_string = render_to_string(
            template_name, context, context_instance, response_format)

        with codecs.open(source, encoding='utf-8', mode='w') as f:
            pdf_string = unicode(rendered_string)

            if context_instance and context_instance['request']:
                pdf_string = pdf_string.replace(
                    "a href=\"/", "a href=\"http://" + RequestSite(context_instance['request']).domain + "/")

            pdf_string.replace("href=\"/", "href=\"")

            pattern = """Content-Type: text/html|<td>\n\W*<div class="content-list-tick">\n\W.*\n.*</div></td>|<th scope="col">Select</th>"""
            pdf_string = re.sub(pattern, "", pdf_string).replace(
                '/static/', 'static/')

            f.write(pdf_string)

        wkpath = getattr(settings, 'WKPATH', './bin/wkhtmltopdf-i386')
        x = subprocess.Popen("{0!s} --print-media-type --orientation {1!s} --page-size {2!s} {3!s} {4!s}".format(wkpath,
                              orientation,
                              page_size,
                              source,
                              output),
                             shell=True,
                             cwd=getattr(settings, 'WKCWD', './'))
        x.wait()

        with open(output) as f:
            response = HttpResponse(f.read(), content_type='application/pdf')

        os.remove(output)
        os.remove(source)

        # response['Content-Disposition'] = 'attachment; filename=%s'%(pdf_name)

        return response

    if 'ajax' in response_format:
        rendered_string = render_to_ajax(
            template_name, context, context_instance)

    else:

        if response_format == 'html' and context_instance and context_instance['request'].path[:3] == '/m/':
            context['response_format'] = response_format = 'mobile'

        if settings.ANAF_FORCE_AJAX_RENDERING:
            context = preprocess_context_ajax(context)

        rendered_string = render_to_string(
            template_name, context, context_instance, response_format)

    response = HttpResponse(rendered_string, content_type=content_type)

    return response