|
Astronomy
Download this file
#!/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)
# touch all files
for root, dirs, files in os.walk( newFile ):
for cf in files:
currentFile = os.path.join( newFile, cf )
if os.path.exists( currentFile ):
os.utime( currentFile, None )
for cd in dirs:
currentDir = os.path.join( newFile, cd )
if os.path.exists(currentDir):
os.utime( currentDir, None )
os.utime( newFile, None )
########################################################################
# 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 currentArg in fetchedFiles:
fetchedFiles.append( currentArg )
# find out which files have already been fetched
for currentTorrent in torrentList:
if currentTorrent in fetchedFiles:
if not currentTorrent in 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( newTorrents, TORRENT_LIST )
if fetchedFilesLock:
writeArrayToFile( fetchedFiles, FETCHED_FILES )
fetchedFilesLock.unlock()
torrentListLock.unlock()