|
|
- #!/usr/bin/python
- # coding: utf-8
- # From Youtube samples : https://raw.githubusercontent.com/youtube/api-samples/master/python/upload_video.py # noqa
-
- import httplib
- import httplib2
- import random
- import time
- import copy
- import json
- from os.path import splitext, basename, exists
- import google.oauth2.credentials
-
- from googleapiclient.discovery import build
- from googleapiclient.errors import HttpError
- from googleapiclient.http import MediaFileUpload
- from google_auth_oauthlib.flow import InstalledAppFlow
-
-
- # Explicitly tell the underlying HTTP transport library not to retry, since
- # we are handling retry logic ourselves.
- httplib2.RETRIES = 1
-
- # Maximum number of times to retry before giving up.
- MAX_RETRIES = 10
-
- # Youtube retriables cases
- 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]
-
-
- CLIENT_SECRETS_FILE = 'youtube_secret.json'
- CREDENTIALS_PATH = ".youtube_credentials.json"
- SCOPES = ['https://www.googleapis.com/auth/youtube.upload']
- API_SERVICE_NAME = 'youtube'
- 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 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(',')
-
- 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)
- )
-
- 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:
- 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.')
-
- 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 as e:
- print('Youtube : An HTTP error %d occurred:\n%s' % (e.resp.status,
- e.content))
|