mirror of
https://github.com/prowler-cloud/prowler.git
synced 2026-10-21 02:44:16 +00:00
32 lines
862 B
Python
32 lines
862 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):
|
|
aws_account_id = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = CloudAccount
|
|
fields = "__all__"
|
|
extra_fields = ["aws_account_id"]
|
|
|
|
def get_aws_account_id(self, obj):
|
|
return obj.account_id.aws_account_id
|
|
|
|
|
|
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
|