pep8

lib/yt_upload.py

@@ -1,18 +1,15 @@
 #!/usr/bin/python
 # coding: utf-8
-# From Youtube samples : https://raw.githubusercontent.com/youtube/api-samples/master/python/upload_video.py
+# From Youtube samples : https://raw.githubusercontent.com/youtube/api-samples/master/python/upload_video.py  # noqa
 
-import argparse
 import httplib
 import httplib2
-import os
 import random
 import time
 import copy
 import json
-
+from os.path import splitext, basename, exists
 import google.oauth2.credentials
-import google_auth_oauthlib.flow
 
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
@@ -28,10 +25,17 @@ httplib2.RETRIES = 1
 MAX_RETRIES = 10
 
 # Youtube retriables cases
-RETRIABLE_EXCEPTIONS = (httplib2.HttpLib2Error, IOError, httplib.NotConnected,
-  httplib.IncompleteRead, httplib.ImproperConnectionState,
-  httplib.CannotSendRequest, httplib.CannotSendHeader,
-  httplib.ResponseNotReady, httplib.BadStatusLine)
+RETRIABLE_EXCEPTIONS = (
+    IOError,
+    httplib2.HttpLib2Error,
+    httplib.NotConnected,
+    httplib.IncompleteRead,
+    httplib.ImproperConnectionState,
+    httplib.CannotSendRequest,
+    httplib.CannotSendHeader,
+    httplib.ResponseNotReady,
+    httplib.BadStatusLine,
+)
 
 RETRIABLE_STATUS_CODES = [500, 502, 503, 504]
 
@@ -45,91 +49,98 @@ API_VERSION = 'v3'
 
 # Authorize the request and store authorization credentials.
 def get_authenticated_service():
-  flow = InstalledAppFlow.from_client_secrets_file(CLIENT_SECRETS_FILE, SCOPES)
-  if os.path.exists(CREDENTIALS_PATH):
-    with open(CREDENTIALS_PATH, 'r') as f:
-        credential_params = json.load(f)
-        credentials = google.oauth2.credentials.Credentials(
-            credential_params["token"],
-            refresh_token=credential_params["_refresh_token"],
-            token_uri=credential_params["_token_uri"],
-            client_id=credential_params["_client_id"],
-            client_secret=credential_params["_client_secret"]
-        )
-  else:
-    credentials = flow.run_local_server()
-    with open(CREDENTIALS_PATH, 'w') as f:
-      p = copy.deepcopy(vars(credentials))
-      del p["expiry"]
-      json.dump(p, f)
-  return build(API_SERVICE_NAME, API_VERSION, credentials = credentials)
+    flow = InstalledAppFlow.from_client_secrets_file(
+        CLIENT_SECRETS_FILE, SCOPES)
+    if exists(CREDENTIALS_PATH):
+        with open(CREDENTIALS_PATH, 'r') as f:
+            credential_params = json.load(f)
+            credentials = google.oauth2.credentials.Credentials(
+                credential_params["token"],
+                refresh_token=credential_params["_refresh_token"],
+                token_uri=credential_params["_token_uri"],
+                client_id=credential_params["_client_id"],
+                client_secret=credential_params["_client_secret"]
+            )
+    else:
+        credentials = flow.run_local_server()
+        with open(CREDENTIALS_PATH, 'w') as f:
+            p = copy.deepcopy(vars(credentials))
+            del p["expiry"]
+            json.dump(p, f)
+    return build(API_SERVICE_NAME, API_VERSION, credentials=credentials)
+
 
 def initialize_upload(youtube, options):
-  path = options.get('--file')
-  tags = None
-  if options.get('--tags'):
-    tags = options.get('--tags').split(',')
+    path = options.get('--file')
+    tags = None
+    if options.get('--tags'):
+        tags = options.get('--tags').split(',')
 
-  body=dict(
-    snippet=dict(
-      title=options.get('--name') or os.path.splitext(os.path.basename(path))[0],
-      description=options.get('--description') or "",
-      tags=tags,
-      categoryId=str(options.get('--category') or 1),
-    ),
-    status=dict(
-      privacyStatus=str(options.get('--privacy') or "private"),
+    body = {
+        "snippet": {
+            "title": options.get('--name') or splitext(basename(path))[0],
+            "description": options.get('--description') or "",
+            "tags": tags,
+            "categoryId": str(options.get('--category') or 1),
+        },
+        "status": {"privacyStatus": str(options.get('--privacy') or "private")}
+    }
+
+    # Call the API's videos.insert method to create and upload the video.
+    insert_request = youtube.videos().insert(
+        part=','.join(body.keys()),
+        body=body,
+        media_body=MediaFileUpload(path, chunksize=-1, resumable=True)
     )
-  )
 
-  # Call the API's videos.insert method to create and upload the video.
-  insert_request = youtube.videos().insert(
-    part=','.join(body.keys()),
-    body=body,
-    media_body=MediaFileUpload(path, chunksize=-1, resumable=True)
-  )
+    resumable_upload(insert_request)
 
-  resumable_upload(insert_request)
 
 # This method implements an exponential backoff strategy to resume a
 # failed upload.
 def resumable_upload(request):
-  response = None
-  error = None
-  retry = 0
-  while response is None:
-    try:
-      print 'Youtube : Uploading file...'
-      status, response = request.next_chunk()
-      if response is not None:
-        if 'id' in response:
-          print 'Youtube : Video id "%s" was successfully uploaded.' % response['id']
-        else:
-          exit('Youtube : The upload failed with an unexpected response: %s' % response)
-    except HttpError, e:
-      if e.resp.status in RETRIABLE_STATUS_CODES:
-        error = 'Youtube : A retriable HTTP error %d occurred:\n%s' % (e.resp.status,
-                                                             e.content)
-      else:
-        raise
-    except RETRIABLE_EXCEPTIONS, e:
-      error = 'Youtube : A retriable error occurred: %s' % e
+    response = None
+    error = None
+    retry = 0
+    while response is None:
+        try:
+            print('Youtube : Uploading file...')
+            status, response = request.next_chunk()
+            if response is not None:
+                if 'id' in response:
+                    template = ('Youtube : Video id "%s" was successfully '
+                                'uploaded.')
+                    print(template % response['id'])
+                else:
+                    template = ('Youtube : The upload failed with an '
+                                'unexpected response: %s')
+                    exit(template % response)
+        except HttpError as e:
+            if e.resp.status in RETRIABLE_STATUS_CODES:
+                template = 'Youtube : A retriable HTTP error %d occurred:\n%s'
+                error = template % (e.resp.status, e.content)
+            else:
+                raise
+        except RETRIABLE_EXCEPTIONS as e:
+            error = 'Youtube : A retriable error occurred: %s' % e
 
     if error is not None:
-      print error
-      retry += 1
-      if retry > MAX_RETRIES:
-        exit('Youtube : No longer attempting to retry.')
+        print(error)
+        retry += 1
+        if retry > MAX_RETRIES:
+            exit('Youtube : No longer attempting to retry.')
 
-      max_sleep = 2 ** retry
-      sleep_seconds = random.random() * max_sleep
-      print 'Youtube : Sleeping %f seconds and then retrying...' % sleep_seconds
-      time.sleep(sleep_seconds)
+        max_sleep = 2 ** retry
+        sleep_seconds = random.random() * max_sleep
+        print('Youtube : Sleeping %f seconds and then retrying...'
+              % sleep_seconds)
+        time.sleep(sleep_seconds)
 
 
 def run(options):
-  youtube = get_authenticated_service()
-  try:
-    initialize_upload(youtube, options)
-  except HttpError, e:
-    print 'Youtube : An HTTP error %d occurred:\n%s' % (e.resp.status, e.content)
+    youtube = get_authenticated_service()
+    try:
+        initialize_upload(youtube, options)
+    except HttpError as e:
+        print('Youtube : An HTTP error %d occurred:\n%s' % (e.resp.status,
+                                                            e.content))