mirror of
https://github.com/eitchtee/WYGIWYH.git
synced 2026-01-19 15:26:58 +01:00
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
from django.db.models import Q
|
|
from rest_framework import serializers
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
|
from apps.api.serializers.currencies import CurrencySerializer
|
|
from apps.accounts.models import AccountGroup, Account
|
|
from apps.currencies.models import Currency
|
|
|
|
|
|
class AccountGroupSerializer(serializers.ModelSerializer):
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
class Meta:
|
|
model = AccountGroup
|
|
fields = "__all__"
|
|
|
|
|
|
class AccountSerializer(serializers.ModelSerializer):
|
|
group = AccountGroupSerializer(read_only=True)
|
|
group_id = serializers.PrimaryKeyRelatedField(
|
|
queryset=AccountGroup.objects.all(),
|
|
source="group",
|
|
write_only=True,
|
|
allow_null=True,
|
|
)
|
|
|
|
currency = CurrencySerializer(read_only=True)
|
|
currency_id = serializers.PrimaryKeyRelatedField(
|
|
queryset=Currency.objects.all(), source="currency", write_only=True
|
|
)
|
|
exchange_currency = CurrencySerializer(read_only=True)
|
|
exchange_currency_id = serializers.PrimaryKeyRelatedField(
|
|
queryset=Currency.objects.all(),
|
|
source="exchange_currency",
|
|
write_only=True,
|
|
allow_null=True,
|
|
)
|
|
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
class Meta:
|
|
model = Account
|
|
fields = [
|
|
"id",
|
|
"name",
|
|
"group",
|
|
"group_id",
|
|
"currency",
|
|
"currency_id",
|
|
"exchange_currency",
|
|
"exchange_currency_id",
|
|
"is_asset",
|
|
]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
request = self.context.get("request")
|
|
if request and request.user.is_authenticated:
|
|
# Reload the queryset to get an updated version with the requesting user
|
|
self.fields["group_id"].queryset = AccountGroup.objects.all()
|
|
|
|
def create(self, validated_data):
|
|
return Account.objects.create(**validated_data)
|
|
|
|
def update(self, instance, validated_data):
|
|
for attr, value in validated_data.items():
|
|
setattr(instance, attr, value)
|
|
instance.save()
|
|
return instance
|
|
|
|
|
|
class AccountBalanceSerializer(serializers.Serializer):
|
|
"""Serializer for account balance response."""
|
|
|
|
current_balance = serializers.DecimalField(max_digits=20, decimal_places=10)
|
|
projected_balance = serializers.DecimalField(max_digits=20, decimal_places=10)
|
|
currency = CurrencySerializer()
|
|
|