|
Astronomy
#!/usr/local/bin/python
###########################################################################
# TODO
# - consult .stats.db to find out if a 1:1 ratio is achieved. if not, do not
# let torrent stop without -f flag (force)
#
# BUGS
# X a paused torrent cannot be stopped, it has to be started again and then
# stopped. should probably check to see if INCOMING/<mdsum> exists, then
# proceed with writing the output
#
###########################################################################
import os
import time
import sys
import os.path
from sha import *
from BitTornado.bencode import *
from shutil import *
import getopt
from common import *
escapeFilenames = True
#==============================================================================
# handy list of torrents one has downloaded so they can potentially script
# the scp'ing of them to their local machines
def recordLocalTorrent(path):
filename = os.path.join( os.getenv( "HOME" ), 'torrents.list' )
fn = LockFile( filename, mode='a', timeout=5, step=0.1 )
fn.write( path )
fn.write( '\n' )
fn.close()
#==============================================================================
def printUsageAndExit(appName):
print 'USAGE: %s [--delete/-d] [--no-escape] [--pause/-p] file1.torrent ... fileN.torrent' % appName
print 'Option \'no-escape\' will prevent filenames from being escaped.'
print 'Option \'delete\' will simply remove the torrent without writing any output.'
print 'Option \'pause\' will pause a torrent.'
sys.exit(2)
#==============================================================================
deleteOnly = False
pauseOnly = False
forceStop = False
try:
opts, args = getopt.getopt(sys.argv[1:], 'dpf', ['pause', 'no-escape', 'escape=','delete','force'])
except getopt.GetoptError:
printUsageAndExit(sys.argv[0])
for opt,arg in opts:
if opt == "--no-escape":
escapeFilename = False
if opt == "--escape":
print 'escape=REGEX not yet implemented'
if opt in ("--delete", "-d"):
deleteOnly = True
if opt in("--pause", "-p"):
pauseOnly = True
if opt in("-f","--force"):
forceStop = True
if len(args) == 0:
printUsageAndExit(sys.argv[0])
for torrent_name in args:
metainfo_name = findTorrent(torrent_name)
if metainfo_name == '':
print "Could not locate anything matching '%s', sorry!" % torrent_name
continue
if not isFileOwnerCurrentUser(metainfo_name):
print 'You do not own torrent \'%s\'' % metainfo_name
continue
info = infoFromTorrent(metainfo_name)
info_hash = sha( bencode( info ) ).hexdigest()
# figure out what the name of the torrent is
torrentPath = os.path.join( INCOMING_TORRENT_DIR, info_hash )
torrentName = torrentPath + '.torrent'
# torrent may be paused so we'll check for the path first
if exists( torrentPath ):
if not pauseOnly:
## log that this torrent was downloaded
recordDownloadedTorrent(info)
if exists( torrentName ):
os.remove( torrentName )
if pauseOnly:
print 'Paused torrent %s' % basename( metainfo_name )
continue
print 'Shutting down torrent %s' % basename( metainfo_name )
# - wait 20 seconds for btlaunchmanycurses to notice the rm'ed
# torrent file.
print 'Sleeping for %d seconds...' % MAX_SLEEP_TIME
printSleepingStatus( MAX_SLEEP_TIME )
if deleteOnly:
try:
sys.stdout.write( 'Removing contents...' )
sys.stdout.flush()
rmtree( torrentPath )
sys.stdout.write( 'Done!\n' )
sys.stdout.flush()
except:
sys.stdout.flush()
print 'Failed to remove torrent contents (%s)' % sys.exc_info()
continue
else:
# - move INCOMING_TORRENT_DIR/mdsum/ to /share/torrents/%torrentFileNameMinus-dot_torrent%
# UNLESS the torrent has a directory specified in it, in which
# case move the mdsum dir to that directory.
cwd = os.path.realpath( os.getcwd() )
if escapeFilenames:
output = os.path.join( cwd, escapeFilename( info['name'] ) )
else:
output = os.path.join( cwd, info['name'] )
if info.has_key('length'): ## assume a single file
torrentedFile = os.path.join( torrentPath, info['name'] )
os.rename(torrentedFile, output)
## now wipe the directory
os.rmdir( torrentPath )
print 'Wrote output to %s' % output
else:
os.rename( torrentPath, output )
print 'Wrote contents to dir %s' % output
## log to our list of torrents
recordLocalTorrent(output)
## now remove the local torrent file
try:
os.remove( metainfo_name )
except:
print "Could not remove local torrent."
else:
print 'No torrent found with a signature of %s' % info_hash