|
Astronomy
#!/usr/local/bin/python
#
# find torrents >15 days
#
from sys import *
from os.path import *
from sha import *
import os
from shutil import *
import datetime
import time
from common import *
########################################################################
# delete a dir hierarchy
def deleteDownloadedTorrent(dir):
if os.path.isfile(dir):
os.unlink(dir)
else:
for root, dirs, files in os.walk(dir):
for currentDir in dirs:
deleteDownloadedTorrent(os.path.join( dir, currentDir ) )
for currentFile in files:
fullFilePath = os.path.join( dir, currentFile )
os.remove( fullFilePath )
os.rmdir( dir )
########################################################################
# find a string in a file
def fileNameExistsInFile(fname,readFile):
try:
fn = open( readFile, 'r' )
except:
#print 'Cant open %s' % readFile
# if we can't open torrents.list, dont bother
return False
for line in fn:
line = line[:-1]
if basename(line) == fname:
fn.close()
return True
fn.close()
return False
########################################################################
# stat()s a dir/file, checks to see if its older than n number of days
def isFileOld(fn, daysOld=15):
stats = os.stat(fn)
lastmod = datetime.date.fromtimestamp(stats[8])
today = datetime.date.today()
oldTime = today + datetime.timedelta(days=-1 * daysOld)
return oldTime > lastmod
########################################################################
# find old files
def findOldTorrents(rootdir,daysOld=15):
oldFiles=[]
for root, dirs, files in os.walk(rootdir):
for currentDir in dirs:
fn = os.path.join(root,currentDir)
if isFileOld( fn,daysOld ):
#print '%s' % fn
oldFiles.append( fn )
for currentFile in files:
fn = os.path.join(root, currentFile)
if isFileOld( fn, daysOld ):
#print '%s' % fn
oldFiles.append( fn )
return oldFiles
########################################################################
# main
####
if os.getuid() > 0:
print 'Only root can run this!'
exit()
if len(argv) > 1:
oldtime = int(argv[1])
else:
oldtime=15
oldies = findOldTorrents( EXPIRED_TORRENT_DIR, oldtime )
for old in oldies:
oldfn = basename(old)
canDelete = True
## TODO: consult a list to obtain torrent users
for user in ['howie', 'adam', 'nvilla']:
tfile = os.path.join( '/home', user, 'torrents.list')
if fileNameExistsInFile( oldfn, tfile):
print '%s needs to grab %s' % (user, old)
canDelete = False
if canDelete:
if os.path.exists(old):
deleteDownloadedTorrent(old)
print 'Removed: %s' % old