#!/usr/bin/env python2 # coding: utf-8 import os import mimetypes import json import logging import datetime import pytz from os.path import splitext, basename, abspath from tzlocal import get_localzone from ConfigParser import RawConfigParser from requests_oauthlib import OAuth2Session from oauthlib.oauth2 import LegacyApplicationClient from requests_toolbelt.multipart.encoder import MultipartEncoder import utils PEERTUBE_SECRETS_FILE = 'peertube_secret' PEERTUBE_PRIVACY = { "public": 1, "unlisted": 2, "private": 3 } def get_authenticated_service(secret): peertube_url = str(secret.get('peertube', 'peertube_url')).rstrip("/") oauth_client = LegacyApplicationClient( client_id=str(secret.get('peertube', 'client_id')) ) try: oauth = OAuth2Session(client=oauth_client) oauth.fetch_token( token_url=str(peertube_url + '/api/v1/users/token'), # lower as peertube does not store uppercase for pseudo username=str(secret.get('peertube', 'username').lower()), password=str(secret.get('peertube', 'password')), client_id=str(secret.get('peertube', 'client_id')), client_secret=str(secret.get('peertube', 'client_secret')) ) except Exception as e: if hasattr(e, 'message'): logging.error("Peertube: Error: " + str(e.message)) exit(1) else: logging.error("Peertube: Error: " + str(e)) exit(1) return oauth def get_default_channel(user_info): return user_info['videoChannels'][0]['id'] def get_channel_by_name(user_info, options): for channel in user_info["videoChannels"]: if channel['displayName'].encode('utf8') == str(options.get('--channel')): return channel['id'] def create_channel(oauth, url, options): template = ('Peertube: Channel %s does not exist, creating it.') logging.info(template % (str(options.get('--channel')))) channel_name = utils.cleanString(str(options.get('--channel'))) # Peertube allows 20 chars max for channel name channel_name = channel_name[:19] data = '{"name":"' + channel_name +'", \ "displayName":"' + str(options.get('--channel')) +'", \ "description":null}' headers = { 'Content-Type': "application/json" } try: response = oauth.post(url + "/api/v1/video-channels/", data=data, headers=headers) except Exception as e: if hasattr(e, 'message'): logging.error("Error: " + str(e.message)) else: logging.error("Error: " + str(e)) if response is not None: if response.status_code == 200: jresponse = response.json() jresponse = jresponse['videoChannel'] return jresponse['id'] if response.status_code == 409: logging.error('Peertube: Error: It seems there is a conflict with an existing channel, please beware ' 'Peertube internal name is compiled from 20 firsts characters of channel name.' ' Please check your channel name and retry.') exit(1) else: logging.error(('Peertube: Creating channel failed with an unexpected response: ' '%s') % response) exit(1) def get_default_playlist(user_info): return user_info['videoChannels'][0]['id'] def get_playlist_by_name(user_playlists, options): for playlist in user_playlists["data"]: if playlist['displayName'].encode('utf8') == str(options.get('--playlist')): return playlist['id'] def create_playlist(oauth, url, options, channel): template = ('Peertube: Playlist %s does not exist, creating it.') logging.info(template % (str(options.get('--playlist')))) # We use files for form-data Content # see https://requests.readthedocs.io/en/latest/user/quickstart/#post-a-multipart-encoded-file # None is used to mute "filename" field files = {'displayName': (None, str(options.get('--playlist'))), 'privacy': (None, "1"), 'description': (None, "null"), 'videoChannelId': (None, str(channel)), 'thumbnailfile': (None, "null")} try: response = oauth.post(url + "/api/v1/video-playlists/", files=files) except Exception as e: if hasattr(e, 'message'): logging.error("Error: " + str(e.message)) else: logging.error("Error: " + str(e)) if response is not None: if response.status_code == 200: jresponse = response.json() jresponse = jresponse['videoPlaylist'] return jresponse['id'] else: logging.error(('Peertube: Creating the playlist failed with an unexpected response: ' '%s') % response) exit(1) def set_playlist(oauth, url, video_id, playlist_id): logging.info('Peertube: add video to playlist.') data = '{"videoId":"' + str(video_id) + '"}' headers = { 'Content-Type': "application/json" } try: response = oauth.post(url + "/api/v1/video-playlists/"+str(playlist_id)+"/videos", data=data, headers=headers) except Exception as e: if hasattr(e, 'message'): logging.error("Error: " + str(e.message)) else: logging.error("Error: " + str(e)) if response is not None: if response.status_code == 200: logging.info('Peertube: Video is successfully added to the playlist.') else: logging.error(('Peertube: Configuring the playlist failed with an unexpected response: ' '%s') % response) exit(1) def upload_video(oauth, secret, options): def get_userinfo(): return json.loads(oauth.get(url+"/api/v1/users/me").content) def get_file(path): mimetypes.init() return (basename(path), open(abspath(path), 'rb'), mimetypes.types_map[splitext(path)[1]]) def get_playlist(username): return json.loads(oauth.get(url+"/api/v1/accounts/"+username+"/video-playlists").content) path = options.get('--file') url = str(secret.get('peertube', 'peertube_url')).rstrip('/') user_info = get_userinfo() user_playlists = get_playlist(str(secret.get('peertube', 'username').lower())) # We need to transform fields into tuple to deal with tags as # MultipartEncoder does not support list refer # https://github.com/requests/toolbelt/issues/190 and # https://github.com/requests/toolbelt/issues/205 fields = [ ("name", options.get('--name') or splitext(basename(options.get('--file')))[0]), ("licence", "1"), ("description", options.get('--description') or "default description"), ("nsfw", str(int(options.get('--nsfw')) or "0")), ("videofile", get_file(path)) ] if options.get('--tags'): tags = options.get('--tags').split(',') for strtag in tags: # Empty tag crashes Peertube, so skip them if strtag == "": continue # Tag more than 30 chars crashes Peertube, so exit and check tags if len(strtag) >= 30: logging.warning("Peertube: Sorry, Peertube does not support tag with more than 30 characters, please reduce tag: " + strtag) exit(1) fields.append(("tags[]", strtag)) if options.get('--category'): fields.append(("category", str(utils.getCategory(options.get('--category'), 'peertube')))) else: # if no category, set default to 2 (Films) fields.append(("category", "2")) if options.get('--language'): fields.append(("language", str(utils.getLanguage(options.get('--language'), "peertube")))) else: # if no language, set default to 1 (English) fields.append(("language", "en")) if options.get('--disable-comments'): fields.append(("commentsEnabled", "0")) else: fields.append(("commentsEnabled", "1")) privacy = None if options.get('--privacy'): privacy = options.get('--privacy').lower() if options.get('--publishAt'): publishAt = options.get('--publishAt') publishAt = datetime.datetime.strptime(publishAt, '%Y-%m-%dT%H:%M:%S') tz = get_localzone() tz = pytz.timezone(str(tz)) publishAt = tz.localize(publishAt).isoformat() fields.append(("scheduleUpdate[updateAt]", publishAt)) fields.append(("scheduleUpdate[privacy]", str(PEERTUBE_PRIVACY["public"]))) fields.append(("privacy", str(PEERTUBE_PRIVACY["private"]))) else: fields.append(("privacy", str(PEERTUBE_PRIVACY[privacy or "private"]))) if options.get('--thumbnail'): fields.append(("thumbnailfile", get_file(options.get('--thumbnail')))) fields.append(("previewfile", get_file(options.get('--thumbnail')))) if options.get('--channel'): channel_id = get_channel_by_name(user_info, options) if not channel_id and options.get('--channelCreate'): channel_id = create_channel(oauth, url, options) elif not channel_id: logging.warning("Channel `" + options.get('--channel') + "` is unknown, using default channel.") channel_id = get_default_channel(user_info) else: channel_id = get_default_channel(user_info) fields.append(("channelId", str(channel_id))) if options.get('--playlist'): playlist_id = get_playlist_by_name(user_playlists, options) if not playlist_id and options.get('--playlistCreate'): playlist_id = create_playlist(oauth, url, options, channel_id) elif not playlist_id: logging.warning("Playlist `" + options.get('--playlist') + "` does not exist, please set --playlistCreate" " if you want to create it") exit(1) multipart_data = MultipartEncoder(fields) headers = { 'Content-Type': multipart_data.content_type } response = oauth.post(url + "/api/v1/videos/upload", data=multipart_data, headers=headers) if response is not None: if response.status_code == 200: jresponse = response.json() jresponse = jresponse['video'] uuid = jresponse['uuid'] video_id = str(jresponse['id']) logging.info('Peertube : Video was successfully uploaded.') template = 'Peertube: Watch it at %s/videos/watch/%s.' logging.info(template % (url, uuid)) # Upload is successful we may set playlist if options.get('--playlist'): set_playlist(oauth, url, video_id, playlist_id) else: logging.error(('Peertube: The upload failed with an unexpected response: ' '%s') % response) exit(1) def run(options): secret = RawConfigParser() try: secret.read(PEERTUBE_SECRETS_FILE) except Exception as e: logging.error("Peertube: Error loading " + str(PEERTUBE_SECRETS_FILE) + ": " + str(e)) exit(1) insecure_transport = secret.get('peertube', 'OAUTHLIB_INSECURE_TRANSPORT') os.environ['OAUTHLIB_INSECURE_TRANSPORT'] = insecure_transport oauth = get_authenticated_service(secret) try: logging.info('Peertube: Uploading video...') upload_video(oauth, secret, options) except Exception as e: if hasattr(e, 'message'): logging.error("Peertube: Error: " + str(e.message)) else: logging.error("Peertube: Error: " + str(e))