python 网站爬虫下载在线盗墓笔记小说到本地的脚本

妖狐艹你老母 2022-08-13 06:00 140阅读 0赞

最近闲着没事想看小说，找到一个全是南派三叔的小说的网站，决定都下载下来看看，于是动手，在很多QQ群里高手的帮助下（本人正则表达式很烂，程序复杂的正则都是一些高手指导的），花了三四天写了一个脚本

需要 BeautifulSoup 和 requests 两个库

（我已经把注释写得尽量详细）

这个程序的执行速度很慢，求高手告诉我优化的方法！！

#-*-coding:utf8-*-
    
    from bs4 import BeautifulSoup
    import requests
    import re
    import os
    
    
    #打开网页将所需的URL读取出来放进一个列表中
    r = requests.get('http://www.nanpaisanshu.org/').content  #打开要读取的网页
    content=BeautifulSoup(r).findAll('a',href=re.compile(r'\Ahttp://www.nanpaisanshu.org/[a-z]+\Z')) #在网页中找到需要的信息
    
    sc = str(content) #转换为string类型
    
    lists=[]
    lists = sc.split(',')   
    lists = list(set(lists)) #删除列表中重复信息
    
    lisy=[]
    
    
    for line in lists:
        p=line.split('"')[1]  #按 " 分割，取出需要的信息写进数组
        lisy.append(p)        #这里已经拥有需要的url
        #print p
    #print lisy
    
    
    #把读取到的URL遍历打开，将所有网页保存到html文件中
    
    s = os.getcwd()#当前路径
    
    d = os.sep  #系统分隔符
    
    namef='aaa' #文件加名称
    
    #b = os.path.exists( s+d+namef) #判断是存在
    
    f=os.path.exists(s+d+namef) #判断是存在
    
    if f==False:
        os.mkdir(s+d+namef)  #如果文件夹不存在就新建一个
    else:
        print u'已经存在'+namef
    
    filenm = s+d+namef+d    #路径
    
    i=1
    for line in lisy:
        r = requests.get(line)   #遍历打开所有url
        print r.content
        print '\n'
        tfile=open(filenm+'neirong'+str(i)+'.html','w')
        i=i+1
        tfile.write(r.content) #将网页内容写入文件
    
    #将URL文件中的符合要求的URL读取出来写进一个txt文件中
    for i in range(1,len(lisy)+1):
        fp = open(filenm+'neirong'+str(i)+'.html', "r")
        of = open(filenm+'neirong'+str(i)+'.txt','w')  
        content = fp.read()   #将文件内容读取
    
        p=re.compile(r'http://www\.nanpaisanshu\.org/.*?\.html') #正则匹配
        
        #print p.findall(content)
    
        #print type(p.findall(content))
    
        for line in p.findall(content):  
            #print line+'\n'
            #if line !='http://www.nanpaisanshu.org/9701.html':
            of.write(line+'\n')  #将匹配到的文件写入另一个文件中
            #else:
                #continue
    
            #of.write(str(p.findall(content)))
    
    #关闭文件
    of.close()
    fp.close()
    tfile.close()
    
    
    #将txt
    
    for i in range(1,len(lisy)+1):
        ot=open(filenm+'neirong'+str(i)+'.txt','r')
        outfile=open(filenm+'quanbu'+str(i)+'.txt','a+')
    
    
        li=[]
        for line in ot:
            line = line.replace('\n','')
            li.append(line)   #将url文件中的数据放进列表中
    
        li = sorted(li)  #给列表排序
    
        for line in li:
            print line
            #line = line.replace('\n','')
            r = requests.get(line).content  #遍历打开所有url
            title=BeautifulSoup(r).find("div",{'class':"post_title"}).h2   #取出标题
            content=BeautifulSoup(r).findAll("div",{'class':"post_entry"}) #取出内容
            sti=str(title).replace('<h2>','').replace('</h2>','')  #处理标题，只保留文字
    
            #处理内容，只保留文字
            scon = str(content).replace('<p>','  ').replace('</p>','  ').replace('<br/>','\n')
            #print str(urllist)
            scon = re.sub("<.*>", "", scon)
            scon = re.sub("(.*?);","",scon) 
            #scon = scon.strip()
            scon = '\n'.join(scon.split())
    
            print scon
            outfile.write(sti+'\n'+line+'\n'+scon) #将标题和内容写进文件中
        #i=i+1
        #print 
    #print urllist
    
    print '=========================下载结束======================='
    
    
    #关闭文件
    outfile.close()
    ot.close()
    
    
    
    #取出指定文件夹下的所有文件名
    targetDir=s+d+namef
    for line in os.listdir(targetDir):
    
        p=re.compile(r'neirong[0-9]{1}') #用正则匹配
        if p.match(line)!=None:
            print "需要删除的文件"+s+d+namef+d+line+'!!'
            os.remove(s+d+namef+d+line)  #匹配成功就删除这个文件，os.remove()中需要完整路径
        else:
            print '保留文件！'
            continue

有时候会显示连接失败，然后程序报错，应该判断一下  requests.get(url).status\_code != 200  不过我加了以后发现运行更慢，每个页面都判断，汗，可能是因为我这里网速几K的原因才会异常

下面是修改后的完善版，慎用，速度极其的慢，判断的东西和次数增加的缘故：

#-*-coding:utf8-*-
    
    #下载盗墓笔记小说
    #2014-10-14
    #ZJL
    
    from bs4 import BeautifulSoup
    import requests
    import re
    import os
    
    
    #打开网页将所需的URL读取出来放进一个列表中
    r = requests.get('http://www.nanpaisanshu.org/').content  #打开要读取的网页
    content=BeautifulSoup(r).findAll('a',href=re.compile(r'\Ahttp://www.nanpaisanshu.org/[a-z]+\Z')) #在网页中找到需要的信息
    
    sc = str(content) #转换为string类型
    
    lists=[]
    lists = sc.split(',')   
    lists = list(set(lists)) #删除列表中重复信息
    
    lisy=[]
    
    
    for line in lists:
        p=line.split('"')[1]  #按 " 分割，取出需要的信息写进数组
        lisy.append(p)        #这里已经拥有需要的url
        #print p
    #print lisy
    
    
    #把读取到的URL遍历打开，将所有网页保存到html文件中
    
    s = os.getcwd()#当前路径
    
    d = os.sep  #系统分隔符
    
    namef='aaa' #文件加名称
    
    #b = os.path.exists( s+d+namef) #判断是存在
    
    f=os.path.exists(s+d+namef) #判断是存在
    
    if f==False:
        os.mkdir(s+d+namef)  #如果文件夹不存在就新建一个
    else:
        print u'已经存在'+namef
    
    filenm = s+d+namef+d    #路径
    
    i=1
    for line in lisy:
        r = requests.get(line)   #遍历打开所有url
        print r.content
        print '\n'
        tfile=open(filenm+'neirong'+str(i)+'.html','w')
        i=i+1
        tfile.write(r.content) #将网页内容写入文件
    
    #将URL文件中的符合要求的URL读取出来写进一个txt文件中
    for i in range(1,len(lisy)+1):
        fp = open(filenm+'neirong'+str(i)+'.html', "r")
        of = open(filenm+'neirong'+str(i)+'.txt','w')  
        content = fp.read()   #将文件内容读取
    
        p=re.compile(r'http://www\.nanpaisanshu\.org/.*?\.html') #正则匹配
        
        #print p.findall(content)
    
        #print type(p.findall(content))
    
        for line in p.findall(content):  
            #print line+'\n'
            #if line !='http://www.nanpaisanshu.org/9701.html':
            of.write(line+'\n')  #将匹配到的文件写入另一个文件中
            #else:
                #continue
    
            #of.write(str(p.findall(content)))
    
    #关闭文件
    of.close()
    fp.close()
    tfile.close()
    
    
    #将txt
    
    for i in range(1,len(lisy)+1):
        ot=open(filenm+'neirong'+str(i)+'.txt','r')
        if os.path.exists(filenm+'quanbu'+str(i)+'.txt')==True:
            print "已经存在"+filenm+'quanbu'+str(i)+'.txt'+'会先删除再创建'
            os.remove(filenm+'quanbu'+str(i)+'.txt')
            outfile=open(filenm+'quanbu'+str(i)+'.txt','a+')   #防止第二次下载时内容在文件后面追加(不知道会不会重新新建文件覆盖掉原来的文件所以这么做)
    
        else:
            print "新建"+filenm+'quanbu'+str(i)+'.txt'
            outfile=open(filenm+'quanbu'+str(i)+'.txt','a+')
    
    
        
        li=[]
        for line in ot:
            line = line.replace('\n','')
            li.append(line)   #将url文件中的数据放进列表中
    
        li = sorted(li)  #给列表排序
    
        for line in li:
        #print line
            #line = line.replace('\n','')
    
            if requests.get(line).status_code != 200:
                print '因为网络原因，这个章节为空!'
                outfile.write('因为网络原因，这个章节为空')   #判断网络连接情况，防止连接失败后程序报错
            elif requests.get(line).status_code == 200:
                print '连接成功！'
                r = requests.get(line).content  #遍历打开所有url
                title=BeautifulSoup(r).find("div",{'class':"post_title"}).h2   #取出标题
                content=BeautifulSoup(r).findAll("div",{'class':"post_entry"}) #取出内容
                sti=str(title).replace('<h2>','').replace('</h2>','')  #处理标题，只保留文字
    
                #处理内容，只保留文字
                scon = str(content).replace('<p>','  ').replace('</p>','  ').replace('<br/>','\n')
                #print str(urllist)
                scon = re.sub("<.*>", "", scon)
                scon = re.sub("(.*?);","",scon) 
                #scon = scon.strip()
                scon = '\n'.join(scon.split())
    
                print scon
                outfile.write(sti+'\n'+line+'\n'+scon) #将标题，链接，内容写进文件中
            #i=i+1
            #print 
            #print urllist
    
    print '=========================下载结束======================='
    
    
    #关闭文件
    outfile.close()
    ot.close()
    
    
    
    #取出指定文件夹下的所有文件名
    targetDir=s+d+namef
    for line in os.listdir(targetDir):
    
        p=re.compile(r'neirong[0-9]{1}') #用正则匹配
        if p.match(line)!=None:
            print "需要删除的文件"+s+d+namef+d+line+'!!'
            os.remove(s+d+namef+d+line)  #匹配成功就删除这个文件，os.remove()中需要完整路径
        else:
            print '保留文件！'
            continue