|
#!/usr/bin/python
|
|
# coding: utf-8
|
|
|
|
import os
|
|
import mimetypes
|
|
import httplib
|
|
import httplib2
|
|
import json
|
|
import array
|
|
|
|
from ConfigParser import RawConfigParser
|
|
from requests_oauthlib import OAuth2Session
|
|
from oauthlib.oauth2 import LegacyApplicationClient
|
|
from requests_toolbelt.multipart.encoder import MultipartEncoder
|
|
|
|
PEERTUBE_SECRETS_FILE = 'peertube_secret'
|
|
|
|
def get_authenticated_service(config):
|
|
oauth = OAuth2Session(client=LegacyApplicationClient(client_id=str(config.get('peertube', 'client_id'))))
|
|
oauth.fetch_token(token_url=str(config.get('peertube', 'peertube_url')) + '/api/v1/users/token',
|
|
username=str(config.get('peertube', 'username').lower()), #lower as peertube does not store uppecase for pseudo
|
|
password=str(config.get('peertube', 'password')),
|
|
client_id=str(config.get('peertube', 'client_id')),
|
|
client_secret=str(config.get('peertube', 'client_secret'))
|
|
)
|
|
return oauth
|
|
|
|
def upload_video(oauth, config, options):
|
|
|
|
def get_userinfo():
|
|
user_info = json.loads(oauth.get(url+"/api/v1/users/me").content)
|
|
return str(user_info["id"])
|
|
|
|
def get_videofile(path):
|
|
mimetypes.init()
|
|
return (os.path.basename(path), open(os.path.abspath(path), 'rb'),
|
|
mimetypes.types_map[os.path.splitext(path)[1]])
|
|
|
|
path = options.get('--file')
|
|
url = config.get('peertube', 'peertube_url')
|
|
tags = None
|
|
tags_tuple=[]
|
|
|
|
# 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 os.path.splitext(os.path.basename(path))[0]),
|
|
("category", str(options.get('--category') or 1)), # look at the list numbers at /videos/categories
|
|
("licence", str(options.get('--licence') or 1)), # look at the list numbers at /videos/licences
|
|
("description", options.get('--description') or "default description"),
|
|
("privacy", str(options.get('--privacy') or 3)), # look at the list numbers at /videos/privacies
|
|
("nsfw", str(options.get('--nsfw') or 0)),
|
|
("commentsEnabled", "1"),
|
|
("channelId", get_userinfo()),
|
|
("videofile", get_videofile(path)) # beware, see validateVideo for supported types
|
|
]
|
|
|
|
if options.get('--tags'):
|
|
tags = options.get('--tags').split(',')
|
|
for strtags in tags:
|
|
fields.append(("tags", strtags))
|
|
|
|
# multipart_data = MultipartEncoder(fields=fields)
|
|
multipart_data = MultipartEncoder(fields)
|
|
|
|
headers = {
|
|
'Content-Type': multipart_data.content_type
|
|
}
|
|
|
|
response = oauth.post(config.get('peertube', 'peertube_url')+"/api/v1/videos/upload", data=multipart_data, headers=headers)
|
|
if response is not None:
|
|
if response.status_code == 200:
|
|
print 'Peertube : Video was successfully uploaded.'
|
|
else:
|
|
exit('Peertube : The upload failed with an unexpected response: %s' % response)
|
|
|
|
|
|
def run(options):
|
|
config = RawConfigParser()
|
|
config.read(PEERTUBE_SECRETS_FILE)
|
|
os.environ['OAUTHLIB_INSECURE_TRANSPORT'] = config.get('peertube', 'OAUTHLIB_INSECURE_TRANSPORT')
|
|
oauth = get_authenticated_service(config)
|
|
try:
|
|
print 'Peertube : Uploading file...'
|
|
upload_video(oauth, config, options)
|
|
except Exception as e:
|
|
if hasattr(e, 'message'):
|
|
print(e.message)
|
|
else:
|
|
print(e)
|