def __call__(self, environ, start_response): """Invoke the Controller""" def check_url(): "check if user should have totals calculated" if ('format' in environ['pylons.routes_dict'] and environ['pylons.routes_dict']['format'] in ['csv', 'pdf']): return False if environ['PATH_INFO'] == '/jsi18n.js': return False return True self.identity = environ.get('repoze.who.identity') if (self.identity is not None and 'user' in self.identity and environ['pylons.routes_dict']['controller'] != 'error' and check_url()): if self.identity['user']: totals = DailyTotals(Session, self.identity['user']) mailq = MailQueue(Session, self.identity['user']) c.baruwa_totals = totals.get() c.baruwa_inbound = mailq.get(1)[0] c.baruwa_outbound = mailq.get(2)[0] if self.identity['user'].is_admin: c.baruwa_status = cluster_status() tzinfo = self.identity['user'].timezone or UTC c.tzinfo = make_tz(tzinfo) try: return WSGIController.__call__(self, environ, start_response) finally: Session.remove()
def command(self): "command" self.init() if asbool(self.conf.get('ms.quarantine.shared')): lock_name = 'cleanquarantine' else: lock_name = 'cleanquarantine-%s' % system_hostname() if acquire_lock(lock_name, self.conf): try: days_to_retain = int( self.conf.get('ms.quarantine.days_to_keep', 0)) quarantine_dir = get_config_option('QuarantineDir') if (quarantine_dir.startswith(('/etc', '/lib', '/home', '/bin', '/sbin', '..'))): return False if ((not os.path.exists(quarantine_dir)) or (days_to_retain == 0)): return False ignore_dirs = ['spam', 'mcp', 'nonspam'] def process_dir(dirs, process_path, direc): "process dirs" if os.path.exists(os.path.join(process_path, direc)): dirs.extend([f for f in os.listdir( os.path.join(process_path, direc))]) dirs = [f for f in os.listdir(quarantine_dir) if os.path.isdir(os.path.join(quarantine_dir, f)) and QDIR.match(f) and should_be_pruned(f, days_to_retain)] dirs.sort() for direc in dirs: process_path = os.path.join(quarantine_dir, direc) ids = [f for f in os.listdir(process_path) if f not in ignore_dirs] for ignore_dir in ignore_dirs: process_dir(ids, process_path, ignore_dir) year, month, day = (int(direc[:4]), int(direc[4:-2]), int(direc[6:])) startdate = datetime.datetime(year, month, day, 00, 00, 00) enddate = datetime.datetime(year, month, day, 23, 59, 59) localzone = make_tz(self.conf['baruwa.timezone']) startdate = localzone.localize(startdate) enddate = localzone.localize(enddate) startdate = pytz.utc.normalize( startdate.astimezone(pytz.utc)) enddate = pytz.utc.normalize(enddate.astimezone(pytz.utc)) sql = Message.__table__.update().where(and_( Message.messageid.in_(ids), Message.timestamp.between(startdate, enddate) )).values(isquarantined=0) Session.bind.execute(sql) if (os.path.isabs(process_path) and (not os.path.islink(process_path))): try: shutil.rmtree(process_path) except shutil.Error: print >> sys.stderr, ("Failed to remove %(path)s" % dict(path=process_path)) else: print >> sys.stderr, ("%(path)s is a symlink skipping" % dict(path=process_path)) finally: Session.close() release_lock(lock_name, self.conf)