[初学Python]利用某网站的功能写一个磁力链转种子工具

孔子曰：学而时习之~

初学python，一边看语法，一边看参考例程，写了一个小程序，做个存档，否则很容易忘记。

这次的目标是利用某种子下载网站，编写一个磁力链下载种子文件的工具。

常见的种子文件磁力链接的格式是这样的：magnet:?xt=urn:btih:XXX

其中的XXX就是种子文件的哈希值，而这个网站可以输入哈希值，下载对应的种子文件，

但是该网站广告内容太多了，用浏览器打开不太方便呀~，具体是哪个网站我就不说了，哈哈~~~，自己看代码吧。

# 磁力链接转种子文件
# QQ:446252221 <[email protected]>
import io
import os
import time
import urllib.request
import urllib.parse
import html.parser

USER_AGENT = "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/68.0.3440.106 Safari/537.36"

#HTML解析
class html_dumper(html.parser.HTMLParser):
    
    def __init__(self):
        html.parser.HTMLParser.__init__(self)
        self.key = ""
        
    def handle_starttag(self, tag, attrs):
        if tag == "input":
            if attrs[1][1] == "key":
                self.key = attrs[2][1]
        return None
        
    def handle_data(self, data):
        return None
            
    def handle_comment(self, data):
        return None
        
    def get_key(self):
        return self.key
        
#第1步获取key
def step1_get_key(hash):
    req_url = "http://www.storetorrents.com/hash/" + hash
    req = urllib.request.Request(url=req_url, method="GET")
    req.add_header("User-Agent", USER_AGENT)
    try:
        respon = urllib.request.urlopen(req)
        #print("第1步:", respon.status)
        respon_cook = respon.getheader("Set-Cookie")
        respon_body = respon.read()
        dumper = html_dumper()
        dumper.feed(respon_body.decode("utf-8"))
        #print("key:", dumper.get_key())
        return dumper.get_key()
    except BaseException as e:
        print("urlopen error:", e)
        return None
        
#第2步获取captcha
def step2_get_captcha():
    req_url = "https://storetorrents.xyz/captcha"
    req = urllib.request.Request(url=req_url, method="GET")
    req.add_header("User-Agent", USER_AGENT)
    #req.add_header("Cookie", "__cfduid=dac7712102a83c5d8e2d3fb4ec6f407031553505032; PHPSESSID=25d2d5kd6erdqbo1fv4o4e8vs2")
    try:
        respon = urllib.request.urlopen(req)
        #print("第2步:", respon.status)
        respon_body = respon.read()
        respon_cook = respon.getheader("Set-Cookie")
        file_name = "captcha.jpg"
        file_obj = open(file_name, "wb")
        file_obj.write(respon_body)
        file_obj.close()
        os.system(file_name)
        #print("cookie:", respon_cook)
        cookie_list = respon_cook.split(";", -1)
        cookie = cookie_list[0] + ";" + cookie_list[5].split(",", 1)[1]
        #print("cookie:", cookie)
        return cookie
    except BaseException as e:
        print("urlopen error:", e)
        return None
    
    
#第3步下载文件
def step3_download(key, hash, captcha, cookie):
    req_url = "https://storetorrents.xyz/download"
    req_values = {}
    req_values["key"] = key
    req_values["infohash"] = hash
    req_values["captcha"] = captcha
    req_body = urllib.parse.urlencode(req_values)
    #print("key encode:", req_body)
    req = urllib.request.Request(url=req_url, method="POST", data=bytes(req_body, encoding="utf8"))
    req.add_header("User-Agent", USER_AGENT)
    req.add_header("Cookie", cookie)
    try:
        respon = urllib.request.urlopen(req)
        #print("第3步:", respon.status)
        type = respon.getheader("Content-Type")
        #print("Type:", type)
        if type.startswith("text/html"):
            print(respon.read().decode("utf-8"))
            return None
        respon_body = respon.read()
        file_name = hash + ".torrent"
        file_obj = open(file_name, "wb")
        file_obj.write(respon_body)
        file_obj.close()
        print("下载完成!")
    except BaseException as e:
        print("urlopen error:", e)
    return None
    
#入口
os.system("title 磁力转换工具")
os.system("echo off")
while(True):
    hash = input("magnet:?xt=urn:btih:")
    key = step1_get_key(hash)
    if key == None:
        print("未找到该资源!")
        os.system("pause")
        continue
    cookie = step2_get_captcha()
    if cookie == None:
        print("获取验证码失败!")
        os.system("pause")
        continue
    code = input("请输入验证码:")
    step3_download(key, hash, code, cookie)
    os.system("pause")
    

 

简单解释一下代码内容：

整体流程是:

1.访问A页面，解析HTML得到其中的重要参数key.

2.访问B页面，得到cookie和验证码.

3.输入验证码：由于python显示图片要用第三方库实现，所以暂时没有用代码实现。取而代之的是下载图片，然后用Windows系统图片查看器打开它，然后人工识别验证码。

4.访问C页面，下载种子文件，这个页面使用POST方式请求，并附加B页面得到的cookie和3个参数：由A页面得到的key，由B页面得到的验证码，和种子文件的哈希值。

5.下载C页面返回的数据写入到文件，这就是我们要的种子文件啦。