修改用户表假删除

hb_server/apps/system/migrations/0004_auto_20220318_1705.py

@@ -0,0 +1,36 @@
+# Generated by Django 3.2.9 on 2022-03-18 09:05
+
+import apps.system.models
+from django.db import migrations, models
+import django.utils.timezone
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('system', '0003_auto_20210812_0909'),
+    ]
+
+    operations = [
+        migrations.AlterModelManagers(
+            name='user',
+            managers=[
+                ('objects', apps.system.models.SoftDeletableUserManager()),
+            ],
+        ),
+        migrations.AddField(
+            model_name='user',
+            name='create_time',
+            field=models.DateTimeField(default=django.utils.timezone.now, help_text='创建时间', verbose_name='创建时间'),
+        ),
+        migrations.AddField(
+            model_name='user',
+            name='is_deleted',
+            field=models.BooleanField(default=False, help_text='删除标记', verbose_name='删除标记'),
+        ),
+        migrations.AddField(
+            model_name='user',
+            name='update_time',
+            field=models.DateTimeField(auto_now=True, help_text='修改时间', verbose_name='修改时间'),
+        ),
+    ]

hb_server/apps/system/models.py

@@ -4,7 +4,7 @@ from django.db.models.base import Model
 import django.utils.timezone as timezone
 from django.db.models.query import QuerySet
 
-from utils.model import SoftModel, BaseModel
+from utils.model import SoftModel, BaseModel, SoftDeletableManagerMixin
 from simple_history.models import HistoricalRecords
 
 
@@ -101,7 +101,12 @@ class Role(SoftModel):
         return self.name
 
 
-class User(AbstractUser):
+from django.contrib.auth.models import UserManager
+
+class SoftDeletableUserManager(SoftDeletableManagerMixin, UserManager):
+    pass
+
+class User(SoftModel, AbstractUser):
     """
     用户
     """
@@ -117,6 +122,7 @@ class User(AbstractUser):
         'self', null=True, blank=True, on_delete=models.SET_NULL, verbose_name='上级主管')
     roles = models.ManyToManyField(Role, blank=True, verbose_name='角色')
 
+    objects = SoftDeletableUserManager()
     class Meta:
         verbose_name = '用户信息'
         verbose_name_plural = verbose_name