Esempio n. 1
0
def select_pdf_document(request, ctx, obj):
    """
    Views to select pdf files to download.

    :type obj: :class:`.DocumentController`
    """
    if request.GET.get("Download"):
        formset = get_pdf_formset(obj, request.GET)
        if formset.is_valid():
            files = []
            for form in formset.forms:
                selected = form.cleaned_data["selected"]
                if selected:
                    df = form.cleaned_data["id"]
                    files.append(df)
            return download_merged_pdf(obj, files)
    else:
        formset = get_pdf_formset(obj)
    ctx["pdf_formset"] = formset
    return r2r("select_pdf_doc.html", ctx, request)
Esempio n. 2
0
def select_pdf_part(request, ctx, obj):
    """
    View helper to select pdf files to download.

    :type obj: :class:`.PartController`
    """
    data = request.GET if request.GET.get("Download") else None
    # retrieve all pdf files (from documents attached to children parts)
    children = obj.get_children(-1)
    formsets = []
    self_link = FakeLink("self", child=obj.object)
    for level, link in [(0, self_link)] + list(children):
        link.formsets = []
        for l in link.child.documentpartlink_part.now():
            doc = l.document
            ctrl = get_controller(doc.type)(doc, request.user)
            if ctrl.check_readable(raise_=False):
                formset = get_pdf_formset(doc, data,
                        prefix="pdf_%s_%d" % (link.id, doc.id))
                link.formsets.append(formset)
        formsets.append((level, link))

    if data:
        # returns a pdf file if all formsets are valid
        valid = True
        files = []
        for level, link in formsets:
            for formset in link.formsets:
                if formset.is_valid():
                    for form in formset.forms:
                        selected = form.cleaned_data["selected"]
                        if selected:
                            df = form.cleaned_data["id"]
                            files.append(df)
                else:
                    valid = False
        if valid:
            return download_merged_pdf(obj, files)
    ctx["children"] = formsets
    return r2r("select_pdf_part.html", ctx, request)