#------------------------------------------------------------------------------------ # torrentParser1.03，用于解析torrent文件 # 修改了函数getStructure,增加其节点值显示 # 2018年5月9日 #------------------------------------------------------------------------------------ from bencode import bdecode import time #------------------------------------- # 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 main(): tp=torrentParser(filePathname='./8.torrent') print('文件名='+tp.getFilepathname()) print('文件结构: '+tp.getStructure()) print('文件简报: '+str(tp.getBrief())) print('文件内容: '+str(tp.getAllContent())) # Start main()
2018年5月9日