【Python】torrentParser1.04 增加获得磁力链URI功能

时间:2023-03-08 15:07:56
【Python】torrentParser1.04 增加获得磁力链URI功能

代码:

#------------------------------------------------------------------------------------
# torrentParser1.04,用于解析torrent文件
# 增加了函数getMegnet,能将种子文件转磁力链
# 2018年5月13日
#------------------------------------------------------------------------------------
from bencode import bdecode,bencode
import time
import hashlib
import base64

#-------------------------------------
# torrentParser类
#-------------------------------------
class torrentParser(object):
    # 构造函数
    def __init__(self,filePathname):
        self.filePathname=filePathname
        with open(filePathname,'rb') as fObj:
            self.fileDic=bdecode(fObj.read()) 

    # 得到文件路径名
    def getFilepathname(self):
        return self.filePathname

    # 得到所有键值
    def getKeys(self):
        return self.fileDic.keys()

    # 得到全部内容
    def getAllContent(self):
        return self.fileDic;

    # 得到文件结构
    def getStructure(self):
        retval=''
        layer=0

        for key in self.fileDic.keys():
            value=self.fileDic[key]
            retval=retval+self.getNextNode(key,value,layer)

        return retval

    # 向下递归查找文件结构,
    def getNextNode(self,key,value,layer):
        retval="";
        layer+=1

        if type(value)==type({}) and len(value.keys())>0:
            for i in range(1,layer+1):
                retval=retval+"\t"
            retval=retval+str(key)+"\n"

            for k in value.keys():
                v=value[k]
                retval=retval+ self.getNextNode(k,v,layer)
        elif type(value)==type([]) and len(value)>0:
            for i in range(1,layer+1):
                retval=retval+"\t"
            retval=retval+str(key)+"\n"

            arr=value

            for it in arr:
                if type(it)==type({}) and len(it.keys())>0:
                    for nk in it.keys():
                        nv=it[nk]

                        retval=retval+''+self.getNextNode(nk,nv,layer)
        else:
            for i in range(1,layer+1):
                retval=retval+"\t"

            showValue=str(value)[0:50] # 显示的值

            retval=retval+str(key)+":"+showValue+"\n"

        return retval

    # 获得tracker服务器的URL
    def getAnnounce(self):
        if 'announce' in self.fileDic:
            return self.fileDic['announce'].decode('utf-8','ignore')
        return ''

    # 获得tracker服务器的URL列表
    def getAnnounceList(self):
        retval=[]

        if 'announce-list' in self.fileDic:
            arr=self.fileDic['announce-list']

            for childArr in arr:

                if type(childArr)==type([]):
                    for item in childArr:
                        retval.append(item.decode('utf-8','ignore'))
                else:
                    retval.append(childArr.decode('utf-8','ignore'))

        return retval

    # 得到制作日期
    def getCreateTime(self):
        if 'creation date' in self.fileDic:
            unixTimestamp=self.fileDic['creation date']
            firmalTime = time.localtime(unixTimestamp)
            dt = time.strftime('%Y-%m-%d %H:%M:%S', firmalTime)

            return dt
        else:
            return ''

    # 获得编码方式
    def getEncoding(self):
        if 'encoding' in self.fileDic:
            return self.fileDic['encoding'].decode('utf-8','ignore')
        return ''

    # 是否包含多个文件
    def hasMultiFiles(self):
        if 'files' in self.fileDic['info']:
            return True
        else:
            return False

    # 获得文件名
    def getTitle(self):
        arr=[]
        info = self.fileDic['info'] 

        if 'name.utf-8' in info:
            arr=info['name.utf-8']
        else:
            arr = info['name'] 

        #print(str(arr))

        return arr.decode('utf-8','ignore') 

    # 获得备注(可选项)
    def getComment(self):
        if 'comment' in self.fileDic:
            return self.fileDic['comment'].decode('utf-8','ignore')
        return ''

    # 获得创建者(可选项)
    def getCreatedBy(self):
        if 'created by' in self.fileDic:
            return self.fileDic['created by'].decode('utf-8','ignore')
        return ''

    # 多文件的情况下,得到多个文件的个数
    def getFileCount(self):
        return len(self.fileDic['info']['files'])

    # 多文件的情况下,获得所有文件
    def getFiles(self):
        files=[]

        for item in self.fileDic['info']['files']:
            file={}

            for key in item.keys():
                value=item.get(key)

                if key=='path':
                    #print('1.'+str(value))
                    #print('10.'+str(value[0]))
                    path=value[0].decode('utf8','ignore')
                    value=path
                if key=='path.utf-8':
                    #print('2.'+str(value))
                    #print('20.'+str(value[0]))
                    path=value[0].decode('utf8','ignore')
                    value=path

                file[key]=value

            files.append(file)

        return files

    # 单文件情况下,取文件名
    def getSingleFileName(self):
        return self.getTitle();

    # 单文件情况下,取文件长度
    def getSingleFileLength(self):
        return self.fileDic['info']['length']

    # 单文件情况下,取文件md5sum
    def getSingleFileMd5sum(self):
        if 'md5sum' in self.fileDic['info']:
            return self.fileDic['info']['md5sum'].decode('utf-8','ignore')
        else:
            return ''

    # 单文件情况下,取文件长度
    def getSingleFilePieceLength(self):
        return self.fileDic['info']['piece length']

    # 单文件情况下,取文件pieces
    def getSingleFilePieces(self):
        return self.fileDic['info']['pieces']

    # 得到文件简报
    def getBrief(self):
        retval=""
        retval=retval+"File:"+self.filePathname+"\n"
        retval=retval+"announce:"+self.getAnnounce()+"\n"

        arr=self.getAnnounceList()
        if(len(arr)>0):
            retval=retval+"announce list:"+"\n"

            for it in arr:
                retval=retval+"\t"+it+"\n"    

        retval=retval+"Create time:"+self.getCreateTime()+"\n"
        retval=retval+"Ecoding:"+self.getEncoding()+"\n"
        retval=retval+"Title:"+self.getTitle()+"\n"
        retval=retval+"Comment:"+self.getComment()+"\n"
        retval=retval+"Created by:"+self.getCreatedBy()+"\n"

        hasMulti=self.hasMultiFiles()
        retval=retval+"has multi files:"+str(hasMulti)+"\n"
        if hasMulti==True:
            retval=retval+"[多文件结构]"+"\n"

            retval=retval+"包含文件个数为:"+str(self.getFileCount())+"\n"
            retval=retval+"Files:"+"\n"

            files=self.getFiles();
            index=1
            for item in files:
                retval=retval+"\tfile#"+str(index)+"\n"    

                for key in item.keys():
                    value=item.get(key)
                    retval=retval+"\t\t"+str(key)+":"+str(value)+"\n"
                retval=retval+"\n"    

                index=index+1
        else:
            retval=retval+"[单文件结构]"+"\n"
            retval=retval+"文件名为:"+self.getSingleFileName()+"\n"
            retval=retval+"文件长度:"+str(self.getSingleFileLength())+"byte\n"
            retval=retval+"文件md5sum:"+self.getSingleFileMd5sum()+"\n"
            retval=retval+"文件块长度:"+str(self.getSingleFilePieceLength())+"byte\n"

        return retval

    # 得到磁力链
    def getMegnet(self):
        info=self.fileDic['info']
        arr=bencode(info);
        diggest=hashlib.sha1(arr).digest()
        b32hash=base64.b32encode(diggest)
        return 'magnet:?xt=urn:btih:'+str(b32hash,'utf-8')

#-------------------------------------
# 入口
#-------------------------------------
def main():
    tp=torrentParser(filePathname='./5.torrent')
    print('文件名='+tp.getFilepathname())
    print('文件结构:\n'+tp.getStructure())
    print('文件简报:\n'+str(tp.getBrief()))
    #print('文件内容:\n'+str(tp.getAllContent()))
    print('磁力链:\n'+str(tp.getMegnet()))

# Start
main()

输出:

C:\Users\horn1\Desktop\python\46-torrentParser>python torrentParser.py
文件名=./5.torrent
文件结构:
        announce:b'udp://tracker.openbittorrent.com:80'
        announce-list
        comment:b'This torrent is downloaded from btkitty.me , sto
        creation date:1496866967
        encoding:b'UTF-8'
        info
                length:1385343970
                name:b'SIRO-1690 \xa4\xca\xa4\xca \xcc\xd2\xb9\xc8\xbb\
                name.utf-8:b'SIRO-1690 \xe3\x81\xaa\xe3\x81\xaa \xe6\xa1\x83\
                piece length:1048576
                pieces:b'S\x06\xa4\x18\xed\x9d\xa5\xd9r\xc4^O\x1b\xefs\xe
        publisher:b'btkitty.me'
        publisher-url:b'http://btkitty.me'

文件简报:
File:./5.torrent
announce:udp://tracker.openbittorrent.com:80
announce list:
        udp://open.demonii.com:1337
        udp://tracker.coppersurfer.tk:6969
        udp://tracker.leechers-paradise.org:6969
        udp://tracker.blackunicorn.xyz:6969
Create time:2017-06-08 04:22:47
Ecoding:UTF-8
Title:SIRO-1690 なな 桃谷绘里香 桃谷エリカ Erika Momotani20歳 豆腐屋.wmv
Comment:This torrent is downloaded from btkitty.me , storebt.com
Created by:
has multi files:False
[单文件结构]
文件名为:SIRO-1690 なな 桃谷绘里香 桃谷エリカ Erika Momotani20歳 豆腐屋.wmv
文件长度:1385343970byte
文件md5sum:
文件块长度:1048576byte

磁力链:
magnet:?xt=urn:btih:H5X5LYB3CPDSUJH4WKVKKTDI5QQGY4JH
magnet:?xt=urn:btih:H5X5LYB3CPDSUJH4WKVKKTDI5QQGY4JH 就是要获得的磁力链,拷贝到迅雷试试看吧。

2018年5月13日