Exesize

1
from django.contrib.auth import get_user_model
2
from django.contrib.auth.forms import UserCreationForm, UserChangeForm
3
from django.core.exceptions import ValidationError
4

5

6
class CustomUserCreationForm(UserCreationForm):
7
    class Meta:
8
        model = get_user_model()
9
        fields = (
10
            'username',
11
            'first_name',
12
            'last_name',
13
            'email',
14
            'age',
15
            'avatar'
16
        )
17

18
    def clean_age(self):
19
        age = self.cleaned_data.get('age')
20
        if age < 18:
21
            raise ValidationError('Контент 18+')
22
        else:
23
            return age
24

25

26
class CustomUserChangeForm(UserChangeForm):
27
    class Meta:
28
        model = get_user_model()
29
        fields = (
30
            'username',
31
            'first_name',
32
            'last_name',
33
            'email',
34
            'age',
35
            'avatar'
36
        )
37

38
    def clean_age(self):
39
        age = self.cleaned_data.get('age')
40
        if age < 18:
41
            raise ValidationError('Контент 18+')
42
        else:
43
            return age
44