Qrtr/qrtr-services
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated plaid connection view to handle various plaid errors. Updated Plaid connection to only allow connecting to account ids which the user has access.

Browse Source
This commit is contained in:
DJ Gillespie 2020-09-16 15:08:57 -06:00
parent b5b9604e63
commit 1e80dfd7fd
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
connection/views.py
View File

@ -42,25 +42,36 @@ class ConnectionViewSet(viewsets.ModelViewSet):
return Response(
status=status.HTTP_400_BAD_REQUEST,
data="ERROR: missing account_id")
accounts = Account.objects.filter(pk=account_id)
user = request.user
# Filter out any accounts with the right id, but the given user
# is not an owner or admin on that account.
accounts = (Account.objects.filter(pk=account_id, owner=user) |
Account.objects.filter(pk=account_id,
admin_users__in=[user]))
if not accounts:
return Response(
status=status.HTTP_400_BAD_REQUEST,
data="ERROR: invalid account_id")
data="ERROR: Account ID not found")
else:
print(f"Account Found: {accounts[0]}")
account = accounts[0]
print(request)
plaid = importlib.import_module(f"connection.connections.plaid_client")
conn_type = ConnectionType.objects.get(name="Plaid")
try:
plaid_client = plaid.Connection(request.data)
except ValueError:
return Response(status=status.HTTP_503,
data="ERROR: Invalid public_token")
except Exception:
return Response(status=status.HTTP_500,
data="ERROR: Unable to contact Plaid")
conn, created = Connection.objects \
.get_or_create(name=name, type=conn_type,
defaults={
"credentials": request.data,
"account": account
})
plaid_client = plaid.Connection(request.data)
conn.credentials = plaid_client.credentials
conn.save()
return Response(plaid_client.get_accounts())