Fixed the accounts model to include an UserManager and rebuild migrations

Author: dvidsilva

accounts/migrations/0001_initial.py

@@ -1,5 +1,6 @@
-# Generated by Django 2.1.2 on 2018-10-03 02:11
+# Generated by Django 2.1.2 on 2018-10-06 15:20
 
+import accounts.models
 import django.contrib.auth.password_validation
 from django.db import migrations, models
 
@@ -30,5 +31,8 @@ class Migration(migrations.Migration):
             options={
                 'abstract': False,
             },
+            managers=[
+                ('objects', accounts.models.UserManager()),
+            ],
         ),
     ]

accounts/models.py

@@ -1,11 +1,30 @@
 from django.db import models
-from django.contrib.auth.models import AbstractBaseUser
+from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
 from django.contrib.auth import password_validation
 
+class UserManager(BaseUserManager):
+    use_in_migrations = True
+
+    def create_user(self, email, password=None, **kwargs):
+        user = self.model(email=email, password=password, **kwargs)
+
+        user.save(using=self._db)
+        return user
+
+    def create_superuser(self, email, username, password):
+        user = self.create_user(email, password=password, username=username, is_admin=True)
+        user.save(using=self._db)
+        return user
+
+    def get_by_natural_key(self, username):
+        case_insensitive_username_field = '{}__iexact'.format(self.model.USERNAME_FIELD)
+        return self.get(**{case_insensitive_username_field: username})
 
 class User(AbstractBaseUser):
     USERNAME_FIELD = 'username'
 
+    REQUIRED_FIELDS = ['email']
+
     created_at = models.DateTimeField(auto_now_add=True)
     updated_at = models.DateTimeField(auto_now=True)
     password = models.CharField(
@@ -24,5 +43,11 @@ class User(AbstractBaseUser):
     is_admin = models.BooleanField(default=False)
     flagged = models.BooleanField(default=False)
 
+    objects = UserManager()
+
     def __str__(self):
         return self.username
+
+    @property
+    def is_staff(self):
+        return self.is_admin