mirror of
https://github.com/jwinterm/modoboa.git
synced 2026-10-20 06:08:35 +02:00
Antoine Nguyen
2a0e9fdf3a
* New events * Updated events API * New models to implement limits (more flexible)
72 lines
2.8 KiB
Python
72 lines
2.8 KiB
Python
# coding: utf-8
|
|
|
|
from django import forms
|
|
from django.utils.translation import ugettext_noop as _
|
|
from django.contrib.auth.models import User, Group
|
|
from models import *
|
|
|
|
class ResellerForm(forms.ModelForm):
|
|
class Meta:
|
|
model = User
|
|
fields = ("username", "first_name", "last_name", "email")
|
|
|
|
def save(self, commit=True):
|
|
user = super(ResellerForm, self).save(commit=False)
|
|
if self.cleaned_data.has_key("password1"):
|
|
user.set_password(self.cleaned_data["password1"])
|
|
if commit:
|
|
user.save()
|
|
grp = Group.objects.get(name="Resellers")
|
|
if not grp in user.groups.all():
|
|
user.groups.add(grp)
|
|
user.save()
|
|
try:
|
|
pool = user.limitspool
|
|
except LimitsPool.DoesNotExist:
|
|
pool = LimitsPool()
|
|
pool.user = user
|
|
pool.save()
|
|
|
|
for lname in reseller_limits_tpl:
|
|
l = Limit()
|
|
l.name = lname
|
|
l.pool = pool
|
|
l.save()
|
|
|
|
return user
|
|
|
|
class ResellerWithPasswordForm(ResellerForm):
|
|
password1 = forms.CharField(label=_("Password"), widget=forms.PasswordInput)
|
|
password2 = forms.CharField(label=_("Confirmation"),
|
|
widget=forms.PasswordInput,
|
|
help_text = _("Enter the same password as above, for verification."))
|
|
|
|
def clean_password2(self):
|
|
password1 = self.cleaned_data.get("password1", "")
|
|
password2 = self.cleaned_data["password2"]
|
|
if password1 != password2:
|
|
raise forms.ValidationError(_("The two password fields didn't match."))
|
|
return password2
|
|
|
|
class ResellerPoolForm(forms.Form):
|
|
domains_limit = forms.IntegerField(ugettext_noop("Max domains"),
|
|
help_text=ugettext_noop("Maximum number of domains that can be created by this user"))
|
|
domain_aliases_limit = forms.IntegerField(ugettext_noop("Max domain aliases"),
|
|
help_text=ugettext_noop("Maximum number of domain aliases that can be created by this user"))
|
|
mailboxes_limit = forms.IntegerField(ugettext_noop("Max mailboxes"),
|
|
help_text=ugettext_noop("Maximum number of mailboxes that can be created by this user"))
|
|
mailbox_aliases_limit = forms.IntegerField(ugettext_noop("Max mailbox aliases"),
|
|
help_text=ugettext_noop("Maximum number of mailbox aliases that can be created by this user"))
|
|
|
|
def load_from_user(self, user):
|
|
for l in reseller_limits_tpl:
|
|
self.fields[l].initial = user.limitspool.getmaxvalue(l)
|
|
|
|
def save_new_limits(self, pool):
|
|
for lname in reseller_limits_tpl:
|
|
l = pool.limit_set.get(name=lname)
|
|
l.maxvalue = self.cleaned_data[lname]
|
|
l.save()
|
|
|
|
|