mirror of
https://github.com/jwinterm/modoboa.git
synced 2026-10-20 06:08:35 +02:00
103 lines
3.7 KiB
Python
103 lines
3.7 KiB
Python
# coding: utf-8
|
|
|
|
from django import forms
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from modoboa.lib import parameters
|
|
from models import *
|
|
from lib import *
|
|
|
|
|
|
class ResourcePoolForm(forms.Form):
|
|
domain_admins_limit = forms.IntegerField(
|
|
label=ugettext_lazy("Domain admins"),
|
|
help_text=ugettext_lazy("Maximum number of domain administrators this user can create")
|
|
)
|
|
domains_limit = forms.IntegerField(
|
|
label=ugettext_lazy("Domains"),
|
|
help_text=ugettext_lazy("Maximum number of domains this user can create")
|
|
)
|
|
domain_aliases_limit = forms.IntegerField(
|
|
label=ugettext_lazy("Domain aliases"),
|
|
help_text=ugettext_lazy("Maximum number of domain aliases this user can create")
|
|
)
|
|
mailboxes_limit = forms.IntegerField(
|
|
label=ugettext_lazy("Mailboxes"),
|
|
help_text=ugettext_lazy("Maximum number of mailboxes this user can create")
|
|
)
|
|
mailbox_aliases_limit = forms.IntegerField(
|
|
label=ugettext_lazy("Mailbox aliases"),
|
|
help_text=ugettext_lazy("Maximum number of mailbox aliases this user can create")
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
if "instance" in kwargs:
|
|
self.account = kwargs["instance"]
|
|
del kwargs["instance"]
|
|
super(ResourcePoolForm, self).__init__(*args, **kwargs)
|
|
if hasattr(self, "account"):
|
|
if self.account.group != "Resellers":
|
|
del self.fields["domain_admins_limit"]
|
|
del self.fields["domains_limit"]
|
|
del self.fields["domain_aliases_limit"]
|
|
self.load_from_user(self.account)
|
|
|
|
def check_limit_value(self, lname):
|
|
if self.cleaned_data[lname] < -1:
|
|
raise forms.ValidationError(_("Invalid limit"))
|
|
return self.cleaned_data[lname]
|
|
|
|
def clean_domains_limit(self):
|
|
return self.check_limit_value("domains_limit")
|
|
|
|
def clean_domain_aliases_limit(self):
|
|
return self.check_limit_value("domain_aliases_limit")
|
|
|
|
def clean_mailboxes_limit(self):
|
|
return self.check_limit_value("mailboxes_limit")
|
|
|
|
def clean_mailbox_aliases_limit(self):
|
|
return self.check_limit_value("mailbox_aliases_limit")
|
|
|
|
def load_from_user(self, user):
|
|
for l in limits_tpl:
|
|
if not l[0] in self.fields:
|
|
continue
|
|
self.fields[l[0]].initial = user.limitspool.getmaxvalue(l[0])
|
|
# The following lines will become useless in a near
|
|
# future.
|
|
if self.fields[l[0]].initial == -2:
|
|
self.fields[l[0]].initial = parameters.get_admin("DEFLT_%s" % l[0].upper())
|
|
|
|
def allocate_from_pool(self, limit, pool):
|
|
ol = pool.get_limit(limit.name)
|
|
if ol.maxvalue == -2:
|
|
raise BadLimitValue(_("Your resources are not initialized yet"))
|
|
newvalue = self.cleaned_data[limit.name]
|
|
if newvalue == -1 and ol.maxvalue != -1:
|
|
raise BadLimitValue(
|
|
_("You're not allowed to define unlimited values")
|
|
)
|
|
|
|
if limit.maxvalue > -1:
|
|
newvalue -= limit.maxvalue
|
|
if newvalue == 0:
|
|
return
|
|
remain = ol.maxvalue - ol.curvalue
|
|
if newvalue > remain:
|
|
raise UnsufficientResource(ol)
|
|
ol.maxvalue -= newvalue
|
|
ol.save()
|
|
|
|
def save(self):
|
|
from modoboa.lib.permissions import get_object_owner
|
|
|
|
owner = get_object_owner(self.account)
|
|
for ltpl in limits_tpl:
|
|
if not ltpl[0] in self.cleaned_data:
|
|
continue
|
|
l = self.account.limitspool.limit_set.get(name=ltpl[0])
|
|
if not owner.is_superuser:
|
|
self.allocate_from_pool(l, owner.limitspool)
|
|
l.maxvalue = self.cleaned_data[ltpl[0]]
|
|
l.save()
|