|
Astronomy
Download this file
#!/usr/local/bin/python
##############################################################################
# TODO
# - have a list of allowable trackers; if torrent's announce url isnt in
# one of those lists, dont add torrent to incoming dir
##############################################################################
from sys import *
from os.path import *
from sha import *
from BitTornado.bencode import *
import statvfs
import os
from shutil import *
import cookielib
import urllib2
from urllib import unquote_plus
import re
from common import *
import getopt
useSymlinks = False
#useSymlinks = True
## make a symlink from the torrent to the incoming dir?
## if false, copies torrent to incoming dir
cookieFile = os.path.join(os.environ["HOME"], ".btrss", "cookies.txt" )
forceDownload = False
# setup args
try:
opts, args = getopt.getopt(argv[1:], 'f', ['force', 'cookie='])
except getopt.GetoptError:
print '%s file1.torrent [--force/-f] [--cookie=path] fileN.torrent' % argv[0]
exit(2)
for opt,arg in opts:
if opt in ("-f", "--force"):
forceDownload = True
if opt == "--cookie":
cookieFile = os.path.expandvars(os.path.expanduser(arg))
if len(args) == 0:
print '%s file1.torrent file2.torrent file3.torrent ...' % argv[0]
exit(2)
## setup http stuffs
cj = cookielib.MozillaCookieJar()
if os.path.exists( cookieFile ):
cj.load( cookieFile )
opener = urllib2.build_opener( urllib2.HTTPCookieProcessor( cj ) )
urllib2.install_opener( opener )
headers = {'User-Agent' : 'Mozilla/4.0 (compatible; python-based client; talk to liekomglol@gmail.com if this is problematic)'}
## end http stuffs
## get the filesize here so each torrent can take off what it needs
## in terms of disk; will give a more accurate picture of the available space
st = os.statvfs(os.getcwd())
totalSpace = st[statvfs.F_BLOCKS] * st[statvfs.F_FRSIZE]
#freeSpace = st[statvfs.F_BAVAIL] * st[statvfs.F_FRSIZE]
freeSpace = st[statvfs.F_BFREE] * st[statvfs.F_FRSIZE]
## take off 12%; dont let disk get above 88% full
freeSpace -= (totalSpace * PERCENT_KEEP_FREE)
for metainfo_name in args:
if metainfo_name.startswith( 'http://' ) or metainfo_name.startswith( 'https://' ):
# make an http request
try:
req = urllib2.Request(metainfo_name, None, headers)
f = urllib2.urlopen(req)
except urllib2.HttpError, e:
print 'HTTP Error: %d' % e.code
continue # skip this torrent
except urllib2.URLError, e:
print 'Network Error: %s' % e.reason.args[1]
continue # skip this torrent
if f.info().has_key('content-disposition'):
filestr = escapeFilename(f.info()['content-disposition'].replace("inline; filename=\"",'').replace('"','')).replace('attachment__filename_','')
else:
filestr = basename(metainfo_name)
pos = filestr.find('&')
if pos > -1:
eqpos = filestr.find('=',pos)+1 # drop =
filestr = filestr[eqpos:]
filestr = escapeFilename(unquote_plus(filestr))
# write out to a local file
of = open( filestr, "wb" )
of.write( f.read() )
of.close()
metainfo_name = filestr
# owners have to match
if not isFileOwnerCurrentUser( metainfo_name ):
print 'You do not own torrent \'%s\'' % metainfo_name
continue
metainfo_file = open(metainfo_name, 'rb')
metainfo = bdecode(metainfo_file.read())
metainfo_file.close()
info = metainfo['info']
info_hash = sha( bencode( info ) ).hexdigest()
# allowed tracker?
if not isTrackerAllowed( metainfo['announce'] ):
print 'Tracker for \'%s\' is not allowed.' % info['name']
continue
## make sure we havent downloaded this already
if checkDownloadStatus( info ) and not forceDownload:
print "A torrent with the hash %s has already been downloaded." % info_hash
else:
if info.has_key('length'):
fileSize = info['length']
else:
fileSize = 0;
for file in info['files']:
fileSize += file['length']
# filesize > disk free-10%?
if fileSize > freeSpace:
print "Sorry, not enough space for torrent %s!" % metainfo_name
else:
freeSpace -= fileSize
# figure out what to name the torrent
torrentName = ( os.path.join( INCOMING_TORRENT_DIR, info_hash ) + '.torrent' )
if exists( torrentName ):
print 'A torrent the signature %s is already being downloaded' % info_hash
else:
# log the torrent info
if useSymlinks:
os.symlink( os.path.abspath(metainfo_name), torrentName )
os.chmod( metainfo_name, 0640 )
else:
copy( metainfo_name, torrentName )
os.chmod( torrentName, 0640 )
recordActiveTorrent( metainfo_name, info['name'], info_hash )
print 'Will begin downloading %s shortly.' % metainfo_name