From d1c27e34ae57fb89c69b94947d7b35ba08842458 Mon Sep 17 00:00:00 2001
From: Austin Weisgrau <austinweisgrau@gmail.com>
Date: Wed, 24 Apr 2024 13:29:37 -0400
Subject: [PATCH] Allow passing google auth Credentials directly to BQ
 connector

This allows for using oauth2 to authenticate the connector, rather
than a service account.
---
 parsons/google/google_bigquery.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/parsons/google/google_bigquery.py b/parsons/google/google_bigquery.py
index fecb15ed2f..9b2f6330c5 100644
--- a/parsons/google/google_bigquery.py
+++ b/parsons/google/google_bigquery.py
@@ -11,6 +11,7 @@
 from google.cloud import bigquery, exceptions
 from google.cloud.bigquery import dbapi
 from google.cloud.bigquery.job import LoadJobConfig
+from google.oauth2.credentials import Credentials
 
 from parsons.databases.database_connector import DatabaseConnector
 from parsons.databases.table import BaseTable
@@ -143,7 +144,7 @@ class GoogleBigQuery(DatabaseConnector):
 
     def __init__(
         self,
-        app_creds=None,
+        app_creds: Optional[Union[str, dict, Credentials]] = None,
         project=None,
         location=None,
         client_options: dict = {
@@ -156,7 +157,11 @@ def __init__(
     ):
         self.app_creds = app_creds
 
-        setup_google_application_credentials(app_creds)
+        if isinstance(app_creds, Credentials):
+            self.credentials = app_creds
+        else:
+            self.credentials = None
+            setup_google_application_credentials(app_creds)
 
         self.project = project
         self.location = location
@@ -185,6 +190,7 @@ def client(self):
                 project=self.project,
                 location=self.location,
                 client_options=self.client_options,
+                credentials=self.credentials,
             )
 
         return self._client