0x2620/python-ox
2
1
Fork 0
Code Issues Pull requests Releases Wiki Activity

use under_score api in ox.torrent too

Browse source
This commit is contained in:
j 2012-08-21 10:49:36 +02:00
parent d1299f940f
commit a22f07ee26
1 changed files with 14 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
ox/torrent/__init__.py
View file

@ -1,6 +1,6 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4 # vi:si:et:sw=4:sts=4:ts=4
# GPL 2007-2009 # GPL 2007-2012
from threading import Event from threading import Event
from hashlib import sha1 from hashlib import sha1
@ -8,31 +8,26 @@ import os
from bencode import bencode, bdecode from bencode import bencode, bdecode
__all__ = ['createTorrent', 'getInfoHash', 'getTorrentInfoFromFile', 'getTorrentInfo', 'getFiles', 'getTorrentSize'] __all__ = ['create_torrent', 'get_info_hash', 'get_torrent_info', 'get_files', 'get_torrent_size']
def createTorrent(file, url, params = {}, flag = Event(), def create_torrent(file, url, params = {}, flag = Event(),
progress = lambda x: None, progress_percent = 1): progress = lambda x: None, progress_percent = 1):
"Creates a torrent for a given file, using url as tracker url" "Creates a torrent for a given file, using url as tracker url"
from makemetafile import make_meta_file from makemetafile import make_meta_file
return make_meta_file(file, url, params, flag, progress, progress_percent) return make_meta_file(file, url, params, flag, progress, progress_percent)
def getInfoHash(torrentFile): def get_info_hash(torrentFile):
"Returns Torrent Info Hash from torrent file" "Returns Torrent Info Hash from torrent file"
metainfo_file = open(torrentFile, 'rb') metainfo_file = open(torrentFile, 'rb')
metainfo = bdecode(metainfo_file.read()) metainfo = bdecode(metainfo_file.read())
info = metainfo['info'] info = metainfo['info']
return sha1(bencode(info)).hexdigest() return sha1(bencode(info)).hexdigest()
def getTorrentInfoFromFile(torrentFile): def get_torrent_info(data=None, file=None):
f = open(torrentFile, 'rb')
data = f.read()
f.close()
tinfo = getTorrentInfo(data)
tinfo['timestamp'] = os.stat(torrentFile).st_ctime
return tinfo
def getTorrentInfo(data):
from bencode import bencode from bencode import bencode
if file:
with open(file, 'rb') as f:
data = f.read()
"Returns Torrent Info from torrent file" "Returns Torrent Info from torrent file"
tinfo = {} tinfo = {}
@ -56,11 +51,13 @@ def getTorrentInfo(data):
tinfo['size'] = file_length tinfo['size'] = file_length
tinfo['hash'] = sha1(bencode(info)).hexdigest() tinfo['hash'] = sha1(bencode(info)).hexdigest()
tinfo['announce'] = metainfo['announce'] tinfo['announce'] = metainfo['announce']
if file:
tinfo['timestamp'] = os.stat(file).st_ctime
return tinfo return tinfo
def getFiles(data): def get_files(data):
files = [] files = []
info = getTorrentInfo(data) info = get_torrent_info(data=data)
if 'files' in info: if 'files' in info:
for f in info['files']: for f in info['files']:
path = [info['name'], ] path = [info['name'], ]
@ -70,7 +67,7 @@ def getFiles(data):
files.append(info['name']) files.append(info['name'])
return files return files
def getTorrentSize(torrentFile): def get_torrent_size(file):
"Returns Size of files in torrent file in bytes" "Returns Size of files in torrent file in bytes"
return getTorrentInfoFromFile(torrentFile)['size'] return get_torrent_info(file=file)['size']