码迷,mamicode.com
首页 > Web开发 > 详细

html2doc

时间:2017-10-28 19:52:51      阅读:1445      评论:0      收藏:0      [点我收藏+]

标签:error:   rom   headers   就是   oct   mob   定位   _id   agent   

0.

 

1.参考

Python 爬虫:把廖雪峰教程转换成 PDF 电子书 

https://github.com/lzjun567/crawler_html2pdf

wkhtmltopdf 就是一个非常好的工具,它可以用适用于多平台的 html 到 pdf 的转换,pdfkit 是 wkhtmltopdf 的Python封装包。

 

https://www.crummy.com/software/BeautifulSoup/bs4/doc/#

也可以通过 BeautifulSoup 插入删除tag

soup.insert
soup.decompose

 

2.安装

https://wkhtmltopdf.org/downloads.html

下载版本 Windows (MinGW) 0.12.4 32-bit / 64-bit for Windows XP/2003 or later; standalone

添加路径 D:\Program Files\wkhtmltopdf\bin

需要重新打开cmd以及notepad++。。。

 

pip install pdfkit

 

API https://pypi.python.org/pypi/pdfkit

定制options,搜索关键字 https://wkhtmltopdf.org/usage/wkhtmltopdf.txt

options = {
    page-size: Letter,
    margin-top: 0.75in,
    margin-right: 0.75in,
    margin-bottom: 0.75in,
    margin-left: 0.75in,
    encoding: "UTF-8",  #支持中文
    custom-header : [
        (Accept-Encoding, gzip)
    ]
    cookie: [
        (cookie-name1, cookie-value1),
        (cookie-name2, cookie-value2),
    ],
    no-outline: None
}

pdfkit.from_url(http://google.com, out.pdf, options=options)

 

 

3.背景知识

3.1url 相对路径 绝对路径

In [323]: urlparse.urljoin(https://doc.scrapy.org/en/latest/index.html, intro/overview.html)  #相当于 ./intro/overview.html,其中 . 指代当前文件夹 latest
Out[323]: https://doc.scrapy.org/en/latest/intro/overview.html

In [324]: urlparse.urljoin(https://doc.scrapy.org/en/latest/intro/overview.html, #walk-through-of-an-example-spider)  #当前网页某个tag id=walk-through-of-an-example-spider
Out[324]: https://doc.scrapy.org/en/latest/intro/overview.html#walk-through-of-an-example-spider

In [326]: urlparse.urljoin(https://doc.scrapy.org/en/latest/intro/overview.html, install.html)  #相当于 ./install.html
Out[326]: https://doc.scrapy.org/en/latest/intro/install.html

In [327]: urlparse.urljoin(https://doc.scrapy.org/en/latest/intro/overview.html, ../topics/commands.html)  # .. 指代当前文件夹intro的上一层文件夹latest
Out[327]: https://doc.scrapy.org/en/latest/topics/commands.html

 

https://doc.scrapy.org/en/latest/index.html

这一类官方文档一般页脚都为:

? Copyright 2008-2016, Scrapy developers. Revision 65ac0b06.

Built with Sphinx using a theme provided by Read the Docs.

3.2页面布局规律

  • 点击左上角 home 图标转到首页
  • 左边栏页面导航
    • <div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="main navigation">
      • <a class="reference internal" href="intro/overview.html">Scrapy at a glance</a></li>
  • 正文主体
    • <div class="rst-content">

3.2转换pdf时注意事项

  • 提取正文主体后,可以直接将  <div xxxx </div> 保存html,不需要补全 <html>
  • 图片链接相对路径需要转换为绝对路径,才会自动加载图片
    • <img alt="Inspecting elements with Firebug" src="../_images/firebug1.png" style="width: 913px; height: 600px;">
  • pdfkit.from_file 第一个参数 input 为 html文件路径列表,文件名不能是中文。。。
    • pdfkit.from_file(self.htmls_saved, self.netloc+‘.pdf‘, options=options)
  • pdf会根据<h1> <h2>等标题 tag 自动生成目录

4.实践代码

#!usr/bin/env python
#coding:utf-8

import os
import sys
import traceback
import re
import urlparse
import threading
import Queue

import requests
from scrapy import Selector
import pdfkit


s = requests.Session()
# s.headers.update({‘user-agent‘:‘Mozilla/5.0 (iPhone; CPU iPhone OS 9_3_5 like Mac OS X) AppleWebKit/601.1.46 (KHTML, like Gecko) Mobile/13G36 MicroMessenger/6.5.12 NetType/4G‘})
s.headers.update({user-agent:Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36})
# s.headers.update({‘Referer‘:‘https://servicewechat.com/wx55b926152a8c3bef/14/page-frame.html‘})
s.verify = False
s.mount(https://, requests.adapters.HTTPAdapter(pool_connections=1000, pool_maxsize=1000)) 
import copy
sp = copy.deepcopy(s)
proxies = {http: http://127.0.0.1:1080, https: https://127.0.0.1:1080}
sp.proxies = proxies 

from urllib3.exceptions import InsecureRequestWarning
from warnings import filterwarnings
filterwarnings(ignore, category = InsecureRequestWarning)

html_template = u"""
<!DOCTYPE html>

<html>
    <head>
        <meta charset="utf-8" />
        <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
    </head>
    <body>
        <!-- <center><h1>{title}</h1></center> -->
        {content}
    </body>
</html>
"""

# https://wkhtmltopdf.org/usage/wkhtmltopdf.txt
options = {
    page-size: A4,  # Letter
    minimum-font-size: 25,  ###
    # ‘image-dpi‘:1500, ###
    
    margin-top: 0.1in,  #0.75in
    margin-right: 0.1in,
    margin-bottom: 0.1in,
    margin-left: 0.1in,
    encoding: UTF-8,  #支持中文
    custom-header: [
        (Accept-Encoding, gzip)
    ],
    cookie: [
        (cookie-name1, cookie-value1),
        (cookie-name2, cookie-value2),
    ],
    outline-depth: 10,
}


class HTMLtoPDF(object):

    def __init__(self, seed_url, font_size=25, css_links=div[class="wy-menu wy-menu-vertical"] a::attr(href),
                css_content=div.rst-content, threads_count=30):
        self.seed_url = seed_url
        options[minimum-font-size] = font_size
        
        self.netloc = urlparse.urlparse(seed_url).netloc 
        print self.netloc
        self.folder = os.path.join(sys.path[0], self.netloc)
        self.folder_temp = os.path.join(sys.path[0], temp)
        for f in [self.folder, self.folder_temp]:
            if not os.path.isdir(f):
                os.mkdir(f)
        
        self.css_content = css_content
        self.css_links = css_links
        
        self.threads_count = threads_count
        # self.lock = threading.Lock()
        self.links_queue = Queue.Queue()     
        
        self.links_queue.put((0, self.seed_url))
        self.get_links()
        self.htmls_saved = [] 
        
        # 验证 re.findall(pattern, s)
        # <img alt="_images/chapt3_img05_IDE_open.png" class="align-center" src="_images/chapt3_img05_IDE_open.png">
        self.img_scr_pattern = re.compile(r(<img\s+[^>]*?src\s*=\s*")(?P<src>.*?)(".*?>))  #最后不能简写成 " ,否则结果缺 "

        
        # <a class="reference external" href="http://code.google.com/p/selenium/issues/detail?id=1008">issue 1008</a>
        # text为空,也能匹配到 m.group(4)=‘‘
        self.a_href_pattern = re.compile(r(<a\s+[^>]*?href\s*=\s*")(?P<href>.*?)(".*?>)(?P<text>.*?)(</a>))
        
        # http://www.seleniumhq.org/docs/ 合体。。。text为 <img alt="openqa.org logo" id="footerLogo" src="/images/openqa-logo.png"/>
        # <a href="http://openqa.org/"><img alt="openqa.org logo" id="footerLogo" src="/images/openqa-logo.png"/></a>

        
    def get_links(self):
        text = self.load_page(self.seed_url)
        sel = Selector(text=text)
        
        # [u‘#selenium-documentation‘,
        # u‘00_Note_to-the-reader.jsp‘,
        # u‘01_introducing_selenium.jsp‘,
        # u‘01_introducing_selenium.jsp#test-automation-for-web-applications‘,    

        # links = [re.sub(r‘#.*$‘,‘‘, i) for i in sel.css(‘div[class="toctree-wrapper compound"] a::attr(href)‘).extract()]        
        links = [re.sub(r#.*$,‘‘, i) for i in sel.css(self.css_links).extract()]        
        links_seen = [self.seed_url]
        
        for link in links:  #set(links) 会导致乱序,使用urls_seen 去重
            link_abs = urlparse.urljoin(self.seed_url, link)
            if link_abs not in links_seen:
                self.links_queue.put((str(len(links_seen)), link_abs))
                links_seen.append(link_abs)

    def save_html(self):
        while True:
            try:
                (num, url) = self.links_queue.get()
                text = self.load_page(url)
                
                title, content = self.parse_page(url, text)
                
                filename_cn = u{}_{}.html.format(num, re.sub(ur[^\u4e00-\u9fa5\w\s()_-], ‘‘, title))  #ur    
                filename = u{}_{}.html.format(num, re.sub(r[^\w\s()_-], ‘‘, title))  #os.path.abspath(‘en/abc.html‘)合成路径 不能是 /en。。
                
                with open(os.path.join(self.folder, filename_cn),wb) as fp:
                    fp.write(text.encode(utf-8,replace))
                f = os.path.join(self.folder_temp, filename)
                with open(f,wb) as fp:
                    fp.write(content.encode(utf-8,replace)) 
                    # fp.write(html_template.format(content=content, title=title).encode(‘utf-8‘,‘replace‘))
                    self.htmls_saved.append(f)
                    print {}/{}.format(len(self.htmls_saved), self.links_queue.qsize())
                    
                self.links_queue.task_done()
            except Exception as err:
                print {} {} {}.format(url, err, traceback.format_exc())
                
    def run(self):
        threads = []
        for i in range(self.threads_count):
            t = threading.Thread(target=self.save_html)
            threads.append(t)

        for t in threads:
            t.setDaemon(True) 
            t.start() 
            
        self.links_queue.join()
        print load done
        
        def func(filename):
            _, filename =os.path.split(filename)
            return int(filename[:filename.index(_)])
        
        self.htmls_saved.sort(key=lambda x:func(x))
        pdfkit.from_file(self.htmls_saved, self.netloc+.pdf, options=options)
        print self.netloc, pdf done
        

    def load_page(self, url):
            
        resp = sp.get(url)  ###############

        if resp.encoding == ISO-8859-1:
            encodings = requests.utils.get_encodings_from_content(resp.content)  #re.compile(r‘<meta.*?charset
            if encodings:
                resp.encoding = encodings[0]
            else:
                resp.encoding = resp.apparent_encoding  #models.py  chardet.detect(self.content)[‘encoding‘]
            # print ‘ISO-8859-1 changed to %s‘%resp.encoding
            
        return resp.text   

    def parse_page(self, url, text):
        sel = Selector(text=text)

        title = sel.css(head title::text).extract_first() or ‘‘  #固定css
        content = sel.css(self.css_content).extract_first() or ‘‘  #‘div.rst-content‘
        
        # sel = sel.css("div#rst-content")[0]  ###缩小范围
        content = self.clean_content(content)
        content = self.modify_content(url, content)  
        
        return title, content
  
    def clean_content(self, content):
        sel = Selector(text=content)
        # content = content.replace(sel.css(‘div#codeLanguagePreference‘).extract_first(), ‘‘) #可能是None
        for div in sel.css(div#codeLanguagePreference).extract():
            content = content.replace(div, ‘‘)
        
        for lang in [java, csharp, ruby, php, perl, javascript]:
            for div in sel.css(div.highlight-%s%lang).extract():
                # print len(content)
                content = content.replace(div, ‘‘)
                
        return content
        
    def modify_content(self, url, content):
        # m.group(1)=‘abc‘ SyntaxError: can‘t assign to function call 不能直接赋值
        
        # https://doc.scrapy.org/en/latest/topics/firebug.html
        # ../_images/firebug1.png
        # 异常 urlparse.urljoin(self.seed_url, src)
        
        # r‘(<img\s+[^>]*?src\s*=\s*")(?P<src>.*?)(".*?>)‘
        def func_src(m):
            src = m.group(src)  #别名
            if not src.startswith(http):
                src = urlparse.urljoin(url, src)
            return u{}{}{}.format(m.group(1), src, m.group(3))

        content = re.sub(self.img_scr_pattern, func_src, content)

        
        # re.compile(r‘(<a\s+[^>]*?href\s*=\s*")(?P<href>.*?)(".*?>)(?P<text>.*?)(</a>)‘)
        def func_href(m):
            href = m.group(href)
            text = m.group(text)
            if not href.startswith(#):
                if not href.startswith(http):
                    href = urlparse.urljoin(url, href)
                text = u{text} ({href}).format(text=text, href=href)
            return u{g1}{href}{g3}{text}{g5}.format(g1=m.group(1), g3=m.group(3), g5=m.group(5), href=href, text=text)
            #m.string是content全文。。。也不能 return m

        content = re.sub(self.a_href_pattern, func_href, content)   

        return content
        
    
    def modify_content2(self, url, content):
        sel = Selector(text=content)

        # 修改图片链接为绝对链接,否则pdf无法图片        
        # <img alt="_images/chapt3_img05_IDE_open.png" class="align-center" src="_images/chapt3_img05_IDE_open.png">
        for i in sel.css(img[src]):
            tag = i.extract()
            src = i.xpath(./@src).extract_first()
            if not src.startswith(http):
                src_abs = urlparse.urljoin(url, src)
                # print src, src_abs
                tag_new = tag.replace(src, src_abs)     
                content = content.replace(tag, tag_new)  #可能alt(同src...)
                
        # a href 的text添加href信息
        # <a class="reference external" href="http://code.google.com/p/selenium/issues/detail?id=1008">issue 1008</a>
        for i in sel.css(a[href]):
            tag = i.extract()
            href = i.xpath(./@href).extract_first()
            text = i.xpath(./text()).extract_first()
            
            # 补全内部链接,忽略本页面的#定位
            if not href.startswith(http) and not href.startswith(#):
                href_abs = urlparse.urljoin(url, href)
                # print href, href_abs
                tag_new = tag.replace(href, href_abs)
            else:
                href_abs = href
                tag_new = tag
                
            # 图标链接,如果text为None,replace表现异常
            if text and not href.startswith(#):
                text_new = u{} ({}).format(text, href_abs)
                # print text.encode(‘gbk‘,‘replace‘), text_new.encode(‘gbk‘,‘replace‘)
                tag_new = tag_new.replace(text, text_new)          
            
            # 保证整体替换   
            content = content.replace(tag, tag_new)  
        
        return content



if __name__ == __main__:
    url = https://doc.scrapy.org/en/latest/index.html
    # obj = HTMLtoPDF(url)  
    
    url = http://python3-cookbook.readthedocs.io/zh_CN/latest/index.html
    # obj = HTMLtoPDF(url, font_size=20, css_links=‘div[class="toctree-wrapper compound"] a::attr(href)‘) 
    
    url = http://www.seleniumhq.org/docs/
    obj = HTMLtoPDF(url, css_links=div#selenium-documentation a::attr(href), css_content=div#mainContent)  
    
    obj.run()

  

 

html2doc

标签:error:   rom   headers   就是   oct   mob   定位   _id   agent   

原文地址:http://www.cnblogs.com/my8100/p/7738366.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!