django-orchestra/orchestra/contrib/musician/websites/forms.py

121 lines
4.4 KiB
Python
Raw Normal View History

2024-05-18 12:27:03 +00:00
from django import forms
from orchestra.forms.widgets import DynamicHelpTextSelect
from django.core.exceptions import ValidationError
from django.utils.translation import gettext_lazy as _
from django.utils.encoding import force_str
2024-05-18 12:27:03 +00:00
from orchestra.contrib.websites.directives import SiteDirective
2024-05-18 12:27:03 +00:00
from orchestra.contrib.websites.models import Website, Content, WebsiteDirective
from orchestra.contrib.webapps.models import WebApp
from orchestra.contrib.domains.models import Domain
from orchestra.contrib.musician.settings import MUSICIAN_WEBSITES_ENABLE_GROUP_DIRECTIVE
2024-05-18 12:27:03 +00:00
class WebsiteUpdateForm(forms.ModelForm):
class Meta:
model = Website
fields = ("is_active", "protocol", "domains")
help_texts = {
'domains': _('Hold down "Control", or "Command" on a Mac, to select more than one.')
}
def __init__(self, *args, **kwargs):
self.user = kwargs.pop('user')
super().__init__(*args, **kwargs)
# Excluir dominios de otros websites
qs = Website.objects.filter(account=self.user).exclude(id=self.instance.id)
used_domains = []
for website in qs:
dominios = website.domains.all()
for dominio in dominios:
used_domains.append(dominio)
self.fields['domains'].queryset = Domain.objects.filter(account=self.user).exclude(name__in=used_domains)
class WesiteContentCreateForm(forms.ModelForm):
class Meta:
model = Content
fields = ("webapp", "path")
def __init__(self, *args, **kwargs):
self.website = kwargs.pop('website')
self.user = kwargs.pop('user')
super().__init__(*args, **kwargs)
self.fields['webapp'].queryset = WebApp.objects.filter(account=self.user, target_server=self.website.target_server)
def clean(self):
cleaned_data = super().clean()
path = self.cleaned_data.get("path")
path = "/" if path == "" else path
if Content.objects.filter(website=self.website, path=path).exists():
self.add_error('path',_("This Path already exists on this Website."))
return cleaned_data
def save(self, commit=True):
instance = super().save(commit=False)
instance.website = self.website
if commit:
super().save(commit=True)
self.website.save()
return instance
from collections import defaultdict
from orchestra.contrib.websites.utils import normurlpath
class WesiteDirectiveCreateForm(forms.ModelForm):
DIRECTIVES_HELP_TEXT = {
op.name: force_str(op.help_text) for op in SiteDirective.get_plugins()
}
class Meta:
model = WebsiteDirective
fields = ("name", "value")
def __init__(self, *args, **kwargs):
self.website = kwargs.pop('website')
# self.user = kwargs.pop('user')
super().__init__(*args, **kwargs)
target = 'this.id.replace("name", "value")'
self.fields['name'].widget.attrs = DynamicHelpTextSelect(target, self.DIRECTIVES_HELP_TEXT).attrs
self.fields['name'].choices = self.get_allow_choices()
def get_allow_choices(self):
groups = MUSICIAN_WEBSITES_ENABLE_GROUP_DIRECTIVE
yield (None, '-------')
options = SiteDirective.get_option_groups()
for grp in groups:
if grp in options.keys():
yield (grp, [(op.name, op.verbose_name) for op in options[grp]])
2024-05-23 19:46:45 +00:00
def clean(self):
# Recoge todos los paths de directive y contents
locations = set()
for content in Content.objects.filter(website=self.website):
location = content.path
if location is not None:
locations.add(location)
for directive_obj in WebsiteDirective.objects.filter(website=self.website):
location = directive_obj.value
if location is not None:
locations.add(normurlpath(location.split()[0]))
# Comprueva que no se repitan
directive = self.cleaned_data
value = normurlpath(directive.get('value'))
if value:
value = value.split()[0]
if value in locations:
self.add_error('value', f"Location '{value}' already in use by other content/directive.")
2024-05-18 12:27:03 +00:00
def save(self, commit=True):
instance = super().save(commit=False)
instance.website = self.website
if commit:
super().save(commit=True)
self.website.save()
return instance