mediacms/users/forms.py
Markos Gogoulos ba94989e6a
Feat urls (#257)
add new URLS, add swaggger doc, add tests
2021-08-05 13:25:25 +03:00

62 lines
1.7 KiB
Python

from django import forms
from files.methods import is_mediacms_manager
from .models import Channel, User
class SignupForm(forms.Form):
name = forms.CharField(max_length=100, label="Name")
def signup(self, request, user):
user.name = self.cleaned_data["name"]
user.save()
class UserForm(forms.ModelForm):
class Meta:
model = User
fields = (
"name",
"description",
"logo",
"notification_on_comments",
"is_featured",
"advancedUser",
"is_manager",
"is_editor",
# "allow_contact",
)
def clean_logo(self):
image = self.cleaned_data.get("logo", False)
if image:
if image.size > 2 * 1024 * 1024:
raise forms.ValidationError("Image file too large ( > 2mb )")
return image
else:
raise forms.ValidationError("Please provide a logo")
def __init__(self, user, *args, **kwargs):
super(UserForm, self).__init__(*args, **kwargs)
self.fields.pop("is_featured")
if not is_mediacms_manager(user):
self.fields.pop("advancedUser")
self.fields.pop("is_manager")
self.fields.pop("is_editor")
class ChannelForm(forms.ModelForm):
class Meta:
model = Channel
fields = ("banner_logo",)
def clean_banner_logo(self):
image = self.cleaned_data.get("banner_logo", False)
if image:
if image.size > 2 * 1024 * 1024:
raise forms.ValidationError("Image file too large ( > 2mb )")
return image
else:
raise forms.ValidationError("Please provide a banner")