|
|
- #!/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
- import datetime
- import pytz
- import logging
- from tzlocal import get_localzone
-
- from googleapiclient.discovery import build
- from googleapiclient.errors import HttpError
- from googleapiclient.http import MediaFileUpload
- from google_auth_oauthlib.flow import InstalledAppFlow
-
- import utils
-
- logging.basicConfig(format='%(asctime)s %(message)s', level=logging.INFO)
-
-
- # 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, cache_discovery=False)
-
-
- def initialize_upload(youtube, options):
- path = options.get('--file')
- tags = None
- if options.get('--tags'):
- tags = options.get('--tags').split(',')
-
- category = None
- if options.get('--category'):
- category = utils.getCategory(options.get('--category'), 'youtube')
-
- language = None
- if options.get('--language'):
- language = utils.getLanguage(options.get('--language'), "youtube")
-
- license = None
- if options.get('--cca'):
- license = "creativeCommon"
-
- body = {
- "snippet": {
- "title": options.get('--name') or splitext(basename(path))[0],
- "description": options.get('--description') or "default description",
- "tags": tags,
- # if no category, set default to 1 (Films)
- "categoryId": str(category or 1),
- "defaultAudioLanguage": str(language or 'en')
- },
- "status": {
- "privacyStatus": str(options.get('--privacy') or "private"),
- "license": str(license or "youtube"),
- }
- }
-
- if options.get('--publishAt'):
- # Youtube needs microsecond and the local timezone from ISO 8601
- publishAt = options.get('--publishAt') + ".000001"
- publishAt = datetime.datetime.strptime(publishAt, '%Y-%m-%dT%H:%M:%S.%f')
- tz = get_localzone()
- tz = pytz.timezone(str(tz))
- publishAt = tz.localize(publishAt).isoformat()
- body['status']['publishAt'] = str(publishAt)
-
- # 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:
- logging.info('Youtube : Uploading file...')
- status, response = request.next_chunk()
- if response is not None:
- if 'id' in response:
- template = ('Youtube : Video was successfully '
- 'uploaded.\n'
- 'Watch it at https://youtu.be/%s (post-encoding could take some time)')
- logging.info(template % response['id'])
- else:
- template = ('Youtube : The upload failed with an '
- 'unexpected response: %s')
- logging.error(template % response)
- exit(1)
- 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:
- logging.warning(error)
- retry += 1
- if retry > MAX_RETRIES:
- logging.error('Youtube : No longer attempting to retry.')
- exit(1)
-
- max_sleep = 2 ** retry
- sleep_seconds = random.random() * max_sleep
- logging.warning('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:
- logging.error('Youtube : An HTTP error %d occurred:\n%s' % (e.resp.status,
- e.content))
|