|
Astronomy
#!/usr/bin/env python
#
# cycle through download dir and show the torrents being grabbed
#
# additionally consult /share/incoming/.torrents.xml to print stats
#
from sys import *
from common import *
from xml.dom import minidom, Node
import string
import math
import pwd
import getopt
## stolen from btlaunchmanycurses.py
def human_readable(n):
n = long(n)
unit = [' B', 'K', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y']
i = 0
if (n > 999):
i = 1
while i + 1 < len(unit) and (n >> 10) >= 999:
i += 1
n >>= 10
n = float(n) / (1 << 10)
if i > 0:
size = '%.1f' % n + '%s' % unit[i]
else:
size = '%.0f' % n + '%s' % unit[i]
return size
def findNodeName(parentNode, name):
for childNode in parentNode.childNodes:
if name == childNode.nodeName:
content = []
for textNode in childNode.childNodes:
content.append( textNode.nodeValue )
return string.join( content )
return ''
verbose = False
tsize = 0
selectedHashes = []
onlyForThisUser = False
showTotals = True
# opts:
# --verbose/-v ==> verbose = True
# --hash=<hash> ==> selectedHash = hash
# --for-me ==> only show torrents owned by os.getuid()
# setup args
try:
opts, args = getopt.getopt(argv[1:], 'v', ['hash=','torrent=','for-me'])
except getopt.GetoptError:
print 'Usage: %s [file1.torrent ... fileN.torrent]' % argv[0]
print 'Optional arguments: [--verbose/-v]: show stats'
print '--hash=<hash>: only show this hash (implies verbose)'
print '--for-me: only show torrents you\'re downloading'
print 'If torrents are specified, only those stats will be shown.'
exit(2)
for o,a in opts:
if o in("--verbose", "-v"):
verbose = True
elif o == '--hash':
selectedHashes.append(a)
verbose = True
showTotals = False
elif o == '--for-me':
onlyForThisUser = True
for a in args:
if os.path.exists( a ) and a.endswith( '.torrent' ):
info = infoFromTorrent(a)
if info == '':
print 'Failed to find anything matching %s' % (a)
else:
selectedHashes.append( sha( bencode( info ) ).hexdigest() )
doc = minidom.parse( TORRENT_XML )
if len(selectedHashes) > 0:
verbose = True
showTotals = False
totalSpeedUp = 0
totalSpeedDn = 0
totalBytesUp = 0
totalBytesDn = 0
print ''
for torrent in doc.documentElement.childNodes:
if torrent.nodeName == 'torrent':
torrentPath = findNodeName( torrent, 'fullpath' )
hash = findNodeName( torrent, 'hash' )
if len(selectedHashes) > 0 and hash not in(selectedHashes):
continue
ownerUID = int(findNodeName(torrent,'owner'))
if onlyForThisUser and ownerUID != os.getuid():
continue
name = findNodeName( torrent, 'name' )
fsize = human_readable( findNodeName( torrent, 'filesize' ) )
tsize += int( findNodeName( torrent, 'filesize' ) )
bytesUp = int(findNodeName( torrent, 'totalUploadBytes' ))
bytesDn = int(findNodeName( torrent, 'totalDownloadBytes' ))
speedUp = float(findNodeName(torrent, 'uploadRate'))
speedDn = float(findNodeName(torrent, 'downloadRate'))
totalBytesUp += bytesUp
totalBytesDn += bytesDn
totalSpeedUp += speedUp
totalSpeedDn += speedDn
status = findNodeName( torrent, 'status' )
if not onlyForThisUser:
ownerName = pwd.getpwuid(ownerUID)[0]
print '%s: %s [%s]' % (ownerName, name, fsize)
else:
print '%s [%s]' % (name, fsize)
if verbose:
if status != "seeding":
print 'ETA: %s' % findNodeName( torrent, 'eta' )
if bytesDn > 0:
ratio = float(bytesUp) / float(bytesDn)
else:
ratio = -0.00
print 'Uploaded: %s, downloaded: %s [Ratio: %.2f]' % (human_readable(bytesUp), human_readable(bytesDn), ratio)
print ''
if verbose:
print '%s @ %s/s up, %s @ %s/s dn' % (human_readable(totalBytesUp), human_readable(totalSpeedUp), human_readable(totalBytesDn), human_readable(totalSpeedDn))
if showTotals:
print 'A total of %s is being downloaded' % human_readable(tsize)