django-orchestra-test/orchestra/contrib/mailboxes/backends.py

576 lines
23 KiB
Python
Raw Normal View History

2014-10-09 17:04:12 +00:00
import logging
2015-04-29 10:51:30 +00:00
import os
2015-04-28 16:07:16 +00:00
import re
import textwrap
2014-08-22 15:31:44 +00:00
2014-10-06 14:57:02 +00:00
from django.core.exceptions import ObjectDoesNotExist
2014-08-22 15:31:44 +00:00
from django.utils.translation import ugettext_lazy as _
from orchestra.contrib.orchestration import ServiceController
2015-04-05 10:46:24 +00:00
from orchestra.contrib.resources import ServiceMonitor
2014-08-22 15:31:44 +00:00
from . import settings
2014-09-26 15:05:20 +00:00
from .models import Address
2014-08-22 15:31:44 +00:00
2014-10-06 14:57:02 +00:00
2014-10-09 17:04:12 +00:00
logger = logging.getLogger(__name__)
2015-04-28 16:07:16 +00:00
class SieveFilteringMixin(object):
def generate_filter(self, mailbox, context):
name, content = mailbox.get_filtering()
2015-04-28 16:07:16 +00:00
for box in re.findall(r'fileinto\s+"([^"]+)"', content):
2015-04-29 10:51:30 +00:00
# create mailboxes if fileinfo is provided witout ':create' option
2015-04-28 16:07:16 +00:00
context['box'] = box
self.append(textwrap.dedent("""
# Create %(box)s mailbox
2015-08-31 11:58:59 +00:00
su %(user)s --shell /bin/bash << 'EOF'
mkdir -p "%(maildir)s/.%(box)s"
EOF
2015-04-29 10:51:30 +00:00
if [[ ! $(grep '%(box)s' %(maildir)s/subscriptions) ]]; then
echo '%(box)s' >> %(maildir)s/subscriptions
2015-08-31 11:58:59 +00:00
chown %(user)s:%(user)s %(maildir)s/subscriptions
2015-04-29 10:51:30 +00:00
fi
""") % context
)
context['filtering_path'] = settings.MAILBOXES_SIEVE_PATH % context
context['filtering_cpath'] = re.sub(r'\.sieve$', '.svbin', context['filtering_path'])
if content:
2015-04-29 10:51:30 +00:00
context['filtering'] = ('# %(banner)s\n' + content) % context
self.append(textwrap.dedent("""\
# Create and compile orchestra sieve filtering
2015-08-31 11:58:59 +00:00
su %(user)s --shell /bin/bash << 'EOF'
mkdir -p $(dirname "%(filtering_path)s")
cat << ' EOF' > %(filtering_path)s
%(filtering)s
EOF
sievec %(filtering_path)s
EOF
2015-04-29 10:51:30 +00:00
""") % context
)
else:
self.append("echo '' > %(filtering_path)s" % context)
2015-08-31 11:58:59 +00:00
self.append('chown %(user)s:%(group)s %(filtering_path)s' % context)
2015-04-28 16:07:16 +00:00
class UNIXUserMaildirBackend(SieveFilteringMixin, ServiceController):
2015-04-23 19:46:23 +00:00
"""
Assumes that all system users on this servers all mail accounts.
2015-05-12 12:38:40 +00:00
If you want to have system users AND mailboxes on the same server you should consider using virtual mailboxes.
Supports quota allocation via <tt>resources.disk.allocated</tt>.
2015-04-23 19:46:23 +00:00
"""
2015-04-29 10:51:30 +00:00
SHELL = '/dev/null'
2015-04-05 18:02:36 +00:00
verbose_name = _("UNIX maildir user")
2015-03-10 21:51:10 +00:00
model = 'mailboxes.Mailbox'
def save(self, mailbox):
context = self.get_context(mailbox)
self.append(textwrap.dedent("""
# Update/create %(user)s user state
2015-03-10 21:51:10 +00:00
if [[ $( id %(user)s ) ]]; then
old_password=$(getent shadow %(user)s | cut -d':' -f2)
usermod %(user)s \\
--shell %(initial_shell)s \\
--password '%(password)s'
if [[ "$old_password" != '%(password)s' ]]; then
# Postfix SASL caches passwords
RESTART_POSTFIX=1
fi
2015-03-10 21:51:10 +00:00
else
useradd %(user)s \\
--home %(home)s \\
--password '%(password)s'
2015-03-10 21:51:10 +00:00
fi
mkdir -p %(home)s
chmod 751 %(home)s
chown %(user)s:%(group)s %(home)s""") % context
)
if hasattr(mailbox, 'resources') and hasattr(mailbox.resources, 'disk'):
self.set_quota(mailbox, context)
self.generate_filter(mailbox, context)
2015-03-10 21:51:10 +00:00
def set_quota(self, mailbox, context):
2015-05-22 13:15:06 +00:00
allocated = mailbox.resources.disk.allocated
scale = mailbox.resources.disk.resource.get_scale()
context['quota'] = allocated * scale
#unit_to_bytes(mailbox.resources.disk.unit)
2015-03-10 21:51:10 +00:00
self.append(textwrap.dedent("""
# Set Maildir quota for %(user)s
2015-08-31 11:58:59 +00:00
su %(user)s --shell /bin/bash << 'EOF'
mkdir -p %(maildir)s
EOF
2015-04-29 10:51:30 +00:00
if [[ ! -f %(maildir)s/maildirsize ]]; then
echo "%(quota)iS" > %(maildir)s/maildirsize
chown %(user)s:%(group)s %(maildir)s/maildirsize
2015-03-10 21:51:10 +00:00
else
2015-04-29 10:51:30 +00:00
sed -i '1s/.*/%(quota)iS/' %(maildir)s/maildirsize
2015-03-10 21:51:10 +00:00
fi""") % context
)
def delete(self, mailbox):
context = self.get_context(mailbox)
2015-04-26 13:53:00 +00:00
self.append('mv %(home)s %(home)s.deleted || exit_code=$?' % context)
2015-03-10 21:51:10 +00:00
self.append(textwrap.dedent("""
2015-05-12 12:38:40 +00:00
nohup bash -c '{ sleep 2 && killall -u %(user)s -s KILL; }' &> /dev/null &
2015-03-10 21:51:10 +00:00
killall -u %(user)s || true
# Restart because of Postfix SASL caches credentials
2015-05-19 13:27:04 +00:00
userdel %(user)s || true && RESTART_POSTFIX=1
2015-03-10 21:51:10 +00:00
groupdel %(user)s || true""") % context
)
2015-05-19 13:27:04 +00:00
def commit(self):
self.append('[[ $RESTART_POSTFIX -eq 1 ]] && service postfix restart')
super(UNIXUserMaildirBackend, self).commit()
2015-03-10 21:51:10 +00:00
def get_context(self, mailbox):
context = {
'user': mailbox.name,
'group': mailbox.name,
'name': mailbox.name,
2015-03-10 21:51:10 +00:00
'password': mailbox.password if mailbox.active else '*%s' % mailbox.password,
'home': mailbox.get_home(),
2015-04-29 10:51:30 +00:00
'maildir': os.path.join(mailbox.get_home(), 'Maildir'),
'initial_shell': self.SHELL,
'banner': self.get_banner(),
2015-03-10 21:51:10 +00:00
}
return context
2015-03-10 21:51:10 +00:00
2015-04-28 16:07:16 +00:00
class DovecotPostfixPasswdVirtualUserBackend(SieveFilteringMixin, ServiceController):
2015-04-23 19:46:23 +00:00
"""
WARNING: This backends is not fully implemented
"""
DEFAULT_GROUP = 'postfix'
2015-04-05 18:02:36 +00:00
verbose_name = _("Dovecot-Postfix virtualuser")
2014-10-17 10:04:47 +00:00
model = 'mailboxes.Mailbox'
2014-08-22 15:31:44 +00:00
2014-10-06 14:57:02 +00:00
def set_user(self, context):
self.append(textwrap.dedent("""
if [[ $( grep '^%(user)s:' %(passwd_path)s ) ]]; then
2015-03-10 21:51:10 +00:00
sed -i 's#^%(user)s:.*#%(passwd)s#' %(passwd_path)s
else
2014-10-06 14:57:02 +00:00
echo '%(passwd)s' >> %(passwd_path)s
fi""") % context
)
2014-08-22 15:31:44 +00:00
self.append("mkdir -p %(home)s" % context)
2015-02-24 09:34:26 +00:00
self.append("chown %(uid)s:%(gid)s %(home)s" % context)
2014-08-22 15:31:44 +00:00
2014-10-09 17:04:12 +00:00
def set_mailbox(self, context):
self.append(textwrap.dedent("""
if [[ ! $(grep '^%(user)s@%(mailbox_domain)s\s' %(virtual_mailbox_maps)s) ]]; then
2015-03-10 21:51:10 +00:00
echo "%(user)s@%(mailbox_domain)s\tOK" >> %(virtual_mailbox_maps)s
2014-10-09 17:04:12 +00:00
UPDATED_VIRTUAL_MAILBOX_MAPS=1
fi""") % context
)
2014-10-09 17:04:12 +00:00
2014-08-22 15:31:44 +00:00
def save(self, mailbox):
context = self.get_context(mailbox)
2014-10-06 14:57:02 +00:00
self.set_user(context)
2014-10-09 17:04:12 +00:00
self.set_mailbox(context)
2014-08-22 15:31:44 +00:00
self.generate_filter(mailbox, context)
def delete(self, mailbox):
context = self.get_context(mailbox)
self.append(textwrap.dedent("""
2015-05-12 12:38:40 +00:00
nohup bash -c 'sleep 2 && killall -u %(uid)s -s KILL' &> /dev/null &
2015-04-09 14:32:10 +00:00
killall -u %(uid)s || true
sed -i '/^%(user)s:.*/d' %(passwd_path)s
sed -i '/^%(user)s@%(mailbox_domain)s\s.*/d' %(virtual_mailbox_maps)s
UPDATED_VIRTUAL_MAILBOX_MAPS=1""") % context
)
if context['deleted_home']:
2015-04-26 13:53:00 +00:00
self.append("mv %(home)s %(deleted_home)s || exit_code=$?" % context)
2015-04-09 14:32:10 +00:00
else:
self.append("rm -fr %(home)s" % context)
2014-08-22 15:31:44 +00:00
2014-10-06 14:57:02 +00:00
def get_extra_fields(self, mailbox, context):
context['quota'] = self.get_quota(mailbox)
return 'userdb_mail=maildir:~/Maildir {quota}'.format(**context)
def get_quota(self, mailbox):
try:
quota = mailbox.resources.disk.allocated
except (AttributeError, ObjectDoesNotExist):
return ''
unit = mailbox.resources.disk.unit[0].upper()
return 'userdb_quota_rule=*:bytes=%i%s' % (quota, unit)
2014-10-09 17:04:12 +00:00
def commit(self):
context = {
2014-10-17 10:04:47 +00:00
'virtual_mailbox_maps': settings.MAILBOXES_VIRTUAL_MAILBOX_MAPS_PATH
2014-10-09 17:04:12 +00:00
}
self.append(textwrap.dedent("""
2014-11-27 19:17:26 +00:00
[[ $UPDATED_VIRTUAL_MAILBOX_MAPS == 1 ]] && {
postmap %(virtual_mailbox_maps)s
}""") % context
)
2014-10-09 17:04:12 +00:00
2014-08-22 15:31:44 +00:00
def get_context(self, mailbox):
context = {
2014-09-30 14:46:29 +00:00
'name': mailbox.name,
2015-03-10 21:51:10 +00:00
'user': mailbox.name,
2014-10-06 14:57:02 +00:00
'password': mailbox.password if mailbox.active else '*%s' % mailbox.password,
'uid': 10000 + mailbox.pk,
'gid': 10000 + mailbox.pk,
'group': self.DEFAULT_GROUP,
'quota': self.get_quota(mailbox),
2014-10-17 10:04:47 +00:00
'passwd_path': settings.MAILBOXES_PASSWD_PATH,
2015-04-09 14:32:10 +00:00
'home': mailbox.get_home(),
2014-10-09 17:04:12 +00:00
'banner': self.get_banner(),
2014-10-17 10:04:47 +00:00
'virtual_mailbox_maps': settings.MAILBOXES_VIRTUAL_MAILBOX_MAPS_PATH,
'mailbox_domain': settings.MAILBOXES_VIRTUAL_MAILBOX_DEFAULT_DOMAIN,
2014-08-22 15:31:44 +00:00
}
2014-10-06 14:57:02 +00:00
context['extra_fields'] = self.get_extra_fields(mailbox, context)
2015-04-09 14:32:10 +00:00
context.update({
'passwd': '{user}:{password}:{uid}:{gid}::{home}::{extra_fields}'.format(**context),
'deleted_home': settings.MAILBOXES_MOVE_ON_DELETE_PATH % context,
})
return context
2014-08-22 15:31:44 +00:00
2015-05-05 19:42:55 +00:00
class PostfixAddressVirtualDomainBackend(ServiceController):
2015-04-23 19:46:23 +00:00
"""
2015-05-05 19:42:55 +00:00
Secondary SMTP server without mailboxes in it, only syncs virtual domains.
2015-04-23 19:46:23 +00:00
"""
2015-05-05 19:42:55 +00:00
verbose_name = _("Postfix address virtdomain-only")
2014-10-17 10:04:47 +00:00
model = 'mailboxes.Address'
related_models = (
('mailboxes.Mailbox', 'addresses'),
)
2015-04-24 11:39:20 +00:00
doc_settings = (settings,
2015-05-05 19:42:55 +00:00
('MAILBOXES_LOCAL_DOMAIN', 'MAILBOXES_VIRTUAL_ALIAS_DOMAINS_PATH')
2015-04-24 11:39:20 +00:00
)
2014-08-22 15:31:44 +00:00
2015-06-22 14:14:16 +00:00
def is_hosted_domain(self, domain):
2015-05-05 19:42:55 +00:00
""" whether or not domain MX points to this server """
return domain.has_default_mx()
2014-10-09 17:04:12 +00:00
2015-05-05 19:42:55 +00:00
def include_virtual_alias_domain(self, context):
domain = context['domain']
2015-06-22 14:14:16 +00:00
if domain.name != context['local_domain'] and self.is_hosted_domain(domain):
2014-10-09 17:04:12 +00:00
self.append(textwrap.dedent("""
# %(domain)s is a virtual domain belonging to this server
2015-05-12 12:38:40 +00:00
if [[ ! $(grep '^\s*%(domain)s\s*$' %(virtual_alias_domains)s) ]]; then
2015-05-05 19:42:55 +00:00
echo '%(domain)s' >> %(virtual_alias_domains)s
UPDATED_VIRTUAL_ALIAS_DOMAINS=1
2015-05-12 12:38:40 +00:00
fi""") % context
2015-05-05 19:42:55 +00:00
)
2014-08-22 15:31:44 +00:00
2015-05-05 19:42:55 +00:00
def is_last_domain(self, domain):
return not Address.objects.filter(domain=domain).exists()
def exclude_virtual_alias_domain(self, context):
domain = context['domain']
if self.is_last_domain(domain):
self.append(textwrap.dedent("""
# Delete %(domain)s virtual domain
2015-05-12 12:38:40 +00:00
if [[ $(grep '^%(domain)s\s*$' %(virtual_alias_domains)s) ]]; then
sed -i '/^%(domain)s\s*/d' %(virtual_alias_domains)s
2015-05-05 19:42:55 +00:00
UPDATED_VIRTUAL_ALIAS_DOMAINS=1
2015-05-12 12:38:40 +00:00
fi""") % context
2015-05-05 19:42:55 +00:00
)
2014-08-22 15:31:44 +00:00
def save(self, address):
context = self.get_context(address)
2014-10-09 17:04:12 +00:00
self.include_virtual_alias_domain(context)
2015-05-05 19:42:55 +00:00
return context
2014-08-22 15:31:44 +00:00
def delete(self, address):
context = self.get_context(address)
2014-10-09 17:04:12 +00:00
self.exclude_virtual_alias_domain(context)
2015-05-05 19:42:55 +00:00
return context
2014-08-22 15:31:44 +00:00
def commit(self):
context = self.get_context_files()
self.append(textwrap.dedent("""
2015-05-05 19:42:55 +00:00
[[ $UPDATED_VIRTUAL_ALIAS_DOMAINS == 1 ]] && {
service postfix reload
}
exit $exit_code
2014-11-27 19:17:26 +00:00
""") % context
)
2014-08-22 15:31:44 +00:00
def get_context_files(self):
return {
2014-10-17 10:04:47 +00:00
'virtual_alias_domains': settings.MAILBOXES_VIRTUAL_ALIAS_DOMAINS_PATH,
'virtual_alias_maps': settings.MAILBOXES_VIRTUAL_ALIAS_MAPS_PATH
2014-08-22 15:31:44 +00:00
}
def get_context(self, address):
context = self.get_context_files()
context.update({
'domain': address.domain,
'email': address.email,
2015-04-09 14:32:10 +00:00
'local_domain': settings.MAILBOXES_LOCAL_DOMAIN,
2014-08-22 15:31:44 +00:00
})
return context
2014-08-22 15:31:44 +00:00
2015-05-05 19:42:55 +00:00
class PostfixAddressBackend(PostfixAddressVirtualDomainBackend):
"""
Addresses based on Postfix virtual alias domains, includes <tt>PostfixAddressVirtualDomainBackend</tt>.
"""
verbose_name = _("Postfix address")
2015-05-22 13:15:06 +00:00
doc_settings = (settings, (
'MAILBOXES_LOCAL_DOMAIN',
'MAILBOXES_VIRTUAL_ALIAS_DOMAINS_PATH',
'MAILBOXES_VIRTUAL_ALIAS_MAPS_PATH'
))
2015-05-05 19:42:55 +00:00
def update_virtual_alias_maps(self, address, context):
destination = address.destination
if destination:
context['destination'] = destination
self.append(textwrap.dedent("""
# Set virtual alias entry for %(email)s
2015-05-05 19:42:55 +00:00
LINE='%(email)s\t%(destination)s'
if [[ ! $(grep '^%(email)s\s' %(virtual_alias_maps)s) ]]; then
# Add new line
echo "${LINE}" >> %(virtual_alias_maps)s
UPDATED_VIRTUAL_ALIAS_MAPS=1
else
# Update existing line, if needed
if [[ ! $(grep "^${LINE}$" %(virtual_alias_maps)s) ]]; then
sed -i "s/^%(email)s\s.*$/${LINE}/" %(virtual_alias_maps)s
UPDATED_VIRTUAL_ALIAS_MAPS=1
fi
fi""") % context)
else:
logger.warning("Address %i is empty" % address.pk)
self.exclude_virtual_alias_maps(context)
2015-05-05 19:42:55 +00:00
# Virtual mailbox stuff
# destination = []
# for mailbox in address.get_mailboxes():
# context['mailbox'] = mailbox
# destination.append("%(mailbox)s@%(local_domain)s" % context)
# for forward in address.forward:
# if '@' in forward:
# destination.append(forward)
def exclude_virtual_alias_maps(self, context):
self.append(textwrap.dedent("""
# Remove %(email)s virtual alias entry
if [[ $(grep '^%(email)s\s' %(virtual_alias_maps)s) ]]; then
sed -i '/^%(email)s\s/d' %(virtual_alias_maps)s
2015-05-05 19:42:55 +00:00
UPDATED_VIRTUAL_ALIAS_MAPS=1
fi""") % context
)
2015-05-05 19:42:55 +00:00
def save(self, address):
context = super(PostfixAddressBackend, self).save(address)
self.update_virtual_alias_maps(address, context)
def delete(self, address):
context = super(PostfixAddressBackend, self).save(address)
self.exclude_virtual_alias_maps(context)
def commit(self):
context = self.get_context_files()
self.append(textwrap.dedent("""
# Apply changes if needed
2015-05-05 19:42:55 +00:00
[[ $UPDATED_VIRTUAL_ALIAS_DOMAINS == 1 ]] && {
service postfix reload
}
[[ $UPDATED_VIRTUAL_ALIAS_MAPS == 1 ]] && {
postmap %(virtual_alias_maps)s
}
exit $exit_code
""") % context
)
2014-08-22 15:31:44 +00:00
class AutoresponseBackend(ServiceController):
2015-04-23 19:46:23 +00:00
"""
WARNING: not implemented
"""
2014-08-22 15:31:44 +00:00
verbose_name = _("Mail autoresponse")
2015-04-05 18:02:36 +00:00
model = 'mailboxes.Autoresponse'
2014-08-22 15:31:44 +00:00
2015-04-05 18:02:36 +00:00
class DovecotMaildirDisk(ServiceMonitor):
2014-11-18 13:59:21 +00:00
"""
Maildir disk usage based on Dovecot maildirsize file
http://wiki2.dovecot.org/Quota/Maildir
"""
2014-10-17 10:04:47 +00:00
model = 'mailboxes.Mailbox'
2014-08-22 15:31:44 +00:00
resource = ServiceMonitor.DISK
2015-04-05 18:02:36 +00:00
verbose_name = _("Dovecot Maildir size")
delete_old_equal_values = True
2015-04-24 11:39:20 +00:00
doc_settings = (settings,
('MAILBOXES_MAILDIRSIZE_PATH',)
)
2014-08-22 15:31:44 +00:00
2014-11-18 13:59:21 +00:00
def prepare(self):
2015-04-05 18:02:36 +00:00
super(DovecotMaildirDisk, self).prepare()
2014-11-18 13:59:21 +00:00
current_date = self.current_date.strftime("%Y-%m-%d %H:%M:%S %Z")
self.append(textwrap.dedent("""\
function monitor () {
2015-03-23 15:36:51 +00:00
awk 'BEGIN { size = 0 } NR > 1 { size += $1 } END { print size }' $1 || echo 0
2014-11-18 13:59:21 +00:00
}"""))
2014-08-22 15:31:44 +00:00
def monitor(self, mailbox):
context = self.get_context(mailbox)
2014-11-18 13:59:21 +00:00
self.append("echo %(object_id)s $(monitor %(maildir_path)s)" % context)
2014-08-22 15:31:44 +00:00
def get_context(self, mailbox):
2014-10-23 21:25:44 +00:00
context = {
2014-11-17 14:17:33 +00:00
'home': mailbox.get_home(),
2014-10-06 14:57:02 +00:00
'object_id': mailbox.pk
2014-10-23 21:25:44 +00:00
}
2014-11-17 14:17:33 +00:00
context['maildir_path'] = settings.MAILBOXES_MAILDIRSIZE_PATH % context
return context
2015-04-05 18:02:36 +00:00
class PostfixMailscannerTraffic(ServiceMonitor):
"""
2015-04-23 19:46:23 +00:00
A high-performance log parser.
Reads the mail.log file only once, for all users.
"""
model = 'mailboxes.Mailbox'
resource = ServiceMonitor.TRAFFIC
2015-04-05 18:02:36 +00:00
verbose_name = _("Postfix-Mailscanner traffic")
script_executable = '/usr/bin/python'
monthly_sum_old_values = True
2015-04-24 11:39:20 +00:00
doc_settings = (settings,
('MAILBOXES_MAIL_LOG_PATH',)
)
def prepare(self):
2015-04-05 18:02:36 +00:00
mail_log = settings.MAILBOXES_MAIL_LOG_PATH
context = {
'current_date': self.current_date.strftime("%Y-%m-%d %H:%M:%S %Z"),
'mail_logs': str((mail_log, mail_log+'.1')),
}
self.append(textwrap.dedent("""\
import re
import sys
from datetime import datetime
from dateutil import tz
def to_local_timezone(date, tzlocal=tz.tzlocal()):
# Converts orchestra's UTC dates to local timezone
date = datetime.strptime(date, '%Y-%m-%d %H:%M:%S %Z')
date = date.replace(tzinfo=tz.tzutc())
date = date.astimezone(tzlocal)
return date
maillogs = {mail_logs}
end_datetime = to_local_timezone('{current_date}')
end_date = int(end_datetime.strftime('%Y%m%d%H%M%S'))
2015-04-29 10:51:30 +00:00
months = ('Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec')
months = dict((m, '%02d' % n) for n, m in enumerate(months, 1))
def inside_period(month, day, time, ini_date):
global months
global end_datetime
2015-04-09 14:32:10 +00:00
# Mar 9 17:13:22
month = months[month]
year = end_datetime.year
if month == '12' and end_datetime.month == 1:
year = year+1
2015-04-09 14:32:10 +00:00
if len(day) == 1:
day = '0' + day
date = str(year) + month + day
date += time.replace(':', '')
return ini_date < int(date) < end_date
users = {{}}
delivers = {{}}
reverse = {{}}
def prepare(object_id, mailbox, ini_date):
global users
global delivers
global reverse
ini_date = to_local_timezone(ini_date)
ini_date = int(ini_date.strftime('%Y%m%d%H%M%S'))
users[mailbox] = (ini_date, object_id)
delivers[mailbox] = set()
reverse[mailbox] = set()
def monitor(users, delivers, reverse, maillogs):
targets = {{}}
counter = {{}}
user_regex = re.compile(r'\(Authenticated sender: ([^ ]+)\)')
for maillog in maillogs:
try:
with open(maillog, 'r') as maillog:
for line in maillog.readlines():
# Only search for Authenticated sendings
if '(Authenticated sender: ' in line:
username = user_regex.search(line).groups()[0]
try:
sender = users[username]
except KeyError:
continue
else:
month, day, time, __, proc, id = line.split()[:6]
if inside_period(month, day, time, sender[0]):
# Add new email
delivers[id[:-1]] = username
# Look for a MailScanner requeue ID
elif ' Requeue: ' in line:
id, __, req_id = line.split()[6:9]
id = id.split('.')[0]
try:
username = delivers[id]
except KeyError:
pass
else:
2015-04-16 13:15:21 +00:00
targets[req_id] = (username, 0)
reverse[username].add(req_id)
# Look for the mail size and count the number of recipients of each email
else:
try:
month, day, time, __, proc, req_id, __, msize = line.split()[:8]
except ValueError:
# not interested in this line
continue
if proc.startswith('postfix/'):
req_id = req_id[:-1]
if msize.startswith('size='):
try:
target = targets[req_id]
except KeyError:
pass
else:
targets[req_id] = (target[0], int(msize[5:-1]))
elif proc.startswith('postfix/smtp'):
try:
target = targets[req_id]
except KeyError:
pass
else:
if inside_period(month, day, time, users[target[0]][0]):
try:
counter[req_id] += 1
except KeyError:
counter[req_id] = 1
except IOError as e:
sys.stderr.write(str(e))
2015-04-16 13:15:21 +00:00
for username, opts in users.iteritems():
size = 0
for req_id in reverse[username]:
size += targets[req_id][1] * counter.get(req_id, 0)
print opts[1], size
""").format(**context)
)
def commit(self):
self.append('monitor(users, delivers, reverse, maillogs)')
def monitor(self, mailbox):
context = self.get_context(mailbox)
self.append("prepare(%(object_id)s, '%(mailbox)s', '%(last_date)s')" % context)
def get_context(self, mailbox):
2015-04-05 18:02:36 +00:00
context = {
'mailbox': mailbox.name,
'object_id': mailbox.pk,
'last_date': self.get_last_date(mailbox.pk).strftime("%Y-%m-%d %H:%M:%S %Z"),
}
return context