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

337 lines
13 KiB
Python
Raw Normal View History

2015-05-05 19:42:55 +00:00
import re
2014-07-25 15:17:50 +00:00
import textwrap
from django.utils.translation import ugettext_lazy as _
2014-05-08 16:59:35 +00:00
2015-05-04 10:48:09 +00:00
from orchestra.contrib.domains.models import Domain, Record
2015-04-05 18:02:36 +00:00
from orchestra.contrib.orchestration import ServiceController, replace
2015-04-05 10:46:24 +00:00
from orchestra.contrib.resources import ServiceMonitor
2014-05-08 16:59:35 +00:00
2014-07-11 14:48:46 +00:00
from . import settings
2014-10-09 17:04:12 +00:00
from .models import List
2014-07-11 14:48:46 +00:00
2014-05-08 16:59:35 +00:00
2015-05-05 19:42:55 +00:00
class MailmanVirtualDomainBackend(ServiceController):
"""
Only syncs virtualdomains used on mailman addresses
"""
verbose_name = _("Mailman virtdomain-only")
model = 'lists.List'
doc_settings = (settings,
('LISTS_VIRTUAL_ALIAS_DOMAINS_PATH',)
)
def is_local_domain(self, domain):
""" whether or not domain MX points to this server """
return domain.has_default_mx()
def include_virtual_alias_domain(self, context):
domain = context['address_domain']
if domain and self.is_local_domain(domain):
self.append(textwrap.dedent("""
[[ $(grep '^\s*%(address_domain)s\s*$' %(virtual_alias_domains)s) ]] || {
echo '%(address_domain)s' >> %(virtual_alias_domains)s
UPDATED_VIRTUAL_ALIAS_DOMAINS=1
}""") % self.context
)
def is_last_domain(self, domain):
return not List.objects.filter(address_domain=domain).exists()
def exclude_virtual_alias_domain(self, context):
domain = context['address_domain']
if domain and self.is_last_domain(domain):
self.append("sed -i '/^%(address_domain)s\s*$/d' %(virtual_alias_domains)s" % context)
def save(self, mail_list):
context = self.get_context(mail_list)
self.include_virtual_alias_domain(context)
def delete(self, mail_list):
context = self.get_context(mail_list)
self.include_virtual_alias_domain(context)
def commit(self):
context = self.get_context_files()
self.append(textwrap.dedent("""
if [[ $UPDATED_VIRTUAL_ALIAS_DOMAINS == 1 ]]; then
service postfix reload
fi""") % context
)
def get_context_files(self):
return {
'virtual_alias_domains': settings.LISTS_VIRTUAL_ALIAS_DOMAINS_PATH,
}
def get_context(self, mail_list):
context = self.get_context_files()
context.update({
'address_domain': mail_list.address_domain,
})
return replace(context, "'", '"')
class MailmanBackend(MailmanVirtualDomainBackend):
2015-04-23 19:46:23 +00:00
"""
2015-04-24 11:39:20 +00:00
Mailman 2 backend based on <tt>newlist</tt>, it handles custom domains.
2015-05-05 19:42:55 +00:00
Includes <tt>MailmanVirtualDomainBackend</tt>
2015-04-23 19:46:23 +00:00
"""
2014-05-08 16:59:35 +00:00
verbose_name = "Mailman"
2015-05-12 12:38:40 +00:00
address_suffixes = [
2014-10-15 12:47:28 +00:00
'',
'-admin',
'-bounces',
'-confirm',
'-join',
'-leave',
'-owner',
'-request',
'-subscribe',
'-unsubscribe'
]
2015-04-24 11:39:20 +00:00
doc_settings = (settings, (
'LISTS_VIRTUAL_ALIAS_PATH',
'LISTS_VIRTUAL_ALIAS_DOMAINS_PATH',
'LISTS_DEFAULT_DOMAIN',
'LISTS_MAILMAN_ROOT_DIR'
))
2014-10-09 17:04:12 +00:00
def get_virtual_aliases(self, context):
2015-03-18 21:51:12 +00:00
aliases = ['# %(banner)s' % context]
2015-05-12 12:38:40 +00:00
for suffix in self.address_suffixes:
context['suffix'] = suffix
# Because mailman doesn't properly handle lists aliases we need two virtual aliases
aliases.append("%(address_name)s%(suffix)s@%(domain)s\t%(name)s%(suffix)s" % context)
# And another with the original list name; Mailman generates links with it
aliases.append("%(name)s%(suffix)s@%(domain)s\t%(name)s%(suffix)s" % context)
2014-10-09 17:04:12 +00:00
return '\n'.join(aliases)
def save(self, mail_list):
context = self.get_context(mail_list)
2014-10-15 12:47:28 +00:00
# Create list
self.append(textwrap.dedent("""\
2015-04-05 18:02:36 +00:00
[[ ! -e '%(mailman_root)s/lists/%(name)s' ]] && {
2014-10-15 12:47:28 +00:00
newlist --quiet --emailhost='%(domain)s' '%(name)s' '%(admin)s' '%(password)s'
2014-11-27 19:17:26 +00:00
}""") % context)
2014-10-15 12:47:28 +00:00
# Custom domain
2014-10-09 17:04:12 +00:00
if mail_list.address:
2014-11-27 19:17:26 +00:00
context['aliases'] = self.get_virtual_aliases(context)
2014-10-15 12:47:28 +00:00
# Preserve indentation
self.append(textwrap.dedent("""\
aliases='%(aliases)s'
2014-10-15 12:47:28 +00:00
if [[ ! $(grep '\s\s*%(name)s\s*$' %(virtual_alias)s) ]]; then
echo "${aliases}" >> %(virtual_alias)s
2014-10-15 12:47:28 +00:00
UPDATED_VIRTUAL_ALIAS=1
else
if [[ ! $(grep '^\s*%(address_name)s@%(address_domain)s\s\s*%(name)s\s*$' %(virtual_alias)s) ]]; then
2015-05-12 12:38:40 +00:00
sed -i -e '/^.*\s%(name)s\(%(suffixes_regex)s\)\s*$/d' \\
2014-10-27 15:15:22 +00:00
-e 'N; /^\s*\\n\s*$/d; P; D' %(virtual_alias)s
echo "${aliases}" >> %(virtual_alias)s
2014-10-15 12:47:28 +00:00
UPDATED_VIRTUAL_ALIAS=1
fi
2014-11-27 19:17:26 +00:00
fi""") % context
)
self.append(
'echo "require_explicit_destination = 0" | '
'%(mailman_root)s/bin/config_list -i /dev/stdin %(name)s' % context
)
2014-10-15 12:47:28 +00:00
self.append(textwrap.dedent("""\
echo "host_name = '%(address_domain)s'" | \
2014-11-27 19:17:26 +00:00
%(mailman_root)s/bin/config_list -i /dev/stdin %(name)s""") % context
)
2014-10-15 12:47:28 +00:00
else:
# Cleanup shit
self.append(textwrap.dedent("""\
if [[ ! $(grep '\s\s*%(name)s\s*$' %(virtual_alias)s) ]]; then
2014-10-27 14:31:04 +00:00
sed -i "/^.*\s%(name)s\s*$/d" %(virtual_alias)s
2014-11-27 19:17:26 +00:00
fi""") % context
)
2014-10-15 12:47:28 +00:00
# Update
if context['password'] is not None:
2014-11-27 19:17:26 +00:00
self.append(
'%(mailman_root)s/bin/change_pw --listname="%(name)s" --password="%(password)s"' % context
)
2014-10-09 17:04:12 +00:00
self.include_virtual_alias_domain(context)
2015-03-18 21:51:12 +00:00
if mail_list.active:
self.append('chmod 775 %(mailman_root)s/lists/%(name)s' % context)
else:
self.append('chmod 000 %(mailman_root)s/lists/%(name)s' % context)
2014-10-09 17:04:12 +00:00
def delete(self, mail_list):
2014-10-15 12:47:28 +00:00
context = self.get_context(mail_list)
self.exclude_virtual_alias_domain(context)
2015-03-18 21:51:12 +00:00
self.append(textwrap.dedent("""
2015-05-12 12:38:40 +00:00
sed -i -e '/^.*\s%(name)s\(%(suffixes_regex)s\)\s*$/d' \\
2014-11-27 19:17:26 +00:00
-e 'N; /^\s*\\n\s*$/d; P; D' %(virtual_alias)s""") % context
)
2015-03-18 21:51:12 +00:00
self.append(textwrap.dedent("""
# Non-existent list archives produce exit code 1
exit_code=0
rmlist -a %(name)s || exit_code=$?
if [[ $exit_code != 0 && $exit_code != 1 ]]; then
exit $exit_code
fi""") % context
)
2014-10-09 17:04:12 +00:00
def commit(self):
context = self.get_context_files()
self.append(textwrap.dedent("""
2015-03-18 21:51:12 +00:00
if [[ $UPDATED_VIRTUAL_ALIAS == 1 ]]; then
postmap %(virtual_alias)s
fi
if [[ $UPDATED_VIRTUAL_ALIAS_DOMAINS == 1 ]]; then
2015-04-09 14:32:10 +00:00
service postfix reload
2015-03-18 21:51:12 +00:00
fi""") % context
2014-11-27 19:17:26 +00:00
)
2014-10-09 17:04:12 +00:00
def get_context_files(self):
return {
'virtual_alias': settings.LISTS_VIRTUAL_ALIAS_PATH,
2014-10-15 12:47:28 +00:00
'virtual_alias_domains': settings.LISTS_VIRTUAL_ALIAS_DOMAINS_PATH,
2014-10-09 17:04:12 +00:00
}
2014-10-27 14:31:04 +00:00
def get_banner(self, mail_list):
banner = super(MailmanBackend, self).get_banner()
return '%s %s' % (banner, mail_list.name)
2014-10-09 17:04:12 +00:00
def get_context(self, mail_list):
context = self.get_context_files()
context.update({
2014-10-27 15:15:22 +00:00
'banner': self.get_banner(mail_list),
2014-10-09 17:04:12 +00:00
'name': mail_list.name,
'password': mail_list.password,
'domain': mail_list.address_domain or settings.LISTS_DEFAULT_DOMAIN,
2014-10-23 21:25:44 +00:00
'address_name': mail_list.get_address_name(),
2014-10-09 17:04:12 +00:00
'address_domain': mail_list.address_domain,
2015-05-12 12:38:40 +00:00
'suffixes_regex': '\|'.join(self.address_suffixes),
2014-10-09 17:04:12 +00:00
'admin': mail_list.admin_email,
2015-04-16 13:15:21 +00:00
'mailman_root': settings.LISTS_MAILMAN_ROOT_DIR,
2014-10-09 17:04:12 +00:00
})
2015-04-05 18:02:36 +00:00
return replace(context, "'", '"')
2014-07-09 16:17:43 +00:00
class MailmanTraffic(ServiceMonitor):
2015-04-23 19:46:23 +00:00
"""
2015-04-24 11:39:20 +00:00
Parses mailman log file looking for email size and multiples it by <tt>list_members</tt> count.
2015-04-23 19:46:23 +00:00
"""
model = 'lists.List'
resource = ServiceMonitor.TRAFFIC
verbose_name = _("Mailman traffic")
script_executable = '/usr/bin/python'
2015-04-24 11:39:20 +00:00
doc_settings = (settings,
('LISTS_MAILMAN_POST_LOG_PATH',)
)
def prepare(self):
postlog = settings.LISTS_MAILMAN_POST_LOG_PATH
context = {
'postlogs': str((postlog, postlog+'.1')),
'current_date': self.current_date.strftime("%Y-%m-%d %H:%M:%S %Z"),
}
self.append(textwrap.dedent("""\
import re
import subprocess
import sys
from datetime import datetime
from dateutil import tz
2015-04-23 19:46:23 +00:00
def to_local_timezone(date, tzlocal=tz.tzlocal()):
date = datetime.strptime(date, '%Y-%m-%d %H:%M:%S %Z')
date = date.replace(tzinfo=tz.tzutc())
date = date.astimezone(tzlocal)
return date
2015-04-23 19:46:23 +00:00
postlogs = {postlogs}
# Use local timezone
end_date = to_local_timezone('{current_date}')
end_date = int(end_date.strftime('%Y%m%d%H%M%S'))
lists = {{}}
months = {{
'Jan': '01',
'Feb': '02',
'Mar': '03',
'Apr': '04',
'May': '05',
'Jun': '06',
'Jul': '07',
'Aug': '08',
'Sep': '09',
'Oct': '10',
'Nov': '11',
'Dec': '12',
}}
2015-04-23 19:46:23 +00:00
def prepare(object_id, list_name, ini_date):
global lists
ini_date = to_local_timezone(ini_date)
ini_date = int(ini_date.strftime('%Y%m%d%H%M%S'))
lists[list_name] = [ini_date, object_id, 0]
2015-04-23 19:46:23 +00:00
def monitor(lists, end_date, months, postlogs):
for postlog in postlogs:
try:
with open(postlog, 'r') as postlog:
for line in postlog.readlines():
month, day, time, year, __, __, __, list_name, __, __, size = line.split()[:11]
try:
list = lists[list_name]
except KeyError:
continue
else:
date = year + months[month] + day + time.replace(':', '')
if list[0] < int(date) < end_date:
size = size[5:-1]
try:
list[2] += int(size)
except ValueError:
# anonymized post
pass
except IOError as e:
sys.stderr.write(e)
2015-04-02 16:14:55 +00:00
for list_name, opts in lists.items():
__, object_id, size = opts
if size:
cmd = ' '.join(('list_members', list_name, '| wc -l'))
ps = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
subscribers = ps.communicate()[0].strip()
size *= int(subscribers)
print object_id, size
""").format(**context)
2015-03-02 10:37:25 +00:00
)
2014-07-11 14:48:46 +00:00
def monitor(self, user):
context = self.get_context(user)
self.append("prepare(%(object_id)s, '%(list_name)s', '%(last_date)s')" % context)
def commit(self):
self.append('monitor(lists, end_date, months, postlogs)')
2014-07-11 14:48:46 +00:00
def get_context(self, mail_list):
2015-04-05 18:02:36 +00:00
context = {
2014-07-11 14:48:46 +00:00
'list_name': mail_list.name,
'object_id': mail_list.pk,
2015-03-02 10:37:25 +00:00
'last_date': self.get_last_date(mail_list.pk).strftime("%Y-%m-%d %H:%M:%S %Z"),
2014-07-11 14:48:46 +00:00
}
2015-04-05 18:02:36 +00:00
return replace(context, "'", '"')
2014-10-27 13:29:02 +00:00
class MailmanSubscribers(ServiceMonitor):
2015-04-23 19:46:23 +00:00
"""
2015-04-24 11:39:20 +00:00
Monitors number of list subscribers via <tt>list_members</tt>
2015-04-23 19:46:23 +00:00
"""
model = 'lists.List'
verbose_name = _("Mailman subscribers")
def monitor(self, mail_list):
context = self.get_context(mail_list)
self.append('echo %(object_id)i $(list_members %(list_name)s | wc -l)' % context)
def get_context(self, mail_list):
2015-04-05 18:02:36 +00:00
context = {
'list_name': mail_list.name,
'object_id': mail_list.pk,
}
2015-04-05 18:02:36 +00:00
return replace(context, "'", '"')