from os import sep
from robustclient import RobustClient
from episodes_pb2 import All
from datetime import datetime
from epgrabber import idnum, run
from re import compile
db = All()
db.ParseFromString(open("watch.list","rb").read())
series = [(x.name,x.search) for x in db.series]
trans = RobustClient("localhost",6886,"palfrey","epsilon")
torrents = trans.list()
ids = {}
for k in list(torrents.keys()):
details = trans.info(k)
for val in details.files():
f = details.files()[val]['name']
small = f.lower().encode("ascii", "replace")
for (name,search) in series:
if search == "":
search = name
bits = search.replace("eztv","").strip().lower().decode("utf-8","replace").split(" ")
for b in bits:
if b == "":
continue
if b[0] == "-": # an ignore
continue
if small.find(b)==-1:
#print "failed with",b,f
parser.add_option("-i","--ignore-torrent",help="Do action irregardless of torrent existing",action="store_false",default=True,dest="check_torrent")
parser.add_option("-c","--check-dir",default=".",type="string",dest="check_dir",help="Directory to check")
parser.add_option("-d","--dest-dir",default="output",type="string",dest="dest_dir",help="Directory to move files to")
(opts,args) = parser.parse_args()
if not exists(opts.check_dir):
parser.error("Folder '%s' doesn't exist!"%opts.check_dir)
if not exists(opts.dest_dir):
parser.error("Folder '%s' doesn't exist!"%opts.dest_dir)
db = All()
db.ParseFromString(open("watch.list","rb").read())
series = [(x.name,x.search) for x in db.series]
if opts.check_torrent:
trans = RobustClient("localhost",6886,user="******",password="******")
torrents = trans.list()
ids = {}
for k in list(torrents.keys()):
files = trans.info(k).files()
for id in files:
f = files[id]['name']
if f.find(sep)!=-1:
f = f[:f.find(sep)]
ids[f] = k
def remove_dir(top):
for root,dirs,files in walk(top, topdown=False):
for name in files:
remove(join(root,name))
for name in dirs:
