From e2cf769b202c39a8961d44a90059626f154aa214 Mon Sep 17 00:00:00 2001 From: Jay Lee Date: Tue, 24 Jan 2017 14:48:44 -0500 Subject: [PATCH] short URLs fix back --- src/oauth2client/tools.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/oauth2client/tools.py b/src/oauth2client/tools.py index b882429f..f15dfc12 100644 --- a/src/oauth2client/tools.py +++ b/src/oauth2client/tools.py @@ -217,6 +217,16 @@ def run_flow(flow, storage, flags=None, http=None): flow.redirect_uri = oauth_callback authorize_url = flow.step1_get_authorize_url() + if flags.short_url: + try: + from googleapiclient.discovery import build + service = build('urlshortener', 'v1', http=http) + url_result = service.url().insert(body={'longUrl': authorize_url}, + key=u'AIzaSyBlmgbii8QfJSYmC9VTMOfqrAt5Vj5wtzE').execute() + authorize_url = url_result['id'] + except: + pass + if not flags.noauth_local_webserver: import webbrowser webbrowser.open(authorize_url, new=1, autoraise=True)