Python - 获取URL,解析和打印PDF

时间:2012-03-24 13:01:49

标签: python django beautifulsoup reportlab

我正在尝试获取URL的HTML源,解析它然后将结果打印为PDF。

为此,我想依靠BeautifulSoup,urllib2和reportlab,但我缺乏如何正确组合它们。

作为一个错误,我在运行django 1.3.1 dev服务器并访问视图时得到'module' object is not callable

这是我的代码:

from reportlab.pdfgen import canvas
from cStringIO import StringIO
from django.http import HttpResponse
from django.shortcuts import render_to_response
from django.template import RequestContext
# Fetching the URL
import urllib2

# Parsing the HTML
from BeautifulSoup import BeautifulSoup

# The ConverterForm
from django import forms

class ConverterForm(forms.Form):
    # Use textarea instead the default TextInput.
    html_files = forms.CharField(widget=forms.Textarea)
    filename = forms.CharField()

# Create your views here.
def create_pdf(request):
    # If the form has been submitted
    if request.method == 'POST':
        # A form bound to the POST data
        form = ConverterForm(request.POST)
    # All validation rules pass
    if form.is_valid():
        # PDF creation process
        # Assign variables
        html_files = form.cleaned_data['html_files']
        filename = form.cleaned_data['filename']

        # Create the HttpResponse object with the appropriate PDF headers.
        response = HttpResponse(mimetype='application/pdf')
        # The use of attachment forces the Save as dialog to open.
        response['Content-Disposition'] = 'attachment; filename=%s.pdf' % filename

        buffer = StringIO()

        # Get the page source
        page = urllib2.urlopen(html_files)
        html = page.read()

        # Parse the page source
        soup = BeautifulSoup(html)

        # Create the PDF object, using the StringIO() object as its "file".
        p = canvas.Canvas(buffer)

        # Draw things on the PDF and generate the PDF.
        # See ReportLab documentation for full list of functions.
        p.drawString(100, 100, soup)

        # Close the PDF object cleanly.
        p.showPage()
        p.save()

        # Get the value of the StringIO buffer and write it to the response.
        pdf = buffer.getvalue()
        buffer.close()
        response.write(pdf)
        return response

    else:
        # An unbound form
        form = ConverterForm()

    # For RequestContext in relation to csrf see more here:
    # https://docs.djangoproject.com/en/1.3/intro/tutorial04/
    return render_to_response('converter/index.html', {
    'form': form,
    }, context_instance=RequestContext(request))

2 个答案:

答案 0 :(得分:4)

您需要导入BeautifulSoup类:

from BeautifulSoup import BeautifulSoup

可能会让人感到困惑,因为模块和类都具有相同的基本名称。

答案 1 :(得分:1)

这是一种更简单的方法:

import cStringIO as StringIO

import ho.pisa as pisa
import requests

def pdf_maker(request):

    browser = requests.get('http://www.google.com/')
    html = browser.text

    result = StringIO.StringIO()
    source = StringIO.StringIO(html.encode('UTF-8')) # adjust as required

    pdf = pisa.pisaDocument(source,dest=result)

    if not pdf.err:
        response = HttpResponse(result.getvalue(),mimetype='application/pdf')
        response['Content-Disposition'] = 'attachment; filename=the_file.pdf'
        return response

    return render(request,'error.html')

这使用requestspisa。但是,您将对此(以及其他此类解决方案)有一​​些限制。也就是说,您需要找到一种自己获取和嵌入图像的方法,因为PDF转换过程无法直接从Internet加载图像。