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: