如何在 create_superuser 中重用 create_user 代码？

Question

我有一个自定义的 BaseUserManager 和 AbstractBaseUser。我想为我的superuser 代码重用AbstractBaseUser 中的create_user 函数，因为要求是相同的。

据我了解，我可以将传递给create_superuser 的参数直接传递给create_user，然后将其本地保存在函数中，将is_admin 更改为True，然后返回它。这将回收大量代码并保持清洁。这是一个合适的方法吗？

def create_superuser(self, email, var2, var3, var4, etc...):
    user = create_user(email, var2, var3, var4, etc...)
    user.is_admin = True
    user.save(using=self._db)
    return user

我主要想知道在同一用户上调用user.save(using=self._db) 两次，一次在create_user 和一次在create_superuser 上是否安全？

Answer 1

看看Django did it如何：

class MyCustomUserManager(models.Manager):
    def _create_user(self, email, is_staff, is_superuser, ...):
        ...

    def create_user(self, email, ...):
        return self._create_user(email, False, False, ...)

    def create_superuser(self, email, ...):
        return self._create_user(email, True, True, ...)

这将最大限度地减少额外代码和数据库访问。虽然两次保存用户并没有什么特别的问题，但完全没有必要。