django-orchestra-test/orchestra/contrib/webapps/models.py

125 lines
4.1 KiB
Python
Raw Normal View History

2015-03-10 21:51:10 +00:00
import os
2015-04-10 15:03:38 +00:00
from collections import OrderedDict
2014-05-08 16:59:35 +00:00
from django.db import models
2015-03-04 21:06:16 +00:00
from django.utils.functional import cached_property
2014-05-08 16:59:35 +00:00
from django.utils.translation import ugettext_lazy as _
2015-03-04 21:06:16 +00:00
from jsonfield import JSONField
2014-05-08 16:59:35 +00:00
2015-05-04 19:52:53 +00:00
from orchestra.core import validators
2014-05-08 16:59:35 +00:00
from . import settings
2015-03-27 19:50:54 +00:00
from .fields import VirtualDatabaseRelation, VirtualDatabaseUserRelation
from .options import AppOption
2015-03-04 21:06:16 +00:00
from .types import AppType
2014-05-08 16:59:35 +00:00
class WebApp(models.Model):
""" Represents a web application """
2015-04-16 13:15:21 +00:00
name = models.CharField(_("name"), max_length=128, validators=[validators.validate_name],
help_text=_("The app will be installed in %s") % settings.WEBAPPS_BASE_DIR)
2015-04-05 10:46:24 +00:00
type = models.CharField(_("type"), max_length=32, choices=AppType.get_choices())
2014-05-08 16:59:35 +00:00
account = models.ForeignKey('accounts.Account', verbose_name=_("Account"),
2015-04-05 10:46:24 +00:00
related_name='webapps')
2015-03-23 15:36:51 +00:00
data = JSONField(_("data"), blank=True, default={},
2015-04-05 10:46:24 +00:00
help_text=_("Extra information dependent of each service."))
2017-07-05 13:55:13 +00:00
target_server = models.ForeignKey('orchestration.Server', verbose_name=_("Target Server"),
related_name='webapps')
2014-05-08 16:59:35 +00:00
2015-03-27 19:50:54 +00:00
# CMS webapps usually need a database and dbuser, with these virtual fields we tell the ORM to delete them
databases = VirtualDatabaseRelation('databases.Database')
databaseusers = VirtualDatabaseUserRelation('databases.DatabaseUser')
2014-05-08 16:59:35 +00:00
class Meta:
unique_together = ('name', 'account')
verbose_name = _("Web App")
verbose_name_plural = _("Web Apps")
2015-04-02 16:14:55 +00:00
def __str__(self):
2015-03-02 10:37:25 +00:00
return self.name
2014-05-08 16:59:35 +00:00
2014-11-27 19:17:26 +00:00
def get_description(self):
return self.get_type_display()
2015-03-04 21:06:16 +00:00
@cached_property
def type_class(self):
return AppType.get(self.type)
2015-03-04 21:06:16 +00:00
@cached_property
def type_instance(self):
""" Per request lived type_instance """
return self.type_class(self)
2015-03-04 21:06:16 +00:00
2014-11-27 19:17:26 +00:00
def clean(self):
2015-03-04 21:06:16 +00:00
apptype = self.type_instance
apptype.validate()
2015-04-30 11:24:18 +00:00
a = apptype.clean_data()
self.data = apptype.clean_data()
2014-11-27 19:17:26 +00:00
2015-06-05 09:57:06 +00:00
def get_options(self, **kwargs):
2015-06-04 10:23:51 +00:00
options = OrderedDict()
2015-04-20 14:23:10 +00:00
qs = WebAppOption.objects.filter(**kwargs)
for name, value in qs.values_list('name', 'value').order_by('name'):
if name in options:
2015-06-22 22:38:39 +00:00
if AppOption.get(name).comma_separated:
options[name] = options[name].rstrip(',') + ',' + value.lstrip(',')
else:
options[name] = max(options[name], value)
2015-04-20 14:23:10 +00:00
else:
options[name] = value
return options
2014-05-08 16:59:35 +00:00
2014-10-24 10:16:46 +00:00
def get_directive(self):
return self.type_instance.get_directive()
2014-05-08 16:59:35 +00:00
2016-05-26 08:15:34 +00:00
def get_base_path(self):
2014-05-08 16:59:35 +00:00
context = {
'home': self.get_user().get_home(),
2015-03-02 10:37:25 +00:00
'app_name': self.name,
2014-05-08 16:59:35 +00:00
}
2016-05-26 08:15:34 +00:00
return settings.WEBAPPS_BASE_DIR % context
def get_path(self):
path = self.get_base_path()
public_root = self.options.filter(name='public-root').first()
if public_root:
path = os.path.join(path, public_root.value)
2015-03-10 21:51:10 +00:00
return os.path.normpath(path.replace('//', '/'))
2014-10-23 15:38:46 +00:00
2014-10-24 10:16:46 +00:00
def get_user(self):
return self.account.main_systemuser
2014-10-23 15:38:46 +00:00
def get_username(self):
2014-10-24 10:16:46 +00:00
return self.get_user().username
2014-10-23 15:38:46 +00:00
def get_groupname(self):
return self.get_username()
2014-05-08 16:59:35 +00:00
class WebAppOption(models.Model):
webapp = models.ForeignKey(WebApp, verbose_name=_("Web application"),
2015-04-05 10:46:24 +00:00
related_name='options')
name = models.CharField(_("name"), max_length=128,
choices=AppType.get_group_options_choices())
2014-05-08 16:59:35 +00:00
value = models.CharField(_("value"), max_length=256)
class Meta:
unique_together = ('webapp', 'name')
verbose_name = _("option")
verbose_name_plural = _("options")
2015-04-02 16:14:55 +00:00
def __str__(self):
2014-05-08 16:59:35 +00:00
return self.name
@cached_property
def option_class(self):
return AppOption.get(self.name)
@cached_property
def option_instance(self):
""" Per request lived option instance """
return self.option_class(self)
2014-05-08 16:59:35 +00:00
def clean(self):
self.option_instance.validate()