#!/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_playlist(user_info):
|
|
return user_info['videoChannels'][0]['id']
|
|
|
|
|
|
def get_playlist_by_name(user_info, options):
|
|
for playlist in user_info["videoChannels"]:
|
|
if playlist['displayName'] == options.get('--playlist'):
|
|
return playlist['id']
|
|
|
|
|
|
def create_playlist(oauth, url, options):
|
|
template = ('Peertube: Playlist %s does not exist, creating it.')
|
|
logging.info(template % (str(options.get('--playlist'))))
|
|
playlist_name = utils.cleanString(str(options.get('--playlist')))
|
|
# Peertube allows 20 chars max for playlist name
|
|
playlist_name = playlist_name[:19]
|
|
data = '{"name":"' + playlist_name +'", \
|
|
"displayName":"' + str(options.get('--playlist')) +'", \
|
|
"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']
|
|
else:
|
|
logging.error(('Peertube: The upload 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]])
|
|
|
|
path = options.get('--file')
|
|
url = str(secret.get('peertube', 'peertube_url')).rstrip('/')
|
|
user_info = get_userinfo()
|
|
|
|
# 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 your tag size")
|
|
exit(1)
|
|
# If Mastodon compatibility is enabled, clean tags from special characters
|
|
if options.get('--mt'):
|
|
strtag = utils.cleanString(strtag)
|
|
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('--playlist'):
|
|
playlist_id = get_playlist_by_name(user_info, options)
|
|
if not playlist_id and options.get('--playlistCreate'):
|
|
playlist_id = create_playlist(oauth, url, options)
|
|
elif not playlist_id:
|
|
logging.warning("Playlist `" + options.get('--playlist') + "` is unknown, using default playlist.")
|
|
playlist_id = get_default_playlist(user_info)
|
|
else:
|
|
playlist_id = get_default_playlist(user_info)
|
|
fields.append(("channelId", str(playlist_id)))
|
|
|
|
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']
|
|
idvideo = str(jresponse['id'])
|
|
logging.info('Peertube : Video was successfully uploaded.')
|
|
template = 'Peertube: Watch it at %s/videos/watch/%s.'
|
|
logging.info(template % (url, uuid))
|
|
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))
|