From 31d6cbf3a9b91793acbf158846cdfe67af525240 Mon Sep 17 00:00:00 2001 From: caoqianming Date: Fri, 18 Mar 2022 17:07:49 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E7=94=A8=E6=88=B7=E8=A1=A8?= =?UTF-8?q?=E5=81=87=E5=88=A0=E9=99=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../migrations/0004_auto_20220318_1705.py | 36 +++++++++++++++++++ hb_server/apps/system/models.py | 10 ++++-- 2 files changed, 44 insertions(+), 2 deletions(-) create mode 100644 hb_server/apps/system/migrations/0004_auto_20220318_1705.py diff --git a/hb_server/apps/system/migrations/0004_auto_20220318_1705.py b/hb_server/apps/system/migrations/0004_auto_20220318_1705.py new file mode 100644 index 0000000..7979291 --- /dev/null +++ b/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='修改时间'), + ), + ] diff --git a/hb_server/apps/system/models.py b/hb_server/apps/system/models.py index c773fcf..4db171b 100644 --- a/hb_server/apps/system/models.py +++ b/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