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 __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)
