Search code examples
pythondjangodjango-modelsdjango-usersdjango-custom-user

How to change the name of db column last_login created by custom Django user?


I've created a custom User in Django and added some extra fields to UserProfile model. I want to keep all column names without underscore but last_login filed is getting created by base User. After running migrations, the only column which appears with an underscore in User table is "last_login". Please advise how do I change that field/column name to "lastlogin".

models.py looks like this -

class CustomUserManager(BaseUserManager):

    def _create_user(self, username, email, password,
                     is_staff, is_superuser, **extra_fields):
        """
        Creates and saves a User with the given username, email and password.
        """
        if not username:
            raise ValueError('The given username must be set')
        email = self.normalize_email(email)
        user = self.model(username=username, email=email,
                          is_staff=is_staff, is_active=True,
                          is_superuser=is_superuser, **extra_fields)
        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_user(self, username, email=None, password=None,
                    **extra_fields):
        return self._create_user(username, email, password, False, False,
                                 **extra_fields)

    def create_superuser(self, username, email, password, **extra_fields):
        return self._create_user(username, email, password, True, True,
                                 **extra_fields)


class UserProfile(AbstractBaseUser):

    class Meta:
        """
        meta class for user role, defining ordering and db table name
        """
        ordering = ["name"]
        db_table = 'user'

    username = models.CharField(max_length=254, unique=True)
    name = models.CharField(max_length=254, blank=True)
    email = models.EmailField(blank=True, unique=True)
    createdon = models.DateTimeField(auto_now_add=True)
    updatedon = models.DateTimeField(auto_now=True)
    is_active = models.BooleanField(default=True, db_column='status')
    is_admin = models.BooleanField(default=False, db_column='isadmin')
    is_staff = models.BooleanField(default=False, db_column='isstaff')
    is_superuser = models.BooleanField(default=False, db_column='issuperuser')

    USERNAME_FIELD = 'username'
    REQUIRED_FIELDS = ['email']

    objects = CustomUserManager()

    def get_full_name(self):
        return self.username

    def get_short_name(self):
        return self.username

Solution

  • I found a solution for this on this link and it works!

    def myoverridenmeta(name, bases, adict):
        newClass = type(name, bases, adict)
        for field in newClass._meta.fields:
            if field.attname == 'last_login':
                field.column = 'lastlogin'
                field.db_column = 'lastlogin'
        return newClass
    
    class UserProfile(AbstractBaseUser):
    
        class Meta:
            """
            meta class for user role, defining ordering and db table name
            """
            ordering = ["name"]
            db_table = 'user'
    
        username = models.CharField(max_length=254, unique=True)
        name = models.CharField(max_length=254, blank=True)
        email = models.EmailField(blank=True, unique=True)
        createdon = models.DateTimeField(auto_now_add=True)
        updatedon = models.DateTimeField(auto_now=True)
        is_active = models.BooleanField(default=True, db_column='status')
        is_admin = models.BooleanField(default=False, db_column='isadmin')
        is_staff = models.BooleanField(default=False, db_column='isstaff')
        is_superuser = models.BooleanField(default=False, db_column='issuperuser')
    
        USERNAME_FIELD = 'username'
        REQUIRED_FIELDS = ['email']
    
        objects = CustomUserManager()
    
        def get_full_name(self):
            return self.username
    
        def get_short_name(self):
            return self.username
    
        __metaclass__ = myoverridenmeta