溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

如何用python抓取網(wǎng)站教程并制作成PDF文檔

發(fā)布時間:2022-01-18 15:52:52 來源:億速云 閱讀:162 作者:iii 欄目:編程語言

這篇文章主要介紹“如何用python抓取網(wǎng)站教程并制作成PDF文檔”,在日常操作中,相信很多人在如何用python抓取網(wǎng)站教程并制作成PDF文檔問題上存在疑惑,小編查閱了各式資料,整理出簡單好用的操作方法,希望對大家解答”如何用python抓取網(wǎng)站教程并制作成PDF文檔”的疑惑有所幫助!接下來,請跟著小編一起來學習吧!

實現(xiàn)功能:抓取廖雪峰網(wǎng)站的python教程,保存html頁面到本地,然后用pdfkit包將html轉換為pdf。
運行需求:安裝python的pdfkit包和windows下的wkhtmltopdf。

代碼如下:
# -*- coding: utf-8
import urllib2
import urllib
import re,os
import time
import pdfkit
import sys
reload(sys)
sys.setdefaultencoding('utf-8')

class Liao:
    def __init__(self,init_url,out_put_path):
        self.init_url = init_url
        self.out_put_path = out_put_path
        self.user_agent = 'Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)'
        self.headers = { 'User-Agent' : self.user_agent }
        self.file_name = 'liao.pdf'
        self.html_template = """
                            <!DOCTYPE html>
                            <html lang="en">
                            <head>
                                <meta charset="gb2312">
                            </head>
                            <body>
                            {content}
                            </body>
                            </html>

    #獲取網(wǎng)頁內容
    def get_content(self,url):
        try:
            request = urllib2.Request(url,headers=self.headers)
            response = urllib2.urlopen(request)
            content = response.read().decode('utf-8').encode('GBK')
            return content
        except urllib2.URLError, e:
            if hasattr(e,"reason"):
                print u"連接頁面失敗,錯誤原因",e.reason
                return None

    #獲取文章目錄url列表
    def get_article_url(self):
        content = self.get_content(self.init_url)
        #print content
        pattern = re.compile(r'<li id="\d.*?" >.*?<a href="(.*?)">(.*?)</a>.*?</li>',re.S)
        # pattern = re.compile(r'<li id="(\d.*?)" >',re.S)
        result = re.findall(pattern,content)
        i = 1
        print len(result)
        # for x in result:
        #     print x[0],x[1]
        #     i+=1
        return result

    #獲取文章內容,傳入文章的url,返回文章內容
    def get_article_detail(self):
        url_list = self.get_article_url()
        pattern = re.compile(r'<div class="x-wiki-content">(.*?)</div>',re.S)
        # patt_title = re.compile(r'<h5>(.*?)</h5>')
        patt_img = re.compile(r'<img src="(/files/attachments/.*?)"',re.S)
        i = 1
        urls = []
        for item in url_list:
            #獲取文章的url
            article_url = 'http://www.liaoxuefeng.com' + item[1]
            url_ind = article_url.split('/')[-1]

            #提取文章標題
            article_title = item[2].replace('/','_')
            title_level = item[0]
            #title_tag = '<center><h' + title_level + '>' + article_title + '</h' + title_level + '></center>'
            title_tag = '<center><h2>' + article_title + '</h></center>'
            #生成html文件名
            html_file = url_ind + '.html'

            #根據(jù)文章url,抓取文章內容
            content = self.get_content(article_url)
            article_content = str(re.findall(pattern,content)[0])

            #處理圖片的url,加上域名
            def func(m):
                if not m.group(1).startswith("http"):
                    rtn = '<img src="http://www.liaoxuefeng.com' + m.group(1) + '"'
                    return rtn
                else:
                    return '<img src="' + m.group(1) + '"'
            article_content = re.compile(patt_img).sub(func, article_content)

            #將標題加到文章內容中
            article_content = title_tag + article_content
            article_content = self.html_template.format(content=article_content)

            #生成url及文件名列表
            urls.append((url_ind,html_file))

            #寫文件
            print 'saving file ' + html_file
            with open(html_file, 'wb') as f:
                f.write(article_content)
        return urls

    #將html保存成pdf
    def save_pdf(self,htmls):
        """
        把所有html文件保存到pdf文件
        """
        options = {
                'page-size': 'Letter',
                'encoding': "UTF-8",
                'custom-header': [
                    ('Accept-Encoding', 'gzip')
                ]
            }
        pdfkit.from_file(htmls, self.file_name, options=options)

init_url = 'http://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000'
out_put_path = 'd:/python/test/output'
liao = Liao(init_url,out_put_path)
urls = liao.get_article_detail()
htmls = [x[1] for x in urls]
liao.save_pdf(htmls)

到此,關于“如何用python抓取網(wǎng)站教程并制作成PDF文檔”的學習就結束了,希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習,快去試試吧!若想繼續(xù)學習更多相關知識,請繼續(xù)關注億速云網(wǎng)站,小編會繼續(xù)努力為大家?guī)砀鄬嵱玫奈恼拢?/p>

向AI問一下細節(jié)

免責聲明:本站發(fā)布的內容(圖片、視頻和文字)以原創(chuàng)、轉載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權內容。

AI