mirror of https://github.com/snachodog/mybuddy.git
64 lines
1.4 KiB
Python
64 lines
1.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
from django import forms
|
|
from django.contrib.auth.forms import PasswordChangeForm, UserCreationForm
|
|
from django.contrib.auth.models import User
|
|
|
|
from .models import Settings
|
|
|
|
|
|
class UserAddForm(UserCreationForm):
|
|
class Meta:
|
|
model = User
|
|
fields = [
|
|
"username",
|
|
"first_name",
|
|
"last_name",
|
|
"email",
|
|
"is_staff",
|
|
"is_active",
|
|
]
|
|
|
|
def save(self, commit=True):
|
|
user = super(UserAddForm, self).save(commit=False)
|
|
# All Baby Buddy users are superusers.
|
|
user.is_superuser = True
|
|
if commit:
|
|
user.save()
|
|
return user
|
|
|
|
|
|
class UserUpdateForm(forms.ModelForm):
|
|
class Meta:
|
|
model = User
|
|
fields = [
|
|
"username",
|
|
"first_name",
|
|
"last_name",
|
|
"email",
|
|
"is_staff",
|
|
"is_active",
|
|
]
|
|
|
|
|
|
class UserForm(forms.ModelForm):
|
|
class Meta:
|
|
model = User
|
|
fields = ["first_name", "last_name", "email"]
|
|
|
|
|
|
class UserPasswordForm(PasswordChangeForm):
|
|
class Meta:
|
|
fields = ["old_password", "new_password1", "new_password2"]
|
|
|
|
|
|
class UserSettingsForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Settings
|
|
fields = [
|
|
"dashboard_refresh_rate",
|
|
"dashboard_hide_empty",
|
|
"dashboard_hide_age",
|
|
"language",
|
|
"timezone",
|
|
]
|