Como usar deferred.defer com oauth2 no Google App Engine?

Question

Eu tenho decorador:

http = httplib2.Http()
service = discovery.build('gmail', 'v1', http=http)
# Creating decorator for OAuth2 account.
decorator = appengine.oauth2decorator_from_clientsecrets(
    CLIENT_SECRETS,
    scope='https://www.googleapis.com/auth/gmail.readonly',
    message=MISSING_CLIENT_SECRETS_MESSAGE)

E classe que o utiliza:

class CSVGeneratorHandler(webapp2.RedirectHandler):
    @decorator.oauth_required
    def get(self):
        http = decorator.http()
        first_messages = service.users().messages().list(userId='me').execute(http=http)
        template = JINJA_ENVIRONMENT.get_template('templates/success.html')
        self.response.write(template.render({}))

Isso funciona muito bem.

Mas quando tento mover a solicitação da API service.users().messages().list(userId='me').execute(http=http) para a tarefa (com deferred.defer(get_mails)) recebi um erro:"Login necessário".

Entendo que a função get_mails deve receber contexto e/ou credenciais do programa principal.Mas não entendo como fazer isso.

Alguém pode me ajudar?

Answer 1

Depois de algumas horas, encontrei a resposta.

Primeiro de tudo você precisa salvar as credenciais em algum lugar.Como nós, você GAE, temos um tipo de campo especial para isso:

from oauth2client.appengine import StorageByKeyName  
from google.appengine.ext import db

class CredentialsModel(db.Model):
  credentials = CredentialsProperty()

O próximo passo é configurar a autenticação sem decorador e só depois enviar a tarefa para a fila:

    user = users.get_current_user()
    flow = flow_from_clientsecrets(os.path.join(os.path.dirname(os.path.realpath(__file__)), CLIENT_SECRETS),
                           scope='https://www.googleapis.com/auth/gmail.readonly',
                           redirect_uri='http://127.0.0.1:8080/success')
    auth_uri = str(flow.step1_get_authorize_url())
    code = self.request.get('code')
    if not code:
        return self.redirect(auth_uri)
    credentials = flow.step2_exchange(code)

    storage = StorageByKeyName(CredentialsModel, user.user_id(), 'credentials')
    storage.put(credentials)

    deferred.defer(get_mails, user.user_id())

Não se esqueça de obter credenciais na função diferida!

def get_mails(user_id):
    storage = StorageByKeyName(CredentialsModel, user_id, 'credentials')
    credentials = storage.get()

    http = httplib2.Http()
    http = credentials.authorize(http)

    all_messages = []
    service = discovery.build('gmail', 'v1', http=http)
    first_messages = service.users().messages().list(userId='me').execute()