|
Astronomy
#!/usr/local/bin/python
########################################################################
# sort torrents.list based on what's been downloaded
########################################################################
# BUGS
# 1- doesnt take into account the current working dir. maybe os.path.normpath()
# or os.path.abspath() ?
#########################################################################
from sys import exit, argv
from pythonutils import LockFile
import os
import os.path
import getopt
from common import *
import shutil
########################################################################
# move a torrent to the EXPIRED directory
def removeDownloadedTorrent(dir):
if os.path.exists(dir):
# flat out delete USER_DL_DIR files
if dir.startswith( USER_DL_DIR):
if os.path.isfile( dir ):
os.remove(dir)
else:
shutil.rmtree(dir)
else:
newFile = os.path.join( EXPIRED_TORRENT_DIR, basename(dir) )
os.rename(dir,newFile)
########################################################################
# cache a file
def cacheFileContents(fileName):
lines = []
try:
fn = open( fileName, 'r' )
for line in fn.readlines():
line = line[:-1]
lines.append( line )
fn.close()
return lines
except IOError:
return []
########################################################################
# write an array to a file
def writeArrayToFile(array,fileName,newline=True):
f = open( fileName, 'w' )
for line in array:
f.write( line )
if newline:
f.write( '\n' )
f.close()
########################################################################
# see if a string exists in an array
def stringExistsInArray(str,array):
for line in array:
if line == str:
return True
return False
########################################################################
# print usage
def printUsageAndExit(appName):
print "%s will clear downloaded files from your ~/torrents.list" % appName
print "USAGE: %s [--fetched-files=FILENAME] [file1 ... fileN]" % appName
print "--fetched-files= Specify a path to filenames you've downloaded"
print "(defaults to ~/.fetched_files)"
exit(2)
########################################################################
# main
FETCHED_FILES = os.path.join( os.environ["HOME"], ".fetched_files" )
TORRENT_LIST = os.path.join( os.environ["HOME"], "torrents.list" )
# use getopt to see if we're processing a different FETCHED_FILES list
try:
opts, args = getopt.getopt(argv[1:], None, ['fetched-files=','torrent-list='])
except:
printUsageAndExit(argv[0])
for o, a in opts:
if o == "--fetched-files":
FETCHED_FILES=os.path.expandvars(os.path.expanduser(a))
if o == "--torrent-list":
TORRENT_LIST=os.path.expandvars(os.path.expanduser(a))
# dont make the user have a fetched_files list
if os.path.exists(FETCHED_FILES):
fetchedFilesLock = Lock( FETCHED_FILES, timeout=5, step=0.1 )
fetchedFilesLock.lock()
fetchedFiles = cacheFileContents( FETCHED_FILES )
else:
fetchedFiles = []
fetchedFilesLock = None
if not os.path.exists(TORRENT_LIST):
print 'Unable to find %s' % TORRENT_LIST
printUsageAndExit(argv[0])
# lock our files so 'stop' cant interfere until we're done
torrentListLock = Lock( TORRENT_LIST, timeout=5, step=0.1 )
torrentListLock.lock()
# cache our data
torrentList = cacheFileContents( TORRENT_LIST )
newTorrents = []
removeTorrents = []
# if args are specified, clear those torrents
if len(args) > 0:
for currentArg in args:
if not os.path.isabs( currentArg ):
currentArg = os.path.abspath( currentArg )
if os.path.exists(currentArg):
removeTorrents.append( currentArg )
# append to fetched files if not there already
if not stringExistsInArray( currentArg, fetchedFiles ):
fetchedFiles.append( currentArg )
# find out which files have already been fetched
for currentTorrent in torrentList:
if stringExistsInArray( currentTorrent, fetchedFiles ):
if not stringExistsInArray( currentTorrent, removeTorrents ):
removeTorrents.append( currentTorrent )
else: # hasnt been downloaded yet
if os.path.exists( currentTorrent ): # make sure file exists
newTorrents.append( currentTorrent )
else:
removeTorrents.append( currentTorrent )
# wipe all the torrents-to-be-removed
if len(removeTorrents) > 0:
print 'Retiring torrents:'
for currentRmTorrent in removeTorrents:
print "%s" % currentRmTorrent
removeDownloadedTorrent( currentRmTorrent )
# write out to our original files and unlock
writeArrayToFile( fetchedFiles, FETCHED_FILES )
writeArrayToFile( newTorrents, TORRENT_LIST )
if fetchedFilesLock:
fetchedFilesLock.unlock()
torrentListLock.unlock()