class populate( object ): __metaclass__ = Singleton def __init__(self, host=None): self.db = MythDB() self.db.searchRecorded.handler = Recorded self.be = MythBE(db=self.db) self.log = MythLog(db=self.db) self.set_host(host) self.load_backends() self.load_storagegroups() def set_host(self, host): self.host = host if host: # if the host was defined on the command line, check # to make sure such host is defined in the database with self.db as c: c.execute("""SELECT count(1) FROM settings WHERE hostname=? AND value=?""", (host, 'BackendServerIP')) if c.fetchone()[0] == 0: raise Exception('Invalid hostname specified for backend.') def load_backends(self): with self.db as c: c.execute("""SELECT hostname FROM settings WHERE value='BackendServerIP'""") hosts = [r[0] for r in c.fetchall()] self.hosts = [] for host in hosts: # try to access all defined hosts, and # store the ones currently accessible try: MythBE(backend=host) self.hosts.append(host) except: pass def load_storagegroups(self): self.storagegroups = \ [sg for sg in self.db.getStorageGroup() \ if sg.groupname not in ('Videos','Banners','Coverart',\ 'Fanart','Screenshots','Trailers')] def flush(self): self.misplaced = [] self.zerorecs = [] self.pendrecs = [] self.orphrecs = [] self.orphvids = [] self.orphimgs = [] self.dbbackup = [] self.unfiltered = [] def __call__(self): self.refresh_content() return self def refresh_content(self): # scan through all accessible backends to # generate a new listof orphaned content self.flush() unfiltered = {} for host in self.hosts: for sg in self.storagegroups: try: dirs,files,sizes = self.be.getSGList(host, sg.groupname, sg.dirname) for f,s in zip(files, sizes): newfile = File(host, sg.groupname, sg.dirname, f, s, self.db) # each filename should be unique among all storage directories # defined on all backends, but may exist in the same directory # on multiple backends if they are shared if newfile not in unfiltered: # add a new file to the list unfiltered[str(newfile)] = newfile else: # add a reference to the host on which it was found unfiltered[str(newfile)].add_host(host) except: self.log(MythLog.GENERAL, MythLog.INFO, 'Could not access {0.groupname}@{1}{0.dirname}'.format(sg, host)) for rec in self.db.searchRecorded(livetv=True): if rec.hostname not in self.hosts: # recording is on an offline backend, ignore it name = rec.basename.rsplit('.',1)[0] for n in unfiltered.keys(): if name in n: # and anything related to it del unfiltered[n] elif rec.basename in unfiltered: # run through list of recordings, matching basenames # with found files, and removing file from list f = unfiltered[rec.basename] del unfiltered[rec.basename] if f.size < 1024: # file is too small to be of any worth self.zerorecs.append(rec) elif rec.doubleorphan: # file is marked for deletion, but has been forgotten by the backend self.pendrecs.append(rec) elif rec.hostname not in f.hosts: # recording is in the database, but not where it should be self.misplaced.append(rec) name = rec.basename.rsplit('.',1)[0] for f in unfiltered.keys(): if name in f: # file is related to a valid recording, ignore it del unfiltered[f] else: # recording has been orphaned self.orphrecs.append(rec) for n,f in unfiltered.iteritems(): if n.endswith('.mpg') or n.endswith('.nuv'): # filter files with recording extensions self.orphvids.append(f) elif n.endswith('.png'): # filter files with image extensions self.orphimgs.append(f) elif 'sql' in n: # filter for database backups self.dbbackup.append(f) else: self.unfiltered.append(f) def print_results(self): printrecs("Recordings found on the wrong host", self.misplaced) printrecs("Recordings with missing files", self.orphrecs) printrecs("Zero byte recordings", self.zerorecs) printrecs("Forgotten pending deletions", self.pendrecs) printfiles("Orphaned video files", self.orphvids) printfiles("Orphaned snapshots", self.orphimgs) printfiles("Database backups", self.dbbackup) printfiles("Other files", self.unfiltered)
class populate(object): __metaclass__ = Singleton def __init__(self, host=None): self.db = MythDB() self.db.searchRecorded.handler = Recorded self.be = MythBE(db=self.db) self.log = MythLog(db=self.db) self.set_host(host) self.load_backends() self.load_storagegroups() def set_host(self, host): self.host = host if host: # if the host was defined on the command line, check # to make sure such host is defined in the database with self.db as c: c.execute( """SELECT count(1) FROM settings WHERE hostname=? AND value=?""", (host, 'BackendServerIP')) if c.fetchone()[0] == 0: raise Exception('Invalid hostname specified for backend.') def load_backends(self): with self.db as c: c.execute("""SELECT hostname FROM settings WHERE value='BackendServerIP'""") hosts = [r[0] for r in c.fetchall()] self.hosts = [] for host in hosts: # try to access all defined hosts, and # store the ones currently accessible try: MythBE(backend=host) self.hosts.append(host) except: pass def load_storagegroups(self): self.storagegroups = \ [sg for sg in self.db.getStorageGroup() \ if sg.groupname not in ('Videos','Banners','Coverart',\ 'Fanart','Screenshots','Trailers')] def flush(self): self.misplaced = [] self.zerorecs = [] self.pendrecs = [] self.orphrecs = [] self.orphvids = [] self.orphimgs = [] self.dbbackup = [] self.unfiltered = [] def __call__(self): self.refresh_content() return self def refresh_content(self): # scan through all accessible backends to # generate a new listof orphaned content self.flush() unfiltered = {} for host in self.hosts: for sg in self.storagegroups: try: dirs, files, sizes = self.be.getSGList( host, sg.groupname, sg.dirname) for f, s in zip(files, sizes): newfile = File(host, sg.groupname, sg.dirname, f, s, self.db) # each filename should be unique among all storage directories # defined on all backends, but may exist in the same directory # on multiple backends if they are shared if newfile not in unfiltered: # add a new file to the list unfiltered[str(newfile)] = newfile else: # add a reference to the host on which it was found unfiltered[str(newfile)].add_host(host) except: self.log( MythLog.GENERAL, MythLog.INFO, 'Could not access {0.groupname}@{1}{0.dirname}'.format( sg, host)) for rec in self.db.searchRecorded(livetv=True): if rec.hostname not in self.hosts: # recording is on an offline backend, ignore it name = rec.basename.rsplit('.', 1)[0] for n in unfiltered.keys(): if name in n: # and anything related to it del unfiltered[n] elif rec.basename in unfiltered: # run through list of recordings, matching basenames # with found files, and removing file from list f = unfiltered[rec.basename] del unfiltered[rec.basename] if f.size < 1024: # file is too small to be of any worth self.zerorecs.append(rec) elif rec.doubleorphan: # file is marked for deletion, but has been forgotten by the backend self.pendrecs.append(rec) elif rec.hostname not in f.hosts: # recording is in the database, but not where it should be self.misplaced.append(rec) name = rec.basename.rsplit('.', 1)[0] for f in unfiltered.keys(): if name in f: # file is related to a valid recording, ignore it del unfiltered[f] else: # recording has been orphaned self.orphrecs.append(rec) for n, f in unfiltered.iteritems(): if n.endswith('.mpg') or n.endswith('.nuv'): # filter files with recording extensions self.orphvids.append(f) elif n.endswith('.png'): # filter files with image extensions self.orphimgs.append(f) elif 'sql' in n: # filter for database backups self.dbbackup.append(f) else: self.unfiltered.append(f) def print_results(self): printrecs("Recordings found on the wrong host", self.misplaced) printrecs("Recordings with missing files", self.orphrecs) printrecs("Zero byte recordings", self.zerorecs) printrecs("Forgotten pending deletions", self.pendrecs) printfiles("Orphaned video files", self.orphvids) printfiles("Orphaned snapshots", self.orphimgs) printfiles("Database backups", self.dbbackup) printfiles("Other files", self.unfiltered)