Removed hardcoded user model references.

Author: simlist

src/emailuser/admin.py

@@ -1,8 +1,8 @@
+from django.contrib.auth import get_user_model
 from django.contrib import admin
 from django.contrib.auth.admin import UserAdmin
 
 from .forms import EmailUserCreationForm, EmailUserChangeForm
-from .models import EmailUser
 
 
 class EmailUserAdmin(UserAdmin):
@@ -31,4 +31,4 @@ class EmailUserAdmin(UserAdmin):
     ordering = ('name',)
 
 
-admin.site.register(EmailUser, EmailUserAdmin)
+admin.site.register(get_user_model(), EmailUserAdmin)

src/emailuser/models.py

@@ -30,4 +30,5 @@ class Meta:
 
 
 class EmailUser(AbstractEmailUser):
-    pass
+    class Meta(AbstractEmailUser.Meta):
+        swappable = 'AUTH_USER_MODEL'

src/emailuser/serializers.py

@@ -1,7 +1,6 @@
+from django.contrib.auth import get_user_model
 from rest_framework import serializers
 
-from .models import EmailUser
-
 
 class EmailUserSerializer(serializers.ModelSerializer):
     password = serializers.CharField(
@@ -11,12 +10,12 @@ class EmailUserSerializer(serializers.ModelSerializer):
         )
 
     class Meta:
-        model = EmailUser
+        model = get_user_model()
         fields = ['id', 'email', 'name', 'password']
         read_only_fields = ['id',]
 
     def create(self, validated_data):
-        return EmailUser.objects.create_user(**validated_data)
+        return get_user_model().objects.create_user(**validated_data)
 
     def update(self, instance, validated_data):
         password = validated_data.pop('password', None)

src/emailuser/views.py

@@ -1,7 +1,7 @@
+from django.contrib.auth import get_user_model
 from rest_framework.permissions import AllowAny, IsAuthenticatedOrReadOnly
 from rest_framework import generics
 from . import serializers
-from .models import EmailUser
 from .permissions import IsSelfOrReadOnly
 
 
@@ -13,4 +13,4 @@ class RegistrationView(generics.CreateAPIView):
 class RetrieveUpdateUserView(generics.RetrieveUpdateAPIView):
     permission_classes = [IsAuthenticatedOrReadOnly, IsSelfOrReadOnly]
     serializer_class = serializers.EmailUserSerializer
-    queryset = EmailUser.objects.all()
+    queryset = get_user_model().objects.all()