Add Service Account

AdeptMind · Dec 22, 2023 · c79ef4e · c79ef4e
1 parent a843c4d
commit c79ef4e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/api/procurement_api.py b/api/procurement_api.py
@@ -2,6 +2,7 @@
 
 import requests
 from backoff import on_exception, expo
+from google.oauth2 import service_account
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
 from ratelimit import limits, RateLimitException
@@ -21,7 +22,11 @@ class ProcurementApi(object):
     """Utilities for interacting with the Procurement API."""
 
     def __init__(self, project_id):
-        self.service = build(PROCUREMENT_API, "v1", cache_discovery=False)
+        if settings.use_service_account:
+            credentials = service_account.Credentials.from_service_account_file('service-account.json', scopes=[])
+            self.service = build(PROCUREMENT_API, "v1", cache_discovery=False, credentials=credentials)
+        else:
+            self.service = build(PROCUREMENT_API, "v1", cache_discovery=False)
         self.project_id = project_id
 
     ##########################

diff --git a/api/services_api.py b/api/services_api.py
@@ -17,8 +17,11 @@ class ServicesApi(object):
     """Utilities for interacting with the Services API."""
 
     def __init__(self, project_id):
-        credentials = service_account.Credentials.from_service_account_file('service-account.json', scopes=[])
-        self.service = build(SERVICES_API, "v1", cache_discovery=False, credentials=credentials)
+        if settings.use_service_account:
+            credentials = service_account.Credentials.from_service_account_file('service-account.json', scopes=[])
+            self.service = build(SERVICES_API, "v1", cache_discovery=False, credentials=credentials)
+        else:
+            self.service = build(SERVICES_API, "v1", cache_discovery=False)
         self.project_id = project_id
 
     @on_exception(expo, RateLimitException, max_tries=8)