290 lines
11 KiB
Python
290 lines
11 KiB
Python
import textwrap
|
|
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from orchestra.contrib.orchestration import ServiceController, replace
|
|
from orchestra.contrib.resources import ServiceMonitor
|
|
|
|
from . import settings
|
|
from .models import List
|
|
|
|
|
|
class MailmanBackend(ServiceController):
|
|
"""
|
|
Mailman backend based on `newlist`, it handles custom domains.
|
|
LISTS_VIRTUAL_ALIAS_PATH = '%s'
|
|
LISTS_VIRTUAL_ALIAS_DOMAINS_PATH = '%s'
|
|
LISTS_DEFAULT_DOMAIN = '%s'
|
|
LISTS_MAILMAN_ROOT_DIR = '%s'
|
|
"""
|
|
format_docstring = (
|
|
settings.LISTS_VIRTUAL_ALIAS_PATH,
|
|
settings.LISTS_VIRTUAL_ALIAS_DOMAINS_PATH,
|
|
settings.LISTS_DEFAULT_DOMAIN,
|
|
settings.LISTS_MAILMAN_ROOT_DIR,
|
|
)
|
|
verbose_name = "Mailman"
|
|
model = 'lists.List'
|
|
addresses = [
|
|
'',
|
|
'-admin',
|
|
'-bounces',
|
|
'-confirm',
|
|
'-join',
|
|
'-leave',
|
|
'-owner',
|
|
'-request',
|
|
'-subscribe',
|
|
'-unsubscribe'
|
|
]
|
|
|
|
def include_virtual_alias_domain(self, context):
|
|
if context['address_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
|
|
}""") % context
|
|
)
|
|
|
|
def exclude_virtual_alias_domain(self, context):
|
|
address_domain = context['address_domain']
|
|
if not List.objects.filter(address_domain=address_domain).exists():
|
|
self.append("sed -i '/^%(address_domain)s\s*$/d' %(virtual_alias_domains)s" % context)
|
|
|
|
def get_virtual_aliases(self, context):
|
|
aliases = ['# %(banner)s' % context]
|
|
for address in self.addresses:
|
|
context['address'] = address
|
|
aliases.append("%(address_name)s%(address)s@%(domain)s\t%(name)s%(address)s" % context)
|
|
return '\n'.join(aliases)
|
|
|
|
def save(self, mail_list):
|
|
context = self.get_context(mail_list)
|
|
# Create list
|
|
self.append(textwrap.dedent("""\
|
|
[[ ! -e '%(mailman_root)s/lists/%(name)s' ]] && {
|
|
newlist --quiet --emailhost='%(domain)s' '%(name)s' '%(admin)s' '%(password)s'
|
|
}""") % context)
|
|
# Custom domain
|
|
if mail_list.address:
|
|
context['aliases'] = self.get_virtual_aliases(context)
|
|
# Preserve indentation
|
|
self.append(textwrap.dedent("""\
|
|
if [[ ! $(grep '\s\s*%(name)s\s*$' %(virtual_alias)s) ]]; then
|
|
echo '%(aliases)s' >> %(virtual_alias)s
|
|
UPDATED_VIRTUAL_ALIAS=1
|
|
else
|
|
if [[ ! $(grep '^\s*%(address_name)s@%(address_domain)s\s\s*%(name)s\s*$' %(virtual_alias)s) ]]; then
|
|
sed -i -e '/^.*\s%(name)s\(%(address_regex)s\)\s*$/d' \\
|
|
-e 'N; /^\s*\\n\s*$/d; P; D' %(virtual_alias)s
|
|
echo '%(aliases)s' >> %(virtual_alias)s
|
|
UPDATED_VIRTUAL_ALIAS=1
|
|
fi
|
|
fi""") % context
|
|
)
|
|
self.append(
|
|
'echo "require_explicit_destination = 0" | '
|
|
'%(mailman_root)s/bin/config_list -i /dev/stdin %(name)s' % context
|
|
)
|
|
self.append(textwrap.dedent("""\
|
|
echo "host_name = '%(address_domain)s'" | \
|
|
%(mailman_root)s/bin/config_list -i /dev/stdin %(name)s""") % context
|
|
)
|
|
else:
|
|
# Cleanup shit
|
|
self.append(textwrap.dedent("""\
|
|
if [[ ! $(grep '\s\s*%(name)s\s*$' %(virtual_alias)s) ]]; then
|
|
sed -i "/^.*\s%(name)s\s*$/d" %(virtual_alias)s
|
|
fi""") % context
|
|
)
|
|
# Update
|
|
if context['password'] is not None:
|
|
self.append(
|
|
'%(mailman_root)s/bin/change_pw --listname="%(name)s" --password="%(password)s"' % context
|
|
)
|
|
self.include_virtual_alias_domain(context)
|
|
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)
|
|
|
|
def delete(self, mail_list):
|
|
context = self.get_context(mail_list)
|
|
self.exclude_virtual_alias_domain(context)
|
|
self.append(textwrap.dedent("""
|
|
sed -i -e '/^.*\s%(name)s\(%(address_regex)s\)\s*$/d' \\
|
|
-e 'N; /^\s*\\n\s*$/d; P; D' %(virtual_alias)s""") % context
|
|
)
|
|
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
|
|
)
|
|
|
|
def commit(self):
|
|
context = self.get_context_files()
|
|
self.append(textwrap.dedent("""
|
|
if [[ $UPDATED_VIRTUAL_ALIAS == 1 ]]; then
|
|
postmap %(virtual_alias)s
|
|
fi
|
|
if [[ $UPDATED_VIRTUAL_ALIAS_DOMAINS == 1 ]]; then
|
|
service postfix reload
|
|
fi""") % context
|
|
)
|
|
|
|
def get_context_files(self):
|
|
return {
|
|
'virtual_alias': settings.LISTS_VIRTUAL_ALIAS_PATH,
|
|
'virtual_alias_domains': settings.LISTS_VIRTUAL_ALIAS_DOMAINS_PATH,
|
|
}
|
|
|
|
def get_banner(self, mail_list):
|
|
banner = super(MailmanBackend, self).get_banner()
|
|
return '%s %s' % (banner, mail_list.name)
|
|
|
|
def get_context(self, mail_list):
|
|
context = self.get_context_files()
|
|
context.update({
|
|
'banner': self.get_banner(mail_list),
|
|
'name': mail_list.name,
|
|
'password': mail_list.password,
|
|
'domain': mail_list.address_domain or settings.LISTS_DEFAULT_DOMAIN,
|
|
'address_name': mail_list.get_address_name(),
|
|
'address_domain': mail_list.address_domain,
|
|
'address_regex': '\|'.join(self.addresses),
|
|
'admin': mail_list.admin_email,
|
|
'mailman_root': settings.LISTS_MAILMAN_ROOT_DIR,
|
|
})
|
|
return replace(context, "'", '"')
|
|
|
|
|
|
|
|
class MailmanTraffic(ServiceMonitor):
|
|
"""
|
|
Parses mailman log file looking for email size and multiples it by `list_members` count.
|
|
LISTS_MAILMAN_POST_LOG_PATH = '%s'
|
|
"""
|
|
format_docstring = (
|
|
settings.LISTS_MAILMAN_POST_LOG_PATH,
|
|
)
|
|
model = 'lists.List'
|
|
resource = ServiceMonitor.TRAFFIC
|
|
verbose_name = _("Mailman traffic")
|
|
script_executable = '/usr/bin/python'
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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',
|
|
}}
|
|
|
|
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]
|
|
|
|
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)
|
|
|
|
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)
|
|
)
|
|
|
|
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)')
|
|
|
|
def get_context(self, mail_list):
|
|
context = {
|
|
'list_name': mail_list.name,
|
|
'object_id': mail_list.pk,
|
|
'last_date': self.get_last_date(mail_list.pk).strftime("%Y-%m-%d %H:%M:%S %Z"),
|
|
}
|
|
return replace(context, "'", '"')
|
|
|
|
|
|
class MailmanSubscribers(ServiceMonitor):
|
|
"""
|
|
Monitors number of list subscribers via `list_members`
|
|
"""
|
|
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):
|
|
context = {
|
|
'list_name': mail_list.name,
|
|
'object_id': mail_list.pk,
|
|
}
|
|
return replace(context, "'", '"')
|