[英]Authentication by extending User model in DRF (Django Rest Framework)
我已經使用 django 默認 User 類進行了 auth.token 注冊,但我被要求向 User 添加一個新字段,我調查了 AbtractUser、AbstractBaseUser 等。我認為最好的解決方案是 OneToOneField “方法”。
這就是我試圖做的:
模型.py:
class Usuario(models.Model):
user = models.OneToOneField(User, on_delete = models.CASCADE)
es_tecnico = models.BooleanField(name = 'es_tecnico', default = False, blank = True)
序列化程序.py
class UserSerializer(serializers.ModelSerializer):
class Meta:
model = User
fields = ('username',
'password')
class UsuarioSerializer(serializers.ModelSerializer):
user = UserSerializer(required=True)
class Meta:
model = Usuario
fields = ('user',
'es_tecnico')
def create(self, validated_data):
"""
Overriding the default create method of the Model serializer.
:param validated_data: data containing all the details of student
:return: returns a successfully created student record
"""
user_data = validated_data.pop('user')
user = UserSerializer.create(UserSerializer(), validated_data=user_data)
usuario, created = Usuario.objects.update_or_create(user=user,
es_tecnico=validated_data.pop('es_tecnico'))
return usuario
視圖.py:
class UsuarioViewSet(viewsets.ModelViewSet):
lookup_field = 'id'
serializer_class = UsuarioSerializer
queryset = Usuario.objects.all().filter(es_tecnico = False)
class TecnicoViewSet(viewsets.ModelViewSet):
lookup_field = 'id'
serializer_class = UsuarioSerializer
queryset = Usuario.objects.all().filter(es_tecnico = True)
我想讓它盡可能簡單地工作,這就是我使用命令“python3 manage.py makemigrations”時得到的:
Traceback (most recent call last):
File "manage.py", line 15, in <module>
execute_from_command_line(sys.argv)
File "/home/stalker/PycharmProjects/ingSoft/servidor/virenv/lib/python3.5/site-packages/django/core/management/__init__.py", line 371, in execute_from_command_line
utility.execute()
File "/home/stalker/PycharmProjects/ingSoft/servidor/virenv/lib/python3.5/site-packages/django/core/management/__init__.py", line 365, in execute
self.fetch_command(subcommand).run_from_argv(self.argv)
File "/home/stalker/PycharmProjects/ingSoft/servidor/virenv/lib/python3.5/site-packages/django/core/management/base.py", line 288, in run_from_argv
self.execute(*args, **cmd_options)
File "/home/stalker/PycharmProjects/ingSoft/servidor/virenv/lib/python3.5/site-packages/django/core/management/base.py", line 332, in execute
self.check()
File "/home/stalker/PycharmProjects/ingSoft/servidor/virenv/lib/python3.5/site-packages/django/core/management/base.py", line 364, in check
include_deployment_checks=include_deployment_checks,
File "/home/stalker/PycharmProjects/ingSoft/servidor/virenv/lib/python3.5/site-packages/django/core/management/base.py", line 351, in _run_checks
return checks.run_checks(**kwargs)
File "/home/stalker/PycharmProjects/ingSoft/servidor/virenv/lib/python3.5/site-packages/django/core/checks/registry.py", line 73, in run_checks
new_errors = check(app_configs=app_configs)
File "/home/stalker/PycharmProjects/ingSoft/servidor/virenv/lib/python3.5/site-packages/django/contrib/auth/checks.py", line 29, in check_user_model
if not isinstance(cls.REQUIRED_FIELDS, (list, tuple)):
AttributeError: type object 'Usuario' has no attribute 'REQUIRED_FIELDS'
當我嘗試其他 stackoverflow 問題的解決方案時,出現此錯誤:
AttributeError: type object 'Usuario' has no attribute 'USERNAME_FIELD'
如果有使用 OneToOneField 的簡單方法,我會很感激
這是我現在遇到問題的視圖:views.py:
class Registrar(mixins.CreateModelMixin, viewsets.GenericViewSet):
serializer_class = UsuarioSerializer
def create(self, request, *args, **kwargs):
# Creando un nuevo usuario
username = request.POST.get('username')
email = request.POST.get('email')
password = request.POST.get('password')
user = Usuario.objects.create_user(username, email, password)
user.save()
token = Token.objects.create(user=user)
return Response({'detail': 'El usuario fue creado con el token: ' + token.key})
class LoginView(mixins.CreateModelMixin, viewsets.GenericViewSet):
serializer_class = LoginSerializer
def create(self, request):
serializer = LoginSerializer(data=request.data)
serializer.is_valid(raise_exception=True)
user = serializer.validated_data["usuario"]
django_login(request, user)
token, created = Token.objects.get_or_create(user=user)
return Response({"token": token.key}, status=200)
在您的情況下,您只是在擴展 Django User
模型,因此您不應覆蓋AUTH_USER_MODEL
。
從settings.py
文件中刪除這行代碼: AUTH_USER_MODEL = "pedidos.Usuario"
,如果你想在你的代碼中使用你的User
模型,這是你應該使用的:
from django.contrib.auth.models import User
(正如我們所說,您只是在擴展 Django User
模型)或者您可以使用方便的函數get_user_model()
get_user_model():
您應該使用 django.contrib.auth.get_user_model() 來引用用戶模型,而不是直接引用 User。 此方法將返回當前活動的用戶模型 - 如果指定了自定義用戶模型,否則為 User。
如果要獲取User
的屬性es_tecnico
,可以使用:
>>> from django.contrib.auth.models import User
>>> u = User.objects.get(username='Juan')
>>> juan_es_tecnico = u.usuario.es_tecnico
或使用get_user_model()
:
>>> from django.contrib.auth import get_user_model
>>> my_user = get_user_model()
>>> u = my_user.objects.get(username='Juan')
>>> juan_es_tecnico = u.usuario.es_tecnico
編輯:
class Registrar(mixins.CreateModelMixin, viewsets.GenericViewSet):
serializer_class = UsuarioSerializer
def create(self, request, *args, **kwargs):
# Creando un nuevo usuario
username = request.POST.get('username')
email = request.POST.get('email')
password = request.POST.get('password')
user = User.objects.create_user(username, email, password)
user.save()
token = Token.objects.create(user=user)
return Response({'detail': 'El usuario fue creado con el token: ' + token.key})
在Registrar
類中使用User
模型而不是Usuario
。
我希望這將有所幫助。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.