diff --git a/prismedia/yt_upload.py b/prismedia/yt_upload.py index e8809c5..b00dd52 100644 --- a/prismedia/yt_upload.py +++ b/prismedia/yt_upload.py @@ -48,8 +48,8 @@ RETRIABLE_EXCEPTIONS = ( RETRIABLE_STATUS_CODES = [500, 502, 503, 504] -CLIENT_SECRETS_FILE = 'youtube_secret.json' -CREDENTIALS_PATH = ".youtube_credentials.json" +CLIENT_SECRETS_FILE_BASE = 'youtube_secret.json' +CREDENTIALS_PATH_BASE = ".youtube_credentials.json" SCOPES = ['https://www.googleapis.com/auth/youtube.upload', 'https://www.googleapis.com/auth/youtube.force-ssl'] API_SERVICE_NAME = 'youtube' API_VERSION = 'v3' @@ -358,6 +358,13 @@ def hearthbeat(): def run(options): + global CLIENT_SECRETS_FILE, CREDENTIALS_PATH + if options.get('--credentialsdir') : + CLIENT_SECRETS_FILE = os.path.join(options.get('--credentialsdir'), CLIENT_SECRETS_FILE_BASE) + CREDENTIALS_PATH = os.path.join(options.get('--credentialsdir'), CREDENTIALS_PATH_BASE) + else : + CLIENT_SECRETS_FILE = CLIENT_SECRETS_FILE_BASE + CREDENTIALS_PATH = CREDENTIALS_PATH_BASE youtube = get_authenticated_service() try: initialize_upload(youtube, options)