Files
prowler/be_poc/api/serializers.py
T

26 lines
677 B
Python

from rest_framework import serializers
from .models import Account, CloudAccount, Audit
class AccountSerializer(serializers.ModelSerializer):
class Meta:
model = Account
fields = "__all__"
class CloudAccountSerializer(serializers.ModelSerializer):
class Meta:
model = CloudAccount
fields = "__all__"
class AuditSerializer(serializers.ModelSerializer):
account_id = serializers.SerializerMethodField()
class Meta:
model = Audit
fields = ['id', 'audit_complete', 'inserted_at', 'updated_at', 'audit_duration', 'account_id']
def get_account_id(self, obj):
return obj.aws_account_id.account_id.id