函数优化

fineconey

有一个问题请教各位.从开始学python就一直不习惯试用函数.  可能是因为不够直接感性,感觉效率差不多.

所以近期打算评估一下用函数和不用函数在时间上有什么差异. 但是想改的时候,突然发现,不会了,

所以,请诸位大神根据我的代码,改成函数样式

(纯小白,获取某PT网站的资源,练习试用.cookie已更改)

代码如下:

1. # coding:utf-8
   
2. import os
   
3. import re
   
4. import time
   
5. import requests
   
6. from bs4 import BeautifulSoup
   
7. n=0
   
8. headers={
   
9. 'cookie': '__cfduid=ddd8dd90defddd80d1b44a39e73fbeb371f1582866913; c_secure_uid=MzgzNDM%3D; c_secure_pass=be6febb679b71a73e7246d7e68cca7a1; c_secure_ssl=bm9wZQ%3D%3D; c_secure_tracker_ssl=eWVhaA%3D%3D; c_secure_login=bm9wZQ%3D%3D',
   
10. 'referer': 'https://pt.btschool.club/index.php',
    
11. 'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.132 Safari/537.36 Edg/80.0.361.66'
    
12. }
    
13. for i in range(170):
    
14.     url='https://pt.btschool.club/torrents.php?inclbookmarked=0&incldead=1&spstate=0&page={}'.format(i)
    
15.     res=requests.get(url,headers=headers)
    
16.     res.encoding="utf-8"
    
17.     urls=re.findall('</tr></table></td><td class="rowfollow"><b><a href="(.*?)&amp;cmtpage',res.text,re.S)
    
18.     for url in urls:
    
19.         print("正在计数:",n)
    
20.         url=url.replace("amp;","")
    
21.         url="https://pt.btschool.club/"+url
    
22.         r = requests.get(url, headers=headers)
    
23.         r.encoding = "utf-8"
    
24.         html = r.text
    
25.         soup = BeautifulSoup(html, "html.parser")
    
26.         titleA = soup.find("a", class_="index").get_text()
    
27.         titleB = soup.find("td", class_="rowfollow", valign='top').get_text()
    
28.         size = re.findall("<b><b>大小：</b></b>(.*?)&nbsp;&nbsp;&nbsp;<b>类型:</b>", html, re.S)[0]
    
29.         leixing = re.findall("<b>类型:</b>&nbsp;(.*?)&nbsp;&nbsp;&nbsp;<b>", html, re.S)[0]
    
30.         downurl = soup.find_all("td", class_="rowfollow", valign="top", align="left")[3].find("a").get("href")
    
31.         data = soup.find_all("div", id="kdescr")[0].get_text()
    
32.         data = data.replace(
    
33.             "免责声明 本站所有资源均为网友上传，仅供测试宽带所用，严禁用于商业用途，否则产生的一切后果将由您自己承担！本站将不对本站的资源负任何法律责任！如支持作品，请购买正版！ 引用 您的保种是PT站长久发展的重要保证！只要片子在硬盘上不删，请自觉做到开机即做种！自觉保种可使资源的有效期无限延长、下载速度达到极限，同时也可以为你赚得上传流量和魔力值，方便你我他~快乐下载，分享至美！感谢您对比特校园PT的支持~",
    
34.             "")
    
35.         print(n, '即将记录__%s__这个文件' % titleA[11:21])
    
36.         with open("PT_btshcool.txt", "a", encoding="utf-8") as f:
    
37.             try:
    
38.                 f.write(str(n))
    
39.                 f.write("■")
    
40.                 f.write(leixing)
    
41.                 f.write("■")
    
42.                 f.write(titleA)
    
43.                 f.write("■")
    
44.                 f.write(titleB)
    
45.                 f.write("■")
    
46.                 f.write(size)
    
47.                 f.write("■")
    
48.                 f.write(downurl)
    
49.                 f.write("\n")
    
50.                 n += 1
    
51.                 with open("%s.txt" % titleA[11:], "w", encoding="utf-8") as f:
    
52.                     f.write(data)
    
53.                 time.sleep(2)
    
54.             except:
    
55.                 pass
    

复制代码

一个账号

1. # coding:utf-8
   
2. import os
   
3. import re
   
4. import time
   
5. import requests
   
6. from bs4 import BeautifulSoup
   
7. 
   
8. def func():
   
9.     n=0
   
10.     headers={
    
11.     'cookie': '__cfduid=ddd8dd90defddd80d1b44a39e73fbeb371f1582866913; c_secure_uid=MzgzNDM%3D; c_secure_pass=be6febb679b71a73e7246d7e68cca7a1; c_secure_ssl=bm9wZQ%3D%3D; c_secure_tracker_ssl=eWVhaA%3D%3D; c_secure_login=bm9wZQ%3D%3D',
    
12.     'referer': 'https://pt.btschool.club/index.php',
    
13.     'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.132 Safari/537.36 Edg/80.0.361.66'
    
14.     }
    
15.     for i in range(170):
    
16.         url='https://pt.btschool.club/torrents.php?inclbookmarked=0&incldead=1&spstate=0&page={}'.format(i)
    
17.         res=requests.get(url,headers=headers)
    
18.         res.encoding="utf-8"
    
19.         urls=re.findall('</tr></table></td><td class="rowfollow"><b><a href="(.*?)&amp;cmtpage',res.text,re.S)
    
20.         for url in urls:
    
21.             print("正在计数:",n)
    
22.             url=url.replace("amp;","")
    
23.             url="https://pt.btschool.club/"+url
    
24.             r = requests.get(url, headers=headers)
    
25.             r.encoding = "utf-8"
    
26.             html = r.text
    
27.             soup = BeautifulSoup(html, "html.parser")
    
28.             titleA = soup.find("a", class_="index").get_text()
    
29.             titleB = soup.find("td", class_="rowfollow", valign='top').get_text()
    
30.             size = re.findall("<b><b>大小：</b></b>(.*?)&nbsp;&nbsp;&nbsp;<b>类型:</b>", html, re.S)[0]
    
31.             leixing = re.findall("<b>类型:</b>&nbsp;(.*?)&nbsp;&nbsp;&nbsp;<b>", html, re.S)[0]
    
32.             downurl = soup.find_all("td", class_="rowfollow", valign="top", align="left")[3].find("a").get("href")
    
33.             data = soup.find_all("div", id="kdescr")[0].get_text()
    
34.             data = data.replace(
    
35.                 "免责声明 本站所有资源均为网友上传，仅供测试宽带所用，严禁用于商业用途，否则产生的一切后果将由您自己承担！本站将不对本站的资源负任何法律责任！如支持作品，请购买正版！ 引用 您的保种是PT站长久发展的重要保证！只要片子在硬盘上不删，请自觉做到开机即做种！自觉保种可使资源的有效期无限延长、下载速度达到极限，同时也可以为你赚得上传流量和魔力值，方便你我他~快乐下载，分享至美！感谢您对比特校园PT的支持~",
    
36.                 "")
    
37.             print(n, '即将记录__%s__这个文件' % titleA[11:21])
    
38.             with open("PT_btshcool.txt", "a", encoding="utf-8") as f:
    
39.                 try:
    
40.                     f.write(str(n))
    
41.                     f.write("■")
    
42.                     f.write(leixing)
    
43.                     f.write("■")
    
44.                     f.write(titleA)
    
45.                     f.write("■")
    
46.                     f.write(titleB)
    
47.                     f.write("■")
    
48.                     f.write(size)
    
49.                     f.write("■")
    
50.                     f.write(downurl)
    
51.                     f.write("\n")
    
52.                     n += 1
    
53.                     with open("%s.txt" % titleA[11:], "w", encoding="utf-8") as f:
    
54.                         f.write(data)
    
55.                     time.sleep(2)
    
56.                 except:
    
57.                     pass
    
58. 
    
59. func()

复制代码

一个账号

直接包起来就行了