add banks to connection viewset. Change plaid auth to return connection serializer

connection/serializers.py

@@ -1,5 +1,6 @@
 from rest_framework import serializers
 from .models import ConnectionType, Connection
+from qrtr_account.models import BankAccount
 
 
 class ConnectionTypeSerializer(serializers.HyperlinkedModelSerializer):
@@ -11,12 +12,24 @@ class ConnectionTypeSerializer(serializers.HyperlinkedModelSerializer):
             'filename': {'read_only': True}
         }
 
-
-class ConnectionSerializer(serializers.HyperlinkedModelSerializer):
+class ConnectionAccountSerializer(serializers.HyperlinkedModelSerializer):
     class Meta:
         model = Connection
-        fields = ['url', 'name', 'type', 'credentials']
+        fields = ['pk', 'url', 'name', 'type']
+
+class BankAccountReadSerializer(serializers.HyperlinkedModelSerializer):
+    connection = ConnectionAccountSerializer(read_only=True)
+    class Meta:
+        model = BankAccount
+        fields = ['pk', 'url', 'connection']
+
+class ConnectionSerializer(serializers.HyperlinkedModelSerializer):
+    bank_accounts = BankAccountReadSerializer(read_only=True, source='*')
+    class Meta:
+        model = Connection
+        fields = ['url', 'name', 'type', 'credentials', 'bank_accounts']
         extra_kwargs = {
             'type': {'write_only': True},
             'credentials': {'write_only': True}
         }
+

connection/views.py

@@ -78,7 +78,7 @@ class ConnectionViewSet(viewsets.ModelViewSet, OwnedAccountsMixin):
             conn.credentials = plaid_client.credentials
             print(f"CREDS: {plaid_client.credentials}")
             conn.save()
-        return Response(plaid_client.get_accounts())
+        return Response(self.get_serializer(conn).data)
 
     @action(detail=False, methods=['post'], url_path='plaid')
     def authenticate(self, request):