【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+"	"
            retval=retval+str(key)+"
"

            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+"	"
            retval=retval+str(key)+"
"

            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+"	"
                
            showValue=str(value)[0:50] # 显示的值
            
            retval=retval+str(key)+":"+showValue+"
"

        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+"
"
        retval=retval+"announce:"+self.getAnnounce()+"
"

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

            for it in arr:
                retval=retval+"	"+it+"
"    
        
        retval=retval+"Create time:"+self.getCreateTime()+"
"
        retval=retval+"Ecoding:"+self.getEncoding()+"
"
        retval=retval+"Title:"+self.getTitle()+"
"
        retval=retval+"Comment:"+self.getComment()+"
"
        retval=retval+"Created by:"+self.getCreatedBy()+"
"

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

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

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

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

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

        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('文件结构:
'+tp.getStructure())
    print('文件简报:
'+str(tp.getBrief()))
    #print('文件内容:
'+str(tp.getAllContent()))
    print('磁力链:
'+str(tp.getMegnet()))

# Start
main()

输出:

C:Usershorn1Desktoppython46-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 xa4xcaxa4xca xccxd2xb9xc8xbb
                name.utf-8:b'SIRO-1690 xe3x81xaaxe3x81xaa xe6xa1x83
                piece length:1048576
                pieces:b'Sx06xa4x18xedx9dxa5xd9rxc4^Ox1bxefsxe
        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日
原文地址:https://www.cnblogs.com/heyang78/p/9032568.html