# Copyright 2014 Budapest University of Technology and Economics (BME IK) # # This file is part of CIRCLE Cloud. # # CIRCLE is free software: you can redistribute it and/or modify it under # the terms of the GNU General Public License as published by the Free # Software Foundation, either version 3 of the License, or (at your option) # any later version. # # CIRCLE is distributed in the hope that it will be useful, but WITHOUT ANY # WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS # FOR A PARTICULAR PURPOSE. See the GNU General Public License for more # details. # # You should have received a copy of the GNU General Public License along # with CIRCLE. If not, see <http://www.gnu.org/licenses/>. from django.forms import ( ModelForm, ModelChoiceField, ChoiceField, Form, CharField, RadioSelect, Textarea, ValidationError ) from django.utils.translation import ugettext_lazy as _ from django.template import RequestContext from django.template.loader import render_to_string from sizefield.widgets import FileSizeWidget from sizefield.utils import filesizeformat from crispy_forms.helper import FormHelper from crispy_forms.layout import Submit from request.models import ( LeaseType, TemplateAccessType, TemplateAccessAction, ) from dashboard.forms import VmResourcesForm class LeaseTypeForm(ModelForm): @property def helper(self): helper = FormHelper() helper.add_input(Submit("submit", _("Save"), css_class="btn btn-success", )) return helper class Meta: model = LeaseType fields = ["name", "lease", ] class TemplateAccessTypeForm(ModelForm): def __init__(self, *args, **kwargs): super(TemplateAccessTypeForm, self).__init__(*args, **kwargs) @property def helper(self): helper = FormHelper() helper.add_input(Submit("submit", _("Save"), css_class="btn btn-success", )) return helper class Meta: model = TemplateAccessType fields = ["name", "templates", ] class InitialFromFileMixin(object): def __init__(self, *args, **kwargs): request = kwargs.pop("request", None) super(InitialFromFileMixin, self).__init__(*args, **kwargs) self.initial['message'] = render_to_string( self.initial_template, RequestContext(request, {}), ) def clean_message(self): message = self.cleaned_data['message'] if message.strip() == self.initial['message'].strip(): raise ValidationError(_("Fill in the message."), code="invalid") return message.strip() class TemplateRequestForm(InitialFromFileMixin, Form): message = CharField(widget=Textarea, label=_("Message")) template = ModelChoiceField(TemplateAccessType.objects.all(), label=_("Template share")) level = ChoiceField(TemplateAccessAction.LEVELS, widget=RadioSelect, initial=TemplateAccessAction.LEVELS.user) initial_template = "request/initials/template.html" class LeaseRequestForm(InitialFromFileMixin, Form): lease = ModelChoiceField(LeaseType.objects.all(), label=_("Lease")) message = CharField(widget=Textarea, label=_("Message")) initial_template = "request/initials/lease.html" class ResourceRequestForm(InitialFromFileMixin, VmResourcesForm): message = CharField(widget=Textarea, label=_("Message")) initial_template = "request/initials/resources.html" def clean(self): cleaned_data = super(ResourceRequestForm, self).clean() inst = self.instance if (cleaned_data['ram_size'] == inst.ram_size and cleaned_data['num_cores'] == inst.num_cores and int(cleaned_data['priority']) == inst.priority): raise ValidationError( _("You haven't changed any of the resources."), code="invalid") class ResizeRequestForm(InitialFromFileMixin, Form): message = CharField(widget=Textarea, label=_("Message")) size = CharField(widget=FileSizeWidget, label=_('Size'), help_text=_('Size to resize the disk in bytes or with' ' units like MB or GB.')) initial_template = "request/initials/resize.html" def __init__(self, *args, **kwargs): self.disk = kwargs.pop("disk") super(ResizeRequestForm, self).__init__(*args, **kwargs) def clean_size(self): cleaned_data = super(ResizeRequestForm, self).clean() disk = self.disk size_in_bytes = cleaned_data.get("size") if not size_in_bytes.isdigit() and len(size_in_bytes) > 0: raise ValidationError(_("Invalid format, you can use GB or MB!")) if int(size_in_bytes) < int(disk.size): raise ValidationError(_("Disk size must be greater than the actual" "size (%s).") % filesizeformat(disk.size)) return size_in_bytes