# coding: utf-8 import inspect from django.db import models from django.http import JsonResponse from django.utils.timezone import now from django.utils.translation import ugettext_lazy as _ class NoDeleteQuerySet(models.query.QuerySet): def delete(self): return self.update(is_discard=True, discard_time=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 = now() return self.save() class JSONResponseMixin(object): """JSON mixin""" @staticmethod def render_json_response(context): return JsonResponse(context) class IDInFilterMixin(object): def filter_queryset(self, queryset): id_list = self.request.query_params.get('id__in') if id_list: import json try: ids = json.loads(id_list) except Exception as e: return queryset if isinstance(ids, list): queryset = queryset.filter(id__in=ids) return queryset class BulkSerializerMixin(object): """ Become rest_framework_bulk not support uuid as a primary key so rewrite it. https://github.com/miki725/django-rest-framework-bulk/issues/66 """ def to_internal_value(self, data): from rest_framework_bulk import BulkListSerializer ret = super(BulkSerializerMixin, self).to_internal_value(data) id_attr = getattr(self.Meta, 'update_lookup_field', 'id') request_method = getattr(getattr(self.context.get('view'), 'request'), 'method', '') # add update_lookup_field field back to validated data # since super by default strips out read-only fields # hence id will no longer be present in validated_data if all((isinstance(self.root, BulkListSerializer), id_attr, request_method in ('PUT', 'PATCH'))): id_field = self.fields[id_attr] if data.get("id"): id_value = id_field.to_internal_value(data.get("id")) else: id_value = id_field.to_internal_value(data.get("pk")) ret[id_attr] = id_value return ret