django-orchestra-test/orchestra/contrib/saas/services/options.py

123 lines
3.9 KiB
Python
Raw Normal View History

2014-11-09 10:16:07 +00:00
from django.core.exceptions import ValidationError
2014-09-26 19:21:09 +00:00
from django.utils.translation import ugettext_lazy as _
from orchestra import plugins
from orchestra.contrib.databases.models import Database, DatabaseUser
2015-04-07 15:14:49 +00:00
from orchestra.contrib.orchestration import Operation
2014-09-26 19:21:09 +00:00
from orchestra.utils.functional import cached
2015-09-21 10:28:49 +00:00
from orchestra.utils.python import import_class
2014-09-26 19:21:09 +00:00
from .. import settings
2015-09-21 10:28:49 +00:00
from ..forms import SaaSPasswordForm
2014-11-20 15:34:59 +00:00
2014-09-26 19:21:09 +00:00
class SoftwareService(plugins.Plugin):
2015-09-21 10:28:49 +00:00
form = SaaSPasswordForm
site_domain = None
2015-03-23 15:36:51 +00:00
has_custom_domain = False
2014-10-11 12:43:08 +00:00
icon = 'orchestra/icons/apps.png'
class_verbose_name = _("Software as a Service")
2015-03-23 15:36:51 +00:00
plugin_field = 'service'
2014-09-26 19:21:09 +00:00
@classmethod
@cached
def get_plugins(cls):
plugins = []
for cls in settings.SAAS_ENABLED_SERVICES:
plugins.append(import_class(cls))
return plugins
2015-03-04 21:06:16 +00:00
def get_change_readonly_fileds(cls):
2015-03-23 15:36:51 +00:00
fields = super(SoftwareService, cls).get_change_readonly_fileds()
return fields + ('name',)
2015-03-04 21:06:16 +00:00
def get_site_domain(self):
2015-09-29 12:35:22 +00:00
context = {
'site_name': self.instance.name,
'name': self.instance.name,
}
return self.site_domain % context
2014-11-20 15:34:59 +00:00
2015-04-02 16:14:55 +00:00
def clean_data(self):
data = super(SoftwareService, self).clean_data()
if not self.instance.pk:
try:
log = Operation.execute_action(self.instance, 'validate_creation')[0]
except IndexError:
pass
else:
2015-04-20 14:23:10 +00:00
if log.state != log.SUCCESS:
raise ValidationError(_("Validate creation execution has failed."))
2015-04-02 16:14:55 +00:00
errors = {}
if 'user-exists' in log.stdout:
errors['name'] = _("User with this username already exists.")
2015-04-07 15:14:49 +00:00
if 'email-exists' in log.stdout:
2015-04-02 16:14:55 +00:00
errors['email'] = _("User with this email address already exists.")
if errors:
raise ValidationError(errors)
return data
2015-03-23 15:36:51 +00:00
def save(self):
pass
2014-09-26 19:21:09 +00:00
2015-03-23 15:36:51 +00:00
def delete(self):
pass
def get_related(self):
return []
class DBSoftwareService(SoftwareService):
db_name = None
db_user = None
def get_db_name(self):
context = {
'name': self.instance.name,
'site_name': self.instance.name,
}
db_name = self.db_name % context
# Limit for mysql database names
return db_name[:65]
def get_db_user(self):
return self.db_user
@cached
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(DBSoftwareService, self).validate()
create = not self.instance.pk
if create:
account = self.get_account()
# Validated Database
db_user = self.get_db_user()
try:
DatabaseUser.objects.get(username=db_user)
except DatabaseUser.DoesNotExist:
raise ValidationError(
_("Global database user for PHPList '%(db_user)s' does not exists.") % {
'db_user': db_user
}
)
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):
account = self.get_account()
# Database
db_name = self.get_db_name()
db_user = self.get_db_user()
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