• BaiJiangJie's avatar
    Asset favor (#3352) · f3dc9b88
    BaiJiangJie authored
    * [Update] 拆分filter org
    
    * [Update] 修改session支持protocol搜索
    
    * [Bugfix] 修复判断问题
    
    * [Update] 支持收藏资产
    
    * [update] 修改org resource queryset
    
    * [Update] 修改form serializer 对应的多对多字段
    
    * [Bugfix] 修复其他组织取消收藏的bug
    
    * [Update] 去掉debug信息
    
    * [Update] 修改remote app get queryset
    
    * [Update] 修改remote app get queryset
    
    * [Update] 修改没有授权时显示情况
    
    * [Bugfix] 修复组织管理员查看用户权限失败问题
    
    * [Update] 优化forms assets queryset设置
    f3dc9b88
models.py 1.58 KB
# -*- coding: utf-8 -*-
#
import uuid
from django.db import models
from django.utils import timezone
from django.utils.translation import ugettext_lazy as _


__all__ = [
    "NoDeleteManager", "NoDeleteModelMixin", "NoDeleteQuerySet",
    "CommonModelMixin"
]


class NoDeleteQuerySet(models.query.QuerySet):

    def delete(self):
        return self.update(is_discard=True, discard_time=timezone.now())


class NoDeleteManager(models.Manager):

    def get_all(self):
        return NoDeleteQuerySet(self.model, using=self._db)

    def get_queryset(self):
        return NoDeleteQuerySet(self.model, using=self._db).filter(is_discard=False)

    def get_deleted(self):
        return NoDeleteQuerySet(self.model, using=self._db).filter(is_discard=True)


class NoDeleteModelMixin(models.Model):
    is_discard = models.BooleanField(verbose_name=_("is discard"), default=False)
    discard_time = models.DateTimeField(verbose_name=_("discard time"), null=True, blank=True)

    objects = NoDeleteManager()

    class Meta:
        abstract = True

    def delete(self):
        self.is_discard = True
        self.discard_time = timezone.now()
        return self.save()


class CommonModelMixin(models.Model):
    id = models.UUIDField(default=uuid.uuid4, primary_key=True)
    created_by = models.CharField(max_length=32, null=True, blank=True, verbose_name=_('Created by'))
    date_created = models.DateTimeField(auto_now_add=True, null=True, blank=True, verbose_name=_('Date created'))
    date_updated = models.DateTimeField(auto_now=True, verbose_name=_('Date updated'))

    class Meta:
        abstract = True