mirror of https://github.com/snachodog/mybuddy.git
51 lines
1.3 KiB
Python
51 lines
1.3 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)
|
|
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'
|
|
]
|