django-orchestra-test/orchestra/contrib/saas/backends/phplist.py

123 lines
4.8 KiB
Python
Raw Normal View History

import hashlib
2015-03-23 15:36:51 +00:00
import re
2015-07-21 12:23:40 +00:00
import sys
import textwrap
2015-03-23 15:36:51 +00:00
import requests
from django.utils.translation import ugettext_lazy as _
2015-04-05 10:46:24 +00:00
from orchestra.contrib.orchestration import ServiceController
from orchestra.utils.sys import sshrun
2015-03-23 15:36:51 +00:00
2015-07-09 13:04:26 +00:00
from .. import settings
2015-03-23 15:36:51 +00:00
class PhpListSaaSBackend(ServiceController):
2015-04-24 11:39:20 +00:00
"""
Creates a new phplist instance on a phpList multisite installation.
The site is created by means of creating a new database per phpList site,
but all sites share the same code.
2015-04-24 11:39:20 +00:00
<tt>// config/config.php
$site = getenv("SITE");
if ( $site == '' ) {
$site = array_shift((explode(".",$_SERVER['HTTP_HOST'])));
}
2015-04-24 11:39:20 +00:00
$database_name = "phplist_mu_{$site}";</tt>
"""
2015-03-23 15:36:51 +00:00
verbose_name = _("phpList SaaS")
model = 'saas.SaaS'
default_route_match = "saas.service == 'phplist'"
serialize = True
2015-03-23 15:36:51 +00:00
2015-07-21 12:23:40 +00:00
def error(self, msg):
sys.stderr.write(msg + '\n')
raise RuntimeError(msg)
2015-03-29 16:10:07 +00:00
def _save(self, saas, server):
2015-07-09 13:04:26 +00:00
admin_link = 'https://%s/admin/' % saas.get_site_domain()
2015-07-21 12:23:40 +00:00
sys.stdout.write('admin_link: %s\n' % admin_link)
admin_content = requests.get(admin_link, verify=settings.SAAS_PHPLIST_VERIFY_SSL)
admin_content = admin_content.content.decode('utf8')
2015-03-23 15:36:51 +00:00
if admin_content.startswith('Cannot connect to Database'):
2015-07-21 12:23:40 +00:00
self.error("Database is not yet configured.")
2015-03-23 15:36:51 +00:00
install = re.search(r'([^"]+firstinstall[^"]+)', admin_content)
if install:
2015-03-27 19:50:54 +00:00
if not hasattr(saas, 'password'):
2015-07-21 12:23:40 +00:00
self.error("Password is missing.")
2015-03-29 16:10:07 +00:00
install_path = install.groups()[0]
install_link = admin_link + install_path[1:]
2015-03-23 15:36:51 +00:00
post = {
'adminname': saas.name,
2015-03-23 15:36:51 +00:00
'orgname': saas.account.username,
'adminemail': saas.account.username,
'adminpassword': saas.password,
}
2015-07-09 13:04:26 +00:00
response = requests.post(install_link, data=post, verify=settings.SAAS_PHPLIST_VERIFY_SSL)
2015-07-21 12:23:40 +00:00
sys.stdout.write(response.content.decode('utf8')+'\n')
2015-03-23 15:36:51 +00:00
if response.status_code != 200:
2015-07-21 12:23:40 +00:00
self.error("Bad status code %i." % response.status_code)
2015-03-29 16:10:07 +00:00
else:
md5_password = hashlib.md5()
md5_password.update(saas.password.encode('ascii'))
context = self.get_context(saas)
context['digest'] = md5_password.hexdigest()
cmd = textwrap.dedent("""\
mysql \\
--host=%(db_host)s \\
--user=%(db_user)s \\
--password=%(db_pass)s \\
--execute='UPDATE phplist_admin SET password="%(digest)s" where ID=1; \\
UPDATE phplist_user_user SET password="%(digest)s" where ID=1;' \\
%(db_name)s""") % context
2015-07-21 12:23:40 +00:00
sys.stdout.write('cmd: %s\n' % cmd)
sshrun(server.get_address(), cmd)
2015-03-23 15:36:51 +00:00
def save(self, saas):
2015-03-29 16:10:07 +00:00
if hasattr(saas, 'password'):
self.append(self._save, saas)
context = self.get_context(saas)
if context['crontab']:
context['escaped_crontab'] = context['crontab'].replace('$', '\\$')
self.append(textwrap.dedent("""\
# Configuring phpList crontabs
if [[ ! $(crontab -u %(user)s -l | grep 'phpList:"%(site_name)s"') ]]; then
cat << EOF | su %(user)s --shell /bin/bash -c 'crontab'
$(crontab -u %(user)s -l)
# %(banner)s - phpList:"%(site_name)s"
%(escaped_crontab)s
EOF
fi""") % context
)
def delete(self, saas):
context = self.get_context(saas)
if context['crontab']:
context['crontab_regex'] = '\\|'.join(context['crontab'].splitlines())
context['crontab_regex'] = context['crontab_regex'].replace('*', '\\*')
self.append(textwrap.dedent("""\
crontab -u %(user)s -l \\
| grep -v 'phpList:"%(site_name)s"\\|%(crontab_regex)s' \\
| su %(user)s --shell /bin/bash -c 'crontab'
""") % context
)
def get_context(self, saas):
context = {
'banner': self.get_banner(),
'name': saas.name,
'site_name': saas.name,
'phplist_path': settings.SAAS_PHPLIST_PATH,
'user': settings.SAAS_PHPLIST_SYSTEMUSER,
'db_user': settings.SAAS_PHPLIST_DB_USER,
'db_pass': settings.SAAS_PHPLIST_DB_PASS,
'db_name': settings.SAAS_PHPLIST_DB_NAME,
'db_host': settings.SAAS_PHPLIST_DB_HOST,
}
context.update({
'crontab': settings.SAAS_PHPLIST_CRONTAB % context,
'db_name': context['db_name'] % context,
})
return context