from django import forms from django.core.exceptions import ValidationError from django.core.urlresolvers import reverse from django.utils.safestring import mark_safe from django.utils.translation import ugettext_lazy as _ from orchestra.contrib.databases.models import Database, DatabaseUser from orchestra.forms.widgets import SpanWidget from .. import settings from ..forms import SaaSPasswordForm from .options import SoftwareService class PHPListForm(SaaSPasswordForm): admin_username = forms.CharField(label=_("Admin username"), required=False, widget=SpanWidget(display='admin')) def __init__(self, *args, **kwargs): super(PHPListForm, self).__init__(*args, **kwargs) self.fields['name'].label = _("Site name") base_domain = self.plugin.site_base_domain help_text = _("Admin URL http://<site_name>.{}/admin/").format(base_domain) self.fields['site_url'].help_text = help_text class PHPListChangeForm(PHPListForm): database = forms.CharField(label=_("Database"), required=False, help_text=_("Database used for this webapp.")) def __init__(self, *args, **kwargs): super(PHPListChangeForm, self).__init__(*args, **kwargs) site_domain = self.instance.get_site_domain() admin_url = "http://%s/admin/" % site_domain help_text = _("Admin URL {0}").format(admin_url) self.fields['site_url'].help_text = help_text # DB link db = self.instance.database db_url = reverse('admin:databases_database_change', args=(db.pk,)) db_link = mark_safe('%s' % (db_url, db.name)) self.fields['database'].widget = SpanWidget(original=db.name, display=db_link) class PHPListService(SoftwareService): name = 'phplist' verbose_name = "phpList" form = PHPListForm change_form = PHPListChangeForm icon = 'orchestra/icons/apps/Phplist.png' site_base_domain = settings.SAAS_PHPLIST_BASE_DOMAIN def get_db_name(self): db_name = 'phplist_mu_%s' % self.instance.name # Limit for mysql database names return db_name[:65] def get_db_user(self): return settings.SAAS_PHPLIST_DB_USER def get_account(self): account_model = self.instance._meta.get_field_by_name('account')[0] return account_model.rel.to.objects.get_main() def validate(self): super(PHPListService, self).validate() create = not self.instance.pk if create: db_user = self.get_db_user() try: DatabaseUser.objects.get(username=db_user) except DatabaseUser.DoesNotExist: raise ValidationError(_("Global database user for PHPList '%s' does not exists.")) account = self.get_account() db = Database(name=self.get_db_name(), account=account) try: db.full_clean() except ValidationError as e: raise ValidationError({ 'name': e.messages, }) def save(self): db_name = self.get_db_name() db_user = self.get_db_user() account = self.get_account() db, db_created = account.databases.get_or_create(name=db_name, type=Database.MYSQL) user = DatabaseUser.objects.get(username=db_user) db.users.add(user) self.instance.database_id = db.pk