Nathan
112 lines
3.6 KiB
Python
112 lines
3.6 KiB
Python
|
|
import json
|
|
import os
|
|
import bpy
|
|
|
|
from time import sleep
|
|
|
|
# import the requests library
|
|
# http://docs.python-requests.org/en/latest
|
|
# pip install requests
|
|
import requests
|
|
#from .utils import print_message
|
|
from .messages import print_message
|
|
|
|
##
|
|
# Uploading a model to Sketchfab is a two step process
|
|
#
|
|
# 1. Upload a model. If the upload is successful, the API will return
|
|
# the model's uid in the `Location` header, and the model will be placed in the processing queue
|
|
#
|
|
# 2. Poll for the processing status
|
|
# You can use your model id (see 1.) to poll the model processing status
|
|
# The processing status can be one of the following:
|
|
# - PENDING: the model is in the processing queue
|
|
# - PROCESSING: the model is being processed
|
|
# - SUCCESSED: the model has being sucessfully processed and can be view on sketchfab.com
|
|
# - FAILED: the processing has failed. An error message detailing the reason for the failure
|
|
# will be returned with the response
|
|
#
|
|
# HINTS
|
|
# - limit the rate at which you poll for the status (once every few seconds is more than enough)
|
|
##
|
|
|
|
SKETCHFAB_DOMAIN = 'sketchfab.com'
|
|
SKETCHFAB_API_URL = 'https://api.{}/v3'.format(SKETCHFAB_DOMAIN)
|
|
|
|
|
|
def _get_request_payload(apikey, data={}, files={}, json_payload=False):
|
|
"""Helper method that returns the authentication token and proper content
|
|
type depending on whether or not we use JSON payload."""
|
|
headers = {'Authorization': 'Token {}'.format(apikey)}
|
|
|
|
if json_payload:
|
|
headers.update({'Content-Type': 'application/json'})
|
|
data = json.dumps(data)
|
|
|
|
return {'data': data, 'files': files, 'headers': headers}
|
|
|
|
|
|
def upload(model_file, uploadname, apikey):
|
|
"""POST a model to sketchfab.
|
|
|
|
This endpoint only accepts formData as we upload a file.
|
|
"""
|
|
#model_endpoint = os.path.join(SKETCHFAB_API_URL, 'models')
|
|
|
|
model_endpoint = SKETCHFAB_API_URL + '/models'
|
|
|
|
print_message(bpy.context, f"Sketchfab URL is going to be: {model_endpoint}")
|
|
|
|
# Optional parameters
|
|
name = uploadname
|
|
description = 'Uploaded with SimpleBake for Blender 2.8'
|
|
#password = 'my-password' # requires a pro account
|
|
#private = 1 # requires a pro account
|
|
#tags = ['bob', 'character', 'video-games'] # Array of tags
|
|
#categories = ['people'] # Array of categories slugs
|
|
#license = 'CC Attribution' # License label
|
|
isPublished = False, # Model will be on draft instead of published
|
|
#isInspectable = True, # Allow 2D view in model inspector
|
|
|
|
data = {
|
|
'name': name,
|
|
'description': description,
|
|
#'tags': tags,
|
|
#'categories': categories,
|
|
#'license': license,
|
|
#'private': private,
|
|
#'password': password,
|
|
'isPublished': isPublished,
|
|
'source': "simplebake-for-blender"
|
|
#'isInspectable': isInspectable
|
|
}
|
|
|
|
f = open(model_file, 'rb')
|
|
|
|
files = {'modelFile': f}
|
|
|
|
try:
|
|
r = requests.post(
|
|
model_endpoint, **_get_request_payload(apikey,
|
|
data, files=files))
|
|
except requests.exceptions.RequestException as e:
|
|
print_message(bpy.context, 'An error occured: {}'.format(e))
|
|
return False
|
|
finally:
|
|
f.close()
|
|
|
|
print_message(bpy.context, f"Status code from Sketchfab was {r.status_code}");
|
|
|
|
if r.status_code != requests.codes.created:
|
|
print_message(bpy.context, 'Upload failed with error: {}'.format(r.json()))
|
|
return False
|
|
|
|
# Should be https://api.sketchfab.com/v3/models/XXXX
|
|
model_url = r.headers['Location']
|
|
|
|
#For some reason this is wrong. Correct it here.
|
|
return model_url.replace("https://api.sketchfab.com/v3/models/", "https://sketchfab.com/3d-models/")
|
|
|
|
|