mirror of
https://github.com/himool/HimoolERP.git
synced 2024-11-15 11:58:42 +08:00
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
|
from rest_framework.fields import IntegerField, FloatField, DecimalField, BooleanField, NullBooleanField
|
||
|
|
from rest_framework.fields import SerializerMethodField, ImageField, FileField, JSONField
|
||
|
|
from rest_framework.fields import CharField, DateField, DateTimeField
|
||
|
|
from rest_framework.serializers import Serializer, ModelSerializer
|
||
|
|
from django.db.models import Sum, Count, Value, F, Q
|
||
|
|
from extensions.exceptions import ValidationError
|
||
|
|
from django.db.models.functions import Coalesce
|
||
|
|
from django.db import transaction
|
||
|
|
from number_precision import NP
|
||
|
|
import pendulum
|
||
|
|
|
||
|
|
|
||
|
|
class BaseSerializer(ModelSerializer):
|
||
|
|
|
||
|
|
@property
|
||
|
|
def request(self):
|
||
|
|
return self.context['request']
|
||
|
|
|
||
|
|
@property
|
||
|
|
def team(self):
|
||
|
|
return self.context['request'].user.team
|
||
|
|
|
||
|
|
@property
|
||
|
|
def user(self):
|
||
|
|
return self.context['request'].user
|
||
|
|
|
||
|
|
def validate_foreign_key(self, model, instance, message):
|
||
|
|
if instance:
|
||
|
|
if not (instance := model.objects.filter(id=instance.id, team=self.team).first()):
|
||
|
|
raise ValidationError(message)
|
||
|
|
return instance
|
||
|
|
|
||
|
|
def validate_foreign_key_set(self, model, instances, message):
|
||
|
|
if instances:
|
||
|
|
instance_ids = [instance.id for instance in instances]
|
||
|
|
instances = model.objects.filter(id__in=instance_ids, team=self.team)
|
||
|
|
|
||
|
|
if len(instance_ids) != len(instances):
|
||
|
|
raise ValidationError(message)
|
||
|
|
return instances
|
||
|
|
|
||
|
|
def validate_unique(self, fields, message):
|
||
|
|
queryset = self.Meta.model.objects.filter(team=self.team, **fields)
|
||
|
|
if self.instance:
|
||
|
|
queryset = queryset.exclude(id=self.instance.id)
|
||
|
|
|
||
|
|
if queryset.exists():
|
||
|
|
raise ValidationError(message)
|
||
|
|
|
||
|
|
def create(self, validated_data):
|
||
|
|
validated_data['team'] = self.team
|
||
|
|
return super().create(validated_data)
|
||
|
|
|
||
|
|
|
||
|
|
class AmountField(DecimalField):
|
||
|
|
"""金额字段"""
|
||
|
|
|
||
|
|
def __init__(self, coerce_to_string=None, max_value=None, min_value=None, localize=False,
|
||
|
|
rounding=None, **kwargs):
|
||
|
|
kwargs['max_digits'], kwargs['decimal_places'] = 16, 2
|
||
|
|
super().__init__(coerce_to_string=None, max_value=None, min_value=None, localize=False,
|
||
|
|
rounding=None, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
'Serializer', 'ModelSerializer', 'BaseSerializer',
|
||
|
|
'SerializerMethodField', 'ImageField', 'FileField', 'JSONField',
|
||
|
|
'BooleanField', 'NullBooleanField', 'IntegerField', 'FloatField', 'DecimalField', 'AmountField',
|
||
|
|
'CharField', 'DateField', 'DateTimeField',
|
||
|
|
'transaction', 'pendulum', 'NP',
|
||
|
|
'Sum', 'Count', 'Value', 'F', 'Q', 'Coalesce',
|
||
|
|
]
|