Usar service account para acceder a la planilla (#7)

Author: tutecano1995

.gitignore

@@ -1,2 +1,3 @@
 *.swp
 *.pyc
+/service_account.json

Pipfile

@@ -4,7 +4,9 @@ verify_ssl = true
 name = "pypi"
 
 [packages]
-gspread = "==0.2.*"
+requests = "==2.26.*"
+google-auth = "==1.34.*"
+gspread = "==4.0.*"
 httplib2 = "==0.17.*"
 oauth2client = "~=4.1"
 webargs = "~=5.0"

Pipfile.lock

@@ -21,7 +21,7 @@ def get_sheet(worksheet_name):
 
     Utiliza la constante global SPREADSHEET_KEY.
     """
-    client = gspread.authorize(notas_oauth.get_credenciales())
+    client = gspread.authorize(notas_oauth.get_credenciales_spreadsheet())
     spreadsheet = client.open_by_key(SPREADSHEET_KEY)
     return spreadsheet.worksheet(worksheet_name)
 

notas.py

@@ -5,30 +5,50 @@
 import datetime
 import httplib2
 import os
-import sys
 
 import oauth2client.client
+import google.oauth2.service_account
+import google.auth.transport.requests
 
 _CLIENT_ID = os.environ["NOTAS_OAUTH_CLIENT"]
 _CLIENT_SECRET = os.environ["NOTAS_OAUTH_SECRET"]
 _OAUTH_REFRESH = os.environ["NOTAS_REFRESH_TOKEN"]
+_SERVICE_ACCOUNT_JSON = os.environ["NOTAS_SERVICE_ACCOUNT_JSON"]
 
-_creds = oauth2client.client.OAuth2Credentials(
+SCOPES = ["https://www.googleapis.com/auth/spreadsheets", "https://www.googleapis.com/auth/gmail.send"]
+
+# TODO: Unificar autenticacion para planilla y cuenta de mail.
+# Por ahora no encontramos la forma de enviar mails usando la service account.
+# Mantenemos por un lado el service account para acceder a la planilla y el client id/secret para enviar mails.
+_creds_spreadhseet = google.oauth2.service_account.Credentials.from_service_account_file(_SERVICE_ACCOUNT_JSON, scopes=SCOPES)
+_creds_email = oauth2client.client.OAuth2Credentials(
     "", _CLIENT_ID, _CLIENT_SECRET, _OAUTH_REFRESH,
     datetime.datetime(2015, 1, 1),
     "https://accounts.google.com/o/oauth2/token", "notasweb/1.0")
 
 
-def get_credenciales():
-    """Devuelve nuestro objeto OAuth2Credentials, actualizado.
+def get_credenciales(creds, es_valida, refrescar):
+    if not es_valida(creds):
+        refrescar(creds)
+    return creds
+
 
+def get_credenciales_spreadsheet():
+    """Devuelve nuestro objeto OAuth2Credentials para acceder a la planilla, actualizado.
     Esta función llama a _refresh() si el token expira en menos de 5 minutos.
     """
-    now = datetime.datetime.utcnow()
-    valid_until = _creds.token_expiry - datetime.timedelta(minutes=5)
-
-    if valid_until < now:
-        print("Generando nuevo token de acceso.", file=sys.stderr)
-        _creds.refresh(httplib2.Http())
-
-    return _creds
+    return get_credenciales(
+        _creds_spreadhseet,
+        lambda creds: creds.valid,
+        lambda creds: creds.refresh(google.auth.transport.requests.Request())
+    )
+
+def get_credenciales_email():
+    """Devuelve nuestro objeto OAuth2Credentials para acceder al mail, actualizado.
+    Esta función llama a _refresh() si el token expira en menos de 5 minutos.
+    """
+    return get_credenciales(
+        _creds_email,
+        lambda creds: creds.token_expiry - datetime.timedelta(minutes=5) > datetime.datetime.utcnow(),
+        lambda creds: creds.refresh(httplib2.Http())
+    )

notas_oauth.py

@@ -32,7 +32,7 @@ def sendmail(fromname, toaddr, link):
 	msg["From"] = "{} <{}>".format(fromname, ACCOUNT)
 	msg["To"] = toaddr
 
-	creds = notas_oauth.get_credenciales()
+	creds = notas_oauth.get_credenciales_email()
 	xoauth2_tok = "user={}\1" "auth=Bearer {}\1\1".format(
 	    ACCOUNT, creds.access_token).encode("utf-8")
 	server = smtplib.SMTP('smtp.gmail.com', 587)